Update app.py

app.py

@@ -20,4 +20,94 @@ 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_temporal_
+        "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
+
+# -----------------------------
+# Utilities
+# -----------------------------
+def ensure_persistent_folders():
+    """Ensure HF tree persistent folders exist."""
+    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 retries, skip if 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)
+            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}")
+    if os.path.exists(dest + ".part"):
+        os.remove(dest + ".part")
+    return False
+
+def fetch_models():
+    """Download all configured models into HF tree persistent paths."""
+    print("🔽 Starting runtime model downloads (if missing)...")
+    success_all = True
+    for name, info in DOWNLOADS.items():
+        ok = download_file_with_retries(info["url"], info["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.")
+
+# -----------------------------
+# Start WebUI routine (Option B)
+# -----------------------------
+WEBUI_LINK = None
+
+def start_webui():
+    """Launch WebUI in background and detect ready URL."""
+    global WEBUI_LINK
+    ensure_persistent_folders()
+    fetch_models()
+
+    args = ["python", LAUNCH_PY] + shlex.split(WEBUI_ARGS)
+    print("▶️ Launching WebUI with:", " ".join(args))
+
+    proc = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
+
+    # Monitor output for "Running on" URL
+    for line in iter(proc.stdout.readline, ""):
+        print(line, end="")
+        if "Running on local URL:" in line or "Running on" in line:
+            parts = line.split("http")