Update app.py

app.py

@@ -1,75 +1,47 @@
-
 import os
 import subprocess
-import threading
 import gradio as gr
 import requests
 
-# -------------------------------
-# 1️⃣ Install dependencies
-# -------------------------------
+# -----------------------------
+# Runtime downloads
+# -----------------------------
+downloads = {
+    "lora": "https://huggingface.co/latent-consistency/lcm-lora-sdv1-5/resolve/main/pytorch_lora_weights.safetensors",
+    "controlnet_tile": "https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile.pth",
+    "temporalnet": "https://huggingface.co/CiaraRowles/TemporalNet/resolve/main/diff_control_sd15_temporalnet_fp16.safetensors",
+    "civitai_model": "https://civitai.com/api/download/models/143906?type=Model&format=SafeTensor&size=pruned&fp=fp16"
+}
+
+# Ensure persistent folders exist
+os.makedirs("stable-diffusion-webui/models/Stable-diffusion", exist_ok=True)
+os.makedirs("stable-diffusion-webui/models/Lora", exist_ok=True)
+os.makedirs("stable-diffusion-webui/extensions/ControlNet/models", exist_ok=True)
+
+# Download helper
+def download_file(url, dest):
+    if not os.path.exists(dest):
+        print(f"Downloading {url} -> {dest}")
+        response = requests.get(url, stream=True)
+        with open(dest, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+    else:
+        print(f"{dest} already exists, skipping download.")
+
+# Download models
+download_file(downloads["lora"], "stable-diffusion-webui/models/Lora/pytorch_lora_weights.safetensors")
+download_file(downloads["controlnet_tile"], "stable-diffusion-webui/extensions/ControlNet/models/control_v11f1e_sd15_tile.pth")
+download_file(downloads["temporalnet"], "stable-diffusion-webui/extensions/ControlNet/models/diff_control_sd15_temporalnet_fp16.safetensors")
+download_file(downloads["civitai_model"], "stable-diffusion-webui/models/Stable-diffusion/civitai_model.safetensors")
+
+# -----------------------------
+# Install dependencies
+# -----------------------------
 subprocess.run(["pip", "install", "-r", "requirements.txt"])
 
-# -------------------------------
-# 2️⃣ Runtime download helper
-# -------------------------------
-def download_file(url, dest_path):
-    if os.path.exists(dest_path):
-        print(f"✅ Already exists: {dest_path}")
-        return
-    os.makedirs(os.path.dirname(dest_path), exist_ok=True)
-    print(f"⬇️ Downloading {url} -> {dest_path}")
-    response = requests.get(url, stream=True)
-    response.raise_for_status()
-    with open(dest_path, "wb") as f:
-        for chunk in response.iter_content(chunk_size=8192):
-            f.write(chunk)
-    print(f"✅ Download completed: {dest_path}")
-
-def fetch_models():
-    # Stable Diffusion SD1.5 (Civitai pruned fp16)
-    download_file(
-        "https://civitai.com/api/download/models/143906?type=Model&format=SafeTensor&size=pruned&fp=fp16",
-        "stable-diffusion-webui/models/Stable-diffusion/sd15_pruned_fp16.safetensors"
-    )
-
-    # LoRA
-    download_file(
-        "https://huggingface.co/latent-consistency/lcm-lora-sdv1-5/resolve/main/pytorch_lora_weights.safetensors",
-        "stable-diffusion-webui/models/Lora/lcm-lora.safetensors"
-    )
-
-    # ControlNets
-    download_file(
-        "https://huggingface.co/CiaraRowles/TemporalNet/blob/main/diff_control_sd15_temporalnet_fp16.safetensors?raw=true",
-        "stable-diffusion-webui/extensions/ControlNet/models/diff_control_sd15_temporalnet_fp16.safetensors"
-    )
-    download_file(
-        "https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile.pth",
-        "stable-diffusion-webui/extensions/ControlNet/models/control_v11f1e_sd15_tile.pth"
-    )
-
-# -------------------------------
-# 3️⃣ Function to start WebUI
-# -------------------------------
-def start_webui():
-    fetch_models()  # download at runtime
-    os.environ["COMMANDLINE_ARGS"] = "--listen --xformers --enable-insecure-extension-access --medvram"
-    subprocess.run(["python", "launch.py"])
-
-# Launch WebUI in background thread
-threading.Thread(target=start_webui, daemon=True).start()
-
-# -------------------------------
-# 4️⃣ Minimal Gradio interface
-# -------------------------------
-def status():
-    return "✅ WebUI is running in the background. Models are downloaded at runtime."
-
-with gr.Blocks() as demo:
-    gr.Markdown("### Automatic1111 WebUI Launcher with Runtime Downloads")
-    gr.Button("Check status").click(status, outputs=[gr.Textbox()])
-
-demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
-
-
+# -----------------------------
+# Launch WebUI
+# -----------------------------
+os.environ["COMMANDLINE_ARGS"] = "--listen --xformers --enable-insecure-extension-access --medvram"
+subprocess.run(["python", "launch.py"])