File size: 1,605 Bytes
bf13828 d2b7e94 bf13828 d2b7e94 bf13828 627d3d7 bf13828 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/v1.6.0/modules/ui_gradio_extensions.py
import os
from pathlib import Path
import gradio as gr
from modules import config
from .localization import localization_js
GradioTemplateResponseOriginal = gr.routes.templates.TemplateResponse
WEBUI_DIR_PATH = Path(os.path.dirname(os.path.realpath(__file__)))
def read_file(fp):
with open(WEBUI_DIR_PATH / fp, "r", encoding="utf-8") as f:
return f.read()
def javascript_html():
def s(text: str):
return f'<script type="text/javascript">{text}</script>\n'
def src(src: str):
return f"<script src='{src}'></script>\n"
def sf(fp: str):
return s(read_file(fp))
head = ""
head += src("https://jsd.onmicrosoft.cn/npm/marked@12.0.2")
head += s(localization_js(config.runtime_env_vars.language))
head += sf("js/index.js")
head += sf("js/localization.js")
if config.runtime_env_vars.theme:
head += s(f"set_theme('{config.runtime_env_vars.theme}');")
return head
def css_html():
head = f'<style>{read_file("css/style.css")}</style>'
return head
def reload_javascript():
js = javascript_html()
css = css_html()
def template_response(*args, **kwargs):
res = 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
|