Update app.py

app.py

@@ -1,64 +1,72 @@
-from fastapi import FastAPI, HTTPException
+import os
+import uuid
+import time
+from pathlib import Path
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
-import time, os
-from google import genai
-from google.genai import types
+from huggingface_hub import InferenceClient
 
-app = FastAPI(title="Gemini Veo3 Video Generator")
+# --- config ---
+OUT_DIR = Path("outputs")
+OUT_DIR.mkdir(exist_ok=True)
 
-API_KEY = os.getenv("GEMINI_API_KEY")
-if not API_KEY:
-    raise ValueError("❌ GEMINI_API_KEY not found. Add it in Hugging Face → Settings → Repository secrets.")
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise RuntimeError("Missing HF_TOKEN environment variable. Add it in Hugging Face Space settings (Repository secrets).")
 
-client = genai.Client(api_key=API_KEY)
+# Initialize client (Fal.ai provider)
+client = InferenceClient(provider="fal-ai", api_key=HF_TOKEN)
 
-class VideoPrompt(BaseModel):
-    prompt: str
-
-@app.post("/generate_video")
-async def generate_video(data: VideoPrompt):
-    try:
-        print(f"🎬 Generating video for prompt: {data.prompt}")
+app = FastAPI(title="Fal.ai Text-to-Video API")
 
-        operation = client.models.generate_videos(
-            model="veo-3.0-generate-001",
-            prompt=data.prompt,
-            config=types.GenerateVideosConfig(
-                aspect_ratio="16:9",
-                resolution="1080p",
-                negative_prompt="cartoon, unrealistic, text distortion"
-            ),
-        )
+# Serve the outputs folder at /download/<filename>
+app.mount("/download", StaticFiles(directory=str(OUT_DIR)), name="download")
 
-        # Poll until generation completes
-        while not operation.done:
-            print("⏳ Still generating...")
-            time.sleep(10)
-            operation = client.operations.get(name=operation.name)  # ✅ correct call
+class PromptIn(BaseModel):
+    prompt: str
+    model: str = "akhaliq/veo3.1-fast"   # default; override if you want
 
-        if not operation.response.generated_videos:
-            raise HTTPException(status_code=500, detail="No video generated.")
+@app.post("/generate", response_class=JSONResponse)
+def generate_video(payload: PromptIn, request: Request):
+    """
+    Generate a video from `payload.prompt` using Fal.ai-backed model.
+    Returns: { success, file_url, filename }
+    """
+    prompt = payload.prompt.strip()
+    model = payload.model.strip()
+    if not prompt:
+        raise HTTPException(status_code=400, detail="prompt is required")
 
-        generated_video = operation.response.generated_videos[0]
-        video_file_id = generated_video.video.name  # ✅ file name or ID
+    # unique filename
+    filename = f"{uuid.uuid4().hex}.mp4"
+    out_path = OUT_DIR / filename
 
-        # ✅ Correct file retrieval syntax
-        file_info = client.files.get(name=video_file_id)
-        video_url = file_info.uri  # This is the downloadable URL
+    try:
+        # Call text_to_video (synchronous call returns bytes)
+        # Note: This may take some time depending on model & quota.
+        print("⏳ text_to_video request:", model, "prompt len:", len(prompt))
+        video_bytes = client.text_to_video(prompt=prompt, model=model)
+        if not video_bytes:
+            raise RuntimeError("Empty response from text_to_video")
 
-        print(f"✅ Video ready: {video_url}")
+        # Save file
+        with open(out_path, "wb") as f:
+            f.write(video_bytes)
 
-        return {
-            "success": True,
-            "video_url": video_url,
-            "message": "Video generation complete."
-        }
+        # Build absolute URL to download
+        base = str(request.base_url).rstrip("/")
+        file_url = f"{base}/download/{filename}"
 
+        print("✅ Video saved:", out_path)
+        return {"success": True, "filename": filename, "file_url": file_url}
     except Exception as e:
-        print("❌ Error:", e)
+        # log but return friendly message
+        print("❌ Error generating video:", repr(e))
+        # If provider returns specific errors (quota / 429), bubble them up
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @app.get("/")
 def root():
-    return {"message": "✅ Gemini Veo3 API running. Use /docs to test."}
+    return {"message": "Text-to-Video API running. POST /generate with JSON {\"prompt\":\"...\"}."}