webui/webui.py

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)