Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - HF Space launcher (Option B, HF tree aware)
+# app.py — HF Space launcher (Option B) with runtime downloads & WebUI ready link
 import os
 import subprocess
 import threading
@@ -8,7 +8,7 @@ import requests
 import gradio as gr
 
 # -----------------------------
-# Configuration (edit if needed)
+# Configuration
 # -----------------------------
 DOWNLOADS = {
     "lora": {
@@ -20,143 +20,4 @@ DOWNLOADS = {
         "dest": "stable-diffusion-webui/extensions/ControlNet/models/control_v11f1e_sd15_tile.pth"
     },
     "temporalnet": {
-        "url": "https://huggingface.co/CiaraRowles/TemporalNet/resolve/main/diff_control_sd15_temporalnet_fp16.safetensors",
-        "dest": "stable-diffusion-webui/extensions/ControlNet/models/diff_control_sd15_temporalnet_fp16.safetensors"
-    },
-    "civitai_model": {
-        "url": "https://civitai.com/api/download/models/143906?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "dest": "stable-diffusion-webui/models/Stable-diffusion/civitai_model.safetensors"
-    }
-}
-
-WEBUI_ARGS = "--listen --xformers --enable-insecure-extension-access --medvram"
-LAUNCH_PY = "launch.py"  # relative to repo root (per HF tree)
-
-# -----------------------------
-# Utilities
-# -----------------------------
-def ensure_persistent_folders():
-    """Make sure HF tree persistent folders exist (safe no-op if present)."""
-    folders = [
-        "stable-diffusion-webui/deforum/input",
-        "stable-diffusion-webui/deforum/output_committed/frames",
-        "stable-diffusion-webui/deforum/output_committed/video",
-        "stable-diffusion-webui/models/Stable-diffusion",
-        "stable-diffusion-webui/models/Lora",
-        "stable-diffusion-webui/extensions/ControlNet/models",
-    ]
-    for f in folders:
-        os.makedirs(f, exist_ok=True)
-    print("✅ Persistent folders ensured.")
-
-
-def download_file_with_retries(url, dest, max_retries=3, backoff=5):
-    """Download a file with simple retry/backoff. Skips if dest exists."""
-    if os.path.exists(dest):
-        print(f"✅ Already exists: {dest}")
-        return True
-
-    os.makedirs(os.path.dirname(dest), exist_ok=True)
-    attempt = 1
-    while attempt <= max_retries:
-        try:
-            print(f"⬇️ Download attempt {attempt}: {url} -> {dest}")
-            with requests.get(url, stream=True, timeout=60) as r:
-                r.raise_for_status()
-                with open(dest + ".part", "wb") as f:
-                    for chunk in r.iter_content(chunk_size=8192):
-                        if chunk:
-                            f.write(chunk)
-            # atomically move into place
-            os.replace(dest + ".part", dest)
-            print(f"✅ Downloaded: {dest}")
-            return True
-        except Exception as e:
-            print(f"⚠️ Download failed (attempt {attempt}) for {url}: {e}")
-            attempt += 1
-            time.sleep(backoff * attempt)
-    print(f"❌ Failed to download after {max_retries} attempts: {url}")
-    # ensure partial file removed
-    try:
-        if os.path.exists(dest + ".part"):
-            os.remove(dest + ".part")
-    except Exception:
-        pass
-    return False
-
-
-def fetch_models():
-    """Download all configured models into the HF tree persistent paths."""
-    print("🔽 Starting runtime model downloads (if missing)...")
-    success_all = True
-    for name, info in DOWNLOADS.items():
-        url = info["url"]
-        dest = info["dest"]
-        ok = download_file_with_retries(url, dest)
-        success_all = success_all and ok
-    if success_all:
-        print("✅ All runtime downloads done (or already present).")
-    else:
-        print("⚠️ Some runtime downloads failed. Check logs for details.")
-
-
-# -----------------------------
-# Start WebUI routine (clean)
-# -----------------------------
-def start_webui():
-    """
-    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()
-        fetch_models()
-
-        # 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 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:
-        print("❌ Exception in start_webui():", e)
-
-
-# -----------------------------
-# Kick off WebUI in background thread
-# -----------------------------
-threading.Thread(target=start_webui, daemon=True).start()
-print("🧵 WebUI thread started (background).")
-
-# -----------------------------
-# Minimal Gradio UI to keep Space alive + quick controls
-# -----------------------------
-def show_status():
-    msg_lines = ["✅ HF Space running (Option B launcher)."]
-    # quick presence checks
-    if os.path.exists(LAUNCH_PY):
-        msg_lines.append(f"launch.py: found")
-    else:
-        msg_lines.append(f"launch.py: MISSING at root!")
-    # list basic model presence
-    for key, info in DOWNLOADS.items():
-        present = "yes" if os.path.exists(info["dest"]) else "no"
-        msg_lines.append(f"{key}: {present}")
-    return "\n".join(msg_lines)
-
-with gr.Blocks() as demo:
-    gr.Markdown("## Automatic1111 WebUI — HF Space launcher (Option B)")
-    gr.Markdown("This Space launches Automatic1111 in a background thread and keeps a small Gradio UI alive so Hugging Face marks the app as initialized.")
-    status_btn = gr.Button("Check status")
-    status_out = gr.Textbox(lines=10)
-    status_btn.click(fn=show_status, inputs=None, outputs=status_out)
-
-demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+        "url": "https://huggingface.co/CiaraRowles/TemporalNet/resolve/main/diff_control_sd15_temporal_