|
|
|
|
|
import os |
|
import gradio as gr |
|
import args_manager |
|
|
|
from modules.localization import localization_js |
|
|
|
|
|
GradioTemplateResponseOriginal = gr.routes.templates.TemplateResponse |
|
|
|
modules_path = os.path.dirname(os.path.realpath(__file__)) |
|
script_path = os.path.dirname(modules_path) |
|
|
|
|
|
def webpath(fn): |
|
if fn.startswith(script_path): |
|
web_path = os.path.relpath(fn, script_path).replace('\\', '/') |
|
else: |
|
web_path = os.path.abspath(fn) |
|
|
|
return f'file={web_path}?{os.path.getmtime(fn)}' |
|
|
|
|
|
def javascript_html(): |
|
script_js_path = webpath('javascript/script.js') |
|
context_menus_js_path = webpath('javascript/contextMenus.js') |
|
localization_js_path = webpath('javascript/localization.js') |
|
zoom_js_path = webpath('javascript/zoom.js') |
|
edit_attention_js_path = webpath('javascript/edit-attention.js') |
|
viewer_js_path = webpath('javascript/viewer.js') |
|
image_viewer_js_path = webpath('javascript/imageviewer.js') |
|
samples_path = webpath(os.path.abspath('./sdxl_styles/samples/fooocus_v2.jpg')) |
|
head = f'<script type="text/javascript">{localization_js(args_manager.args.language)}</script>\n' |
|
head += f'<script type="text/javascript" src="{script_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{context_menus_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{localization_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{zoom_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{edit_attention_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{viewer_js_path}"></script>\n' |
|
head += f'<script type="text/javascript" src="{image_viewer_js_path}"></script>\n' |
|
head += f'<meta name="samples-path" content="{samples_path}"></meta>\n' |
|
|
|
if args_manager.args.theme: |
|
head += f'<script type="text/javascript">set_theme(\"{args_manager.args.theme}\");</script>\n' |
|
|
|
return head |
|
|
|
|
|
def css_html(): |
|
style_css_path = webpath('css/style.css') |
|
head = f'<link rel="stylesheet" property="stylesheet" href="{style_css_path}">' |
|
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 |
|
|
