add redefinition guard to gradio_routes_templates_response

afaa03c5 · DepFA · GitHub · a9044475 · afaa03c5
Commit afaa03c5 authored Oct 01, 2022 by DepFA Committed by GitHub Oct 01, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 7 deletions

modules/ui.py modules/ui.py +8 -7

No files found.
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1219,12 +1219,13 @@ for filename in sorted(os.listdir(jsdir)):
        javascript += f"\n<script>{jsfile.read()}</script>"


-def template_response(*args, **kwargs):
-    res = gradio_routes_templates_response(*args, **kwargs)
-    res.body = res.body.replace(b'</head>', f'{javascript}</head>'.encode("utf8"))
-    res.init_headers()
-    return res
+if 'gradio_routes_templates_response' not in globals():
+  def template_response(*args, **kwargs):
+      res = gradio_routes_templates_response(*args, **kwargs)
+      res.body = res.body.replace(b'</head>', f'{javascript}</head>'.encode("utf8"))
+      res.init_headers()
+      return res

+  gradio_routes_templates_response = gradio.routes.templates.TemplateResponse
+  gradio.routes.templates.TemplateResponse = template_response

-gradio_routes_templates_response = gradio.routes.templates.TemplateResponse
-gradio.routes.templates.TemplateResponse = template_response