Update app.py

app.py

@@ -7,12 +7,13 @@ import uvicorn
 import os
 import uuid
 import shutil
+import requests
 
 app = FastAPI(title="WAN2 GGUF API", version="1.0")
 
 # ✅ Directories
 MODEL_REPO = "calcuis/wan2-gguf"
-MODEL_FILE = "wan2.2-animate-14b-q4_0.gguf"  # big model
+MODEL_FILE = "wan2.2-animate-14b-q4_0.gguf"
 MODEL_DIR = "models"
 OUTPUT_DIR = "outputs"
 
@@ -42,37 +43,40 @@ def root():
 @app.post("/generate")
 def generate_video(request: PromptRequest):
     """
-    Dummy video generator — for now just copies a placeholder .mp4.
-    Replace this later with actual WAN2 inference code.
+    Dummy video generator — just copies a placeholder .mp4.
+    Replace with actual WAN2 inference later.
     """
 
     # Unique filename
     file_id = str(uuid.uuid4())
     file_path = os.path.join(OUTPUT_DIR, f"{file_id}.mp4")
 
-    # Use a Hugging Face placeholder video
+    # Hugging Face placeholder video
     placeholder_url = (
         "https://huggingface.co/datasets/huggingface/documentation-images/"
         "resolve/main/video-placeholder.mp4"
     )
 
-    # Download placeholder only once
+    # Download placeholder if not already present
     placeholder_file = os.path.join(OUTPUT_DIR, "placeholder.mp4")
     if not os.path.exists(placeholder_file):
-        import requests
         r = requests.get(placeholder_url, stream=True)
         with open(placeholder_file, "wb") as f:
             shutil.copyfileobj(r.raw, f)
 
-    # Copy placeholder to simulate unique output
+    # Copy placeholder to simulate output
     shutil.copy(placeholder_file, file_path)
 
+    # ✅ Build full URL for Hugging Face Space
+    base_url = "https://abrahamdw882-wan2-api.hf.space"
+    video_url = f"{base_url}/file/{file_id}.mp4"
+
     return {
         "status": "success",
         "model_file": MODEL_FILE,
         "prompt": request.prompt,
         "steps": request.steps,
-        "video_url": f"/file/{file_id}.mp4",
+        "video_url": video_url,
         "note": "Replace this with actual inference code."
     }