rework-disable-autolaunch

2 files changed, 2 insertions, 4 deletions

diff --git a/modules/launch_utils.py b/modules/launch_utils.py
index af8d8b37..59008385 100644
--- a/modules/launch_utils.py
+++ b/modules/launch_utils.py
@@ -247,6 +247,7 @@ def prepare_environment():
     try:

         # the existance of this file is a signal to webui.sh/bat that webui needs to be restarted when it stops execution

         os.remove(os.path.join(script_path, "tmp", "restart"))

+        os.environ.setdefault('SD_WEBUI_DISABLE_AUTOLAUNCH', '1')

     except OSError:

         pass

 

diff --git a/webui.py b/webui.py
index 254fada3..1ee0e41d 100644
--- a/webui.py
+++ b/webui.py
@@ -396,7 +396,7 @@ def webui():
             ssl_verify=cmd_opts.disable_tls_verify,

             debug=cmd_opts.gradio_debug,

             auth=gradio_auth_creds,

-            inbrowser=cmd_opts.autolaunch,

+            inbrowser=cmd_opts.autolaunch and os.getenv('SD_WEBUI_DISABLE_AUTOLAUNCH') != '1',

             prevent_thread_lock=True,

             allowed_paths=cmd_opts.gradio_allowed_path,

             app_kwargs={

@@ -407,9 +407,6 @@ def webui():
         if cmd_opts.add_stop_route:

             app.add_route("/_stop", stop_route, methods=["POST"])

 

-        # after initial launch, disable --autolaunch for subsequent restarts

-        cmd_opts.autolaunch = False

-

         startup_timer.record("gradio launch")

 

         # gradio uses a very open CORS policy via app.user_middleware, which makes it possible for