Update app.py

app.py

@@ -4,30 +4,24 @@ import time, os
 from google import genai
 from google.genai import types
 
-# 🚀 Initialize FastAPI
 app = FastAPI(title="Gemini Veo3 Video Generator")
 
-# 🔒 Load Gemini API key from Hugging Face secret
 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.")
 
-# 🔑 Initialize client
 client = genai.Client(api_key=API_KEY)
 
-# 🧠 Request model
 class VideoPrompt(BaseModel):
     prompt: str
 
-# 🎬 Endpoint to generate video
 @app.post("/generate_video")
 async def generate_video(data: VideoPrompt):
     try:
         print(f"🎬 Generating video for prompt: {data.prompt}")
 
         operation = client.models.generate_videos(
-            model="veo-3.0-fast-generate-001",
+            model="veo-3.0-generate-001",
             prompt=data.prompt,
             config=types.GenerateVideosConfig(
                 aspect_ratio="16:9",
@@ -36,20 +30,21 @@ async def generate_video(data: VideoPrompt):
             ),
         )
 
-        # ⏳ Wait for video generation to complete
+        # Poll until generation completes
         while not operation.done:
             print("⏳ Still generating...")
             time.sleep(10)
-            operation = client.operations.get(operation)
+            operation = client.operations.get(name=operation.name)  # ✅ correct call
 
-        # ✅ Check if video is generated
         if not operation.response.generated_videos:
             raise HTTPException(status_code=500, detail="No video generated.")
 
-        # 🎥 Get video file & URL
-        video_file = operation.response.generated_videos[0].video
-        file_info = client.files.get(video_file)
-        video_url = file_info.uri  # Public GCS link
+        generated_video = operation.response.generated_videos[0]
+        video_file_id = generated_video.video.name  # ✅ file name or ID
+
+        # ✅ Correct file retrieval syntax
+        file_info = client.files.get(name=video_file_id)
+        video_url = file_info.uri  # This is the downloadable URL
 
         print(f"✅ Video ready: {video_url}")
 
@@ -60,11 +55,10 @@ async def generate_video(data: VideoPrompt):
         }
 
     except Exception as e:
-        print(f"❌ Error: {e}")
+        print("❌ Error:", e)
         raise HTTPException(status_code=500, detail=str(e))
 
 
-# 🏠 Optional root endpoint for basic health check
 @app.get("/")
 def root():
-    return {"message": "✅ Gemini Veo3 API is running. Go to /docs to test."}
+    return {"message": "✅ Gemini Veo3 API running. Use /docs to test."}