f298ebca76
* feat: adds preview overlay for styles * fix: implement correct path resolution for all OS solves issues for Windows * fix: makes preview overlay also work with other languages * feat: scale down images to 258x200, fix aspect ratio in css reduces file size from 19,6MB to 3,4MB * fix: adjust z-index to correctly overlay gradio border pulse when generating images * update files * version --------- Co-authored-by: Manuel Schmid <9307310+mashb1t@users.noreply.github.com> Co-authored-by: Manuel Schmid <manuel.schmid@odt.net> Co-authored-by: lllyasviel <lyuminzhang@outlook.com>
68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
# based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/v1.6.0/modules/ui_gradio_extensions.py
|
|
|
|
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
|