Update app.py

app.py

@@ -98,7 +98,7 @@ def start_webui():
     port = int(os.environ.get("PORT", 7860))
     cmd = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS) + [f"--port={port}"]
 
-    # Redirect output to avoid HF spam
+    # Redirect output to avoid HF log 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}")
@@ -112,11 +112,17 @@ print("🧵 WebUI background thread started.")
 # Gradio UI
 # -----------------------------
 def show_status():
-    port = int(os.environ.get("PORT", 7860))
-    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 (open in browser when ready): {url}")
+
+    # Show public shareable URL for the Space WebUI
+    try:
+        # share=True generates a public URL automatically
+        port = int(os.environ.get("PORT", 7860))
+        url = demo.share_url  # updated after launch
+        lines.append(f"WebUI URL (open in browser when ready): {url}")
+    except Exception:
+        lines.append("WebUI URL not yet available. Refresh after a few seconds.")
+
     # Model status
     for key, info in DOWNLOADS.items():
         present = "yes" if os.path.exists(info["dest"]) else "no"
@@ -130,8 +136,10 @@ with gr.Blocks() as demo:
     status_out = gr.Textbox(lines=10)
     status_btn.click(fn=show_status, inputs=None, outputs=status_out)
 
+# Launch Gradio with share=True to get a public clickable URL
 demo.launch(
     server_name="0.0.0.0",
     server_port=int(os.environ.get("PORT", 7860)),
-    ssr_mode=False  # prevent Node server port issues
+    ssr_mode=False,
+    share=True  # <--- makes the WebUI URL publicly accessible
 )