no store

routers/gemini.py

@@ -390,8 +390,12 @@ async def download_video(
 ):
     """
     Download a generated video.
-    If the video hasn't been downloaded yet, downloads it from the stored URL.
+    Downloads from Gemini URL, streams to client, then deletes local file.
+    No permanent storage on server.
     """
+    from fastapi.responses import StreamingResponse
+    import httpx
+    
     query = select(GeminiJob).where(
         GeminiJob.job_id == job_id,
         GeminiJob.user_id == user.user_id,
@@ -412,18 +416,6 @@ async def download_video(
             detail="Video not ready for download"
         )
     
-    # Check if video is already downloaded
-    filename = job.output_data.get("filename")
-    if filename:
-        filepath = os.path.join(DOWNLOADS_DIR, filename)
-        if os.path.exists(filepath):
-            return FileResponse(
-                path=filepath,
-                media_type="video/mp4",
-                filename=f"video_{job_id}.mp4"
-            )
-    
-    # Video not downloaded yet, download from stored URL
     video_url = job.output_data.get("video_url")
     if not video_url:
         raise HTTPException(
@@ -431,28 +423,22 @@ async def download_video(
             detail="No video URL available"
         )
     
-    # Download the video on-demand
-    from services.gemini_service import GeminiService
-    service = GeminiService()  # Uses default API key
-    
-    try:
-        filename = await service.download_video(video_url, job.job_id)
-        
-        # Update job with downloaded filename
-        job.output_data = {**job.output_data, "filename": filename}
-        await db.commit()
-        
-        filepath = os.path.join(DOWNLOADS_DIR, filename)
-        return FileResponse(
-            path=filepath,
-            media_type="video/mp4",
-            filename=f"video_{job_id}.mp4"
-        )
-    except Exception as e:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to download video: {str(e)}"
-        )
+    # Stream video directly from Gemini URL to client
+    # No local file storage needed
+    async def stream_video():
+        async with httpx.AsyncClient(timeout=120.0, follow_redirects=True) as client:
+            async with client.stream("GET", video_url) as response:
+                response.raise_for_status()
+                async for chunk in response.aiter_bytes(chunk_size=8192):
+                    yield chunk
+    
+    return StreamingResponse(
+        stream_video(),
+        media_type="video/mp4",
+        headers={
+            "Content-Disposition": f'attachment; filename="video_{job_id}.mp4"'
+        }
+    )
 
 
 @router.post("/job/{job_id}/cancel")