Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - HF Space launcher (Option B, HF tree aware, clean logs)
+# app.py - HF Space launcher (Option B, HF tree aware, clean logs + Gradio UI)
 import os
 import subprocess
 import threading
@@ -53,12 +53,10 @@ def ensure_folders():
 def download_file(url, dest, retries=3, backoff=5):
     """Download a file if missing, with retries."""
     if os.path.exists(dest):
-        print(f"✅ Already exists: {dest}")
         return True
     os.makedirs(os.path.dirname(dest), exist_ok=True)
     for attempt in range(1, retries + 1):
         try:
-            print(f"⬇️ Download attempt {attempt}: {url}")
             with requests.get(url, stream=True, timeout=60) as r:
                 r.raise_for_status()
                 with open(dest + ".part", "wb") as f:
@@ -66,25 +64,20 @@ def download_file(url, dest, retries=3, backoff=5):
                         if chunk:
                             f.write(chunk)
             os.replace(dest + ".part", dest)
-            print(f"✅ Downloaded: {dest}")
             return True
-        except Exception as e:
-            print(f"⚠️ Attempt {attempt} failed: {e}")
+        except Exception:
             time.sleep(backoff * attempt)
-    print(f"❌ Failed to download: {dest}")
     return False
 
 def fetch_models():
-    print("🔽 Fetching runtime models...")
+    """Download runtime models if missing."""
     for key, info in DOWNLOADS.items():
         download_file(info["url"], info["dest"])
-    print("✅ Model downloads done.")
 
 # -----------------------------
-# Check if port is open
+# Wait for WebUI port
 # -----------------------------
 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:
@@ -102,12 +95,10 @@ def wait_for_port(port, host="127.0.0.1", timeout=180):
 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 (logs suppressed)...")
 
-    # Redirect stdout/stderr to avoid spamming HF logs
+    # Redirect output to avoid HF spam
     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}")
@@ -126,6 +117,7 @@ def show_status():
     url = f"https://{space_domain}/?__theme=light"  # HF public URL
     lines = ["✅ HF Space running (Option B launcher)."]
     lines.append(f"WebUI URL (open in browser when ready): {url}")
+    # Model status
     for key, info in DOWNLOADS.items():
         present = "yes" if os.path.exists(info["dest"]) else "no"
         lines.append(f"{key}: {present}")
@@ -141,5 +133,5 @@ with gr.Blocks() as demo:
 demo.launch(
     server_name="0.0.0.0",
     server_port=int(os.environ.get("PORT", 7860)),
-    ssr_mode=False   # <--- this is all we’re adding
+    ssr_mode=False  # prevent Node server port issues
 )