Spaces:
No application file
No application file
from webui.ui.ui import create_ui | |
from .args import args | |
import gradio | |
def launch_webui(): | |
auth = (args.username, args.password) if args.username else None | |
template_response_original = gradio.routes.templates.TemplateResponse | |
# Magic monkeypatch | |
import webui.extensionlib.extensionmanager as em | |
scripts = ''.join([f'<script type="module" src="file={s}"></script>' for s in ['scripts/script.js'] + em.get_scripts()]) | |
def template_response(*args, **kwargs): | |
res = template_response_original(*args, **kwargs) | |
res.body = res.body.replace(b'</body>', | |
f'{scripts}</body>'.encode("utf8")) | |
res.init_headers() | |
return res | |
gradio.routes.templates.TemplateResponse = template_response | |
import webui.extensionlib.callbacks as cb | |
cb.get_manager('webui.init')() | |
create_ui(args.theme).queue().launch(share=args.share, | |
auth=auth, | |
server_name='0.0.0.0' if args.listen else None, | |
server_port=args.port, | |
favicon_path='assets/logo.png', | |
inbrowser=args.launch) | |