diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-11-08 05:35:01 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-11-08 05:35:01 +0000 |
commit | c5334fc56b3d44976425da2e6d0a303ae96836a1 (patch) | |
tree | b4c4cd968ef3f5272d78805b9befe52be1c3f0d9 /modules/ui.py | |
parent | 804d9fb83d0c63ca3acd36378707ce47b8f12599 (diff) | |
download | stable-diffusion-webui-gfx803-c5334fc56b3d44976425da2e6d0a303ae96836a1.tar.gz stable-diffusion-webui-gfx803-c5334fc56b3d44976425da2e6d0a303ae96836a1.tar.bz2 stable-diffusion-webui-gfx803-c5334fc56b3d44976425da2e6d0a303ae96836a1.zip |
fix javascript duplication bug after pressing the restart UI button
Diffstat (limited to 'modules/ui.py')
-rw-r--r-- | modules/ui.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/modules/ui.py b/modules/ui.py index 34c31ef1..67cf1d6a 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1752,7 +1752,7 @@ def create_ui(wrap_gradio_gpu_call): return demo
-def load_javascript(raw_response):
+def reload_javascript():
with open(os.path.join(script_path, "script.js"), "r", encoding="utf8") as jsfile:
javascript = f'<script>{jsfile.read()}</script>'
@@ -1768,7 +1768,7 @@ def load_javascript(raw_response): javascript += f"\n<script>{localization.localization_js(shared.opts.localization)}</script>"
def template_response(*args, **kwargs):
- res = raw_response(*args, **kwargs)
+ res = shared.GradioTemplateResponseOriginal(*args, **kwargs)
res.body = res.body.replace(
b'</head>', f'{javascript}</head>'.encode("utf8"))
res.init_headers()
@@ -1777,4 +1777,5 @@ def load_javascript(raw_response): gradio.routes.templates.TemplateResponse = template_response
-reload_javascript = partial(load_javascript, gradio.routes.templates.TemplateResponse)
+if not hasattr(shared, 'GradioTemplateResponseOriginal'):
+ shared.GradioTemplateResponseOriginal = gradio.routes.templates.TemplateResponse
|