| |
|
|
| 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}">\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 |
|
|
