Upload 2 files

.gitattributes

@@ -41,3 +41,4 @@ direct/install_comfyui.tar filter=lfs diff=lfs merge=lfs -text
 direct/install_comfyui_face.tar filter=lfs diff=lfs merge=lfs -text
 direct/install_comfyui_pack.tar filter=lfs diff=lfs merge=lfs -text
 direct/install_comfyui_pack_20240225.tar filter=lfs diff=lfs merge=lfs -text
+direct/forge/repositories.tar filter=lfs diff=lfs merge=lfs -text

direct/forge/directui.py

@@ -0,0 +1,145 @@
+from __future__ import annotations
+
+import os
+import time
+
+from modules import timer
+from modules import initialize_util
+from modules import initialize
+
+startup_timer = timer.startup_timer
+startup_timer.record("launcher")
+
+initialize.imports()
+
+initialize.check_versions()
+
+
+def create_api(app):
+    from modules.api.api import Api
+    from modules.call_queue import queue_lock
+
+    api = Api(app, queue_lock)
+    return api
+
+
+def api_only():
+    from fastapi import FastAPI
+    from modules.shared_cmd_options import cmd_opts
+
+    initialize.initialize()
+
+    app = FastAPI()
+    initialize_util.setup_middleware(app)
+    api = create_api(app)
+
+    from modules import script_callbacks
+
+    script_callbacks.before_ui_callback()
+    script_callbacks.app_started_callback(None, app)
+
+    print(f"Startup time: {startup_timer.summary()}.")
+    api.launch(
+        server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1",
+        port=cmd_opts.port if cmd_opts.port else 7861,
+        root_path=f"/{cmd_opts.subpath}" if cmd_opts.subpath else "",
+    )
+
+
+def webui():
+    from modules.shared_cmd_options import cmd_opts
+
+    launch_api = cmd_opts.api
+    initialize.initialize()
+
+    from modules import (
+        shared,
+        ui_tempdir,
+        script_callbacks,
+        ui,
+        progress,
+        ui_extra_networks,
+    )
+
+    while 1:
+        if shared.opts.clean_temp_dir_at_start:
+            ui_tempdir.cleanup_tmpdr()
+            startup_timer.record("cleanup temp dir")
+
+        script_callbacks.before_ui_callback()
+        startup_timer.record("scripts before_ui_callback")
+
+        shared.demo = ui.create_ui()
+        startup_timer.record("create ui")
+
+        app, local_url, share_url = shared.demo.launch(
+            height=3000, prevent_thread_lock=True
+        )
+
+        startup_timer.record("gradio launch")
+
+        # gradio uses a very open CORS policy via app.user_middleware, which makes it possible for
+        # an attacker to trick the user into opening a malicious HTML page, which makes a request to the
+        # running web ui and do whatever the attacker wants, including installing an extension and
+        # running its code. We disable this here. Suggested by RyotaK.
+        app.user_middleware = [
+            x for x in app.user_middleware if x.cls.__name__ != "CORSMiddleware"
+        ]
+
+        initialize_util.setup_middleware(app)
+
+        progress.setup_progress_api(app)
+        ui.setup_ui_api(app)
+
+        if launch_api:
+            create_api(app)
+
+        ui_extra_networks.add_pages_to_demo(app)
+
+        startup_timer.record("add APIs")
+
+        with startup_timer.subcategory("app_started_callback"):
+            script_callbacks.app_started_callback(shared.demo, app)
+
+        timer.startup_record = startup_timer.dump()
+        print(f"Startup time: {startup_timer.summary()}.")
+
+        try:
+            while True:
+                server_command = shared.state.wait_for_server_command(timeout=5)
+                if server_command:
+                    if server_command in ("stop", "restart"):
+                        break
+                    else:
+                        print(f"Unknown server command: {server_command}")
+        except KeyboardInterrupt:
+            print("Caught KeyboardInterrupt, stopping...")
+            server_command = "stop"
+
+        if server_command == "stop":
+            print("Stopping server...")
+            # If we catch a keyboard interrupt, we want to stop the server and exit.
+            shared.demo.close()
+            break
+
+        # disable auto launch webui in browser for subsequent UI Reload
+        os.environ.setdefault("SD_WEBUI_RESTARTING", "1")
+
+        print("Restarting UI...")
+        shared.demo.close()
+        time.sleep(0.5)
+        startup_timer.reset()
+        script_callbacks.app_reload_callback()
+        startup_timer.record("app reload callback")
+        script_callbacks.script_unloaded_callback()
+        startup_timer.record("scripts unloaded callback")
+        initialize.initialize_rest(reload_script_modules=True)
+
+
+if __name__ == "__main__":
+    from modules.shared_cmd_options import cmd_opts
+
+    if cmd_opts.nowebui:
+        api_only()
+    else:
+        webui()

direct/forge/repositories.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f5b897a511af311cda44a5be3da40cf33196366734fc9d6ea763ab8c7a638d1
+size 222383104