Update app.py

app.py

@@ -105,25 +105,25 @@ def fetch_models():
 # -----------------------------
 def start_webui():
     """
-    1) Ensure folders exist
-    2) Fetch runtime models (only if missing)
-    3) Start Automatic1111's launch.py in the same foreground process context
-       using a subprocess call built from a proper arg list.
+    Option B — background WebUI launcher aware of HF tree and persistent folders.
+    1) Ensure persistent folders exist.
+    2) Download missing runtime models.
+    3) Launch Automatic1111 WebUI (launch.py) with correct PORT binding.
+    4) Threaded so Gradio UI stays alive.
     """
     try:
         ensure_persistent_folders()
-
-        # Download models (non-blocking safety: we do this before launch so they are present)
         fetch_models()
 
-        # Build the command safely
-        args = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS)
-        print("▶️ Launching WebUI with:", " ".join(args))
+        # Respect HF Space assigned PORT
+        port = int(os.environ.get("PORT", 7860))
+        # Build safe command list
+        cmd = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS) + [f"--port {port}"]
+        print("▶️ Launching WebUI with:", " ".join(cmd))
 
-        # Run launch.py and block this thread until it exits; logs go to HF logs.
-        # We run in a separate Python thread (caller keeps main process alive via Gradio).
-        subprocess.run(args, check=True)
-        print("ℹ️ launch.py exited (subprocess finished). If this happens early, check errors above.")
+        # Run launch.py in the current environment, blocking this thread
+        subprocess.run(cmd, check=True)
+        print("ℹ️ launch.py exited normally.")
     except subprocess.CalledProcessError as cpe:
         print(f"❌ launch.py exited with non-zero status: {cpe.returncode}")
     except Exception as e: