mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
import os
|
|
import gradio as gr
|
|
|
|
from modules import localization, shared, scripts, util
|
|
from modules.paths import script_path, data_path
|
|
|
|
|
|
def webpath(fn):
|
|
return f'file={util.truncate_path(fn)}?{os.path.getmtime(fn)}'
|
|
|
|
|
|
def javascript_html():
|
|
# Ensure localization is in `window` before scripts
|
|
head = f'<script type="text/javascript">{localization.localization_js(shared.opts.localization)}</script>\n'
|
|
|
|
script_js = os.path.join(script_path, "script.js")
|
|
head += f'<script type="text/javascript" src="{webpath(script_js)}"></script>\n'
|
|
|
|
# We want the utils.js script to be imported first since it can be used by multiple other
|
|
# scripts. This resolves dependency issues caused by html <script> ordering.
|
|
js_scripts = scripts.list_scripts("javascript", ".js")
|
|
js_utils_script = [x for x in js_scripts if x.filename == "utils.js"][0]
|
|
head += f'<script type="text/javascript" src="{webpath(js_utils_script.path)}"></script>\n'
|
|
|
|
# Now add all remaining .js scripts excluding the utils.js file.
|
|
for script in [x for x in js_scripts if x.filename != "utils.js"]:
|
|
head += f'<script type="text/javascript" src="{webpath(script.path)}"></script>\n'
|
|
|
|
for script in scripts.list_scripts("javascript", ".mjs"):
|
|
head += f'<script type="module" src="{webpath(script.path)}"></script>\n'
|
|
|
|
if shared.cmd_opts.theme:
|
|
head += f'<script type="text/javascript">set_theme(\"{shared.cmd_opts.theme}\");</script>\n'
|
|
|
|
return head
|
|
|
|
|
|
def css_html():
|
|
head = ""
|
|
|
|
def stylesheet(fn):
|
|
return f'<link rel="stylesheet" property="stylesheet" href="{webpath(fn)}">'
|
|
|
|
for cssfile in scripts.list_files_with_name("style.css"):
|
|
head += stylesheet(cssfile)
|
|
|
|
user_css = os.path.join(data_path, "user.css")
|
|
if os.path.exists(user_css):
|
|
head += stylesheet(user_css)
|
|
|
|
return head
|
|
|
|
|
|
def reload_javascript():
|
|
js = javascript_html()
|
|
css = css_html()
|
|
|
|
def template_response(*args, **kwargs):
|
|
res = shared.GradioTemplateResponseOriginal(*args, **kwargs)
|
|
res.body = res.body.replace(b'</head>', f'{js}</head>'.encode("utf8"))
|
|
res.body = res.body.replace(b'</body>', f'{css}</body>'.encode("utf8"))
|
|
res.init_headers()
|
|
return res
|
|
|
|
gr.routes.templates.TemplateResponse = template_response
|
|
|
|
|
|
if not hasattr(shared, 'GradioTemplateResponseOriginal'):
|
|
shared.GradioTemplateResponseOriginal = gr.routes.templates.TemplateResponse
|