OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-31 18:52:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove preconnects

Browse Source
This commit is contained in:
w-e-w 2024-08-10 18:40:38 +09:00
parent 48239090f1
commit eed3b43252
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/ui_gradio_extensions.py
View File

@ -1,4 +1,5 @@
import os import os
import re
import gradio as gr import gradio as gr
from modules import localization, shared, scripts, util from modules import localization, shared, scripts, util
@ -49,12 +50,19 @@ def css_html():
return head return head
re_preconnect = re.compile(rb'<link\s+rel="preconnect"\s+href="([^"]+)"(?:\s+[^>]*)?/?>')
def reload_javascript(): def reload_javascript():
js = javascript_html() js = javascript_html()
css = css_html() css = css_html()
def template_response(*args, **kwargs): def template_response(*args, **kwargs):
res = shared.GradioTemplateResponseOriginal(*args, **kwargs) res = shared.GradioTemplateResponseOriginal(*args, **kwargs)
# remove preconnects
res.body = re_preconnect.sub(b'', res.body)
res.body = res.body.replace(b'</head>', f'{js}<meta name="referrer" content="no-referrer"/></head>'.encode("utf8")) res.body = res.body.replace(b'</head>', f'{js}<meta name="referrer" content="no-referrer"/></head>'.encode("utf8"))
res.body = res.body.replace(b'</body>', f'{css}</body>'.encode("utf8")) res.body = res.body.replace(b'</body>', f'{css}</body>'.encode("utf8"))
res.init_headers() res.init_headers()