Update app.py

app.py

@@ -81,14 +81,15 @@ def fetch_models():
     print("✅ Model downloads done.")
 
 # -----------------------------
-# Check if port is free
+# Check if port is open
 # -----------------------------
-def wait_for_port(port, host="127.0.0.1", timeout=120):
+def wait_for_port(port, host="127.0.0.1", timeout=180):
     """Wait until the WebUI port is open."""
     start = time.time()
     while time.time() - start < timeout:
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
             try:
+                s.settimeout(1)
                 s.connect((host, port))
                 return True
             except:
@@ -96,24 +97,22 @@ def wait_for_port(port, host="127.0.0.1", timeout=120):
     return False
 
 # -----------------------------
-# Start WebUI thread
+# Start WebUI in background
 # -----------------------------
 def start_webui():
     ensure_folders()
     fetch_models()
 
     port = int(os.environ.get("PORT", 7860))
-    cmd = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS) + [f"--port {port}"]
-    print("▶️ Launching WebUI with:", " ".join(cmd))
+    cmd = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS) + [f"--port={port}"]
+    print("▶️ Launching WebUI (logs suppressed)...")
 
-    # Run launch.py once; logs go to HF logs
-    try:
-        subprocess.run(cmd, check=True)
-        print("ℹ️ launch.py exited normally.")
-    except subprocess.CalledProcessError as e:
-        print(f"❌ launch.py failed: {e}")
-    except Exception as e:
-        print(f"❌ Exception: {e}")
+    # Redirect stdout/stderr to avoid spamming HF logs
+    with subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) as proc:
+        if wait_for_port(port, timeout=180):
+            print(f"✅ WebUI is ready on port {port}")
+        else:
+            print("⚠️ WebUI did not start within timeout.")
 
 threading.Thread(target=start_webui, daemon=True).start()
 print("🧵 WebUI background thread started.")
@@ -123,10 +122,10 @@ print("🧵 WebUI background thread started.")
 # -----------------------------
 def show_status():
     port = int(os.environ.get("PORT", 7860))
-    url = f"http://localhost:{port}"
+    space_domain = os.environ.get("SPACE_DOMAIN", "your-space-name.hf.space")
+    url = f"https://{space_domain}/?__theme=light"  # HF public URL
     lines = ["✅ HF Space running (Option B launcher)."]
-    lines.append(f"WebUI URL: [Click here]({url}) (opens in browser when ready)")
-    # Check models
+    lines.append(f"WebUI URL (open in browser when ready): {url}")
     for key, info in DOWNLOADS.items():
         present = "yes" if os.path.exists(info["dest"]) else "no"
         lines.append(f"{key}: {present}")