Update app.py

app.py

@@ -1,12 +1,75 @@
 
 import os
 import subprocess
+import threading
 import gradio as gr
+import requests
 
-# Install dependencies (from root)
+# -------------------------------
+# 1️⃣ Install dependencies
+# -------------------------------
 subprocess.run(["pip", "install", "-r", "requirements.txt"])
 
-# Launch WebUI
-os.environ["COMMANDLINE_ARGS"] = "--listen --xformers --enable-insecure-extension-access --medvram"
-subprocess.run(["python", "launch.py"])
+# -------------------------------
+# 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)))
+