Update app.py

app.py

@@ -835,22 +835,36 @@ async def fetch_airforce_video(prompt: str, duration: int = 5) -> bytes:
 
 @app.get("/gen/video/airforce/{prompt}")
 @app.post("/gen/video/airforce")
-async def genvideo_airforce(request: Request, prompt: str = None, duration: int = 5):
+async def genvideo_airforce(request: Request, prompt: str = None):
+    data = await request.json()
     if prompt is None:
         body = await request.json()
         prompt = body.get("prompt")
         duration = body.get("duration", 5)
 
-    if not prompt:
-        raise HTTPException(400, "Prompt is required")
+    body = {
+        "model": "grok-imagine-video",
+        "prompt": prompt,
+    }
+
+    async with httpx.AsyncClient(timeout=600) as client:
+        resp = await client.post(
+            "https://api.airforce/v1/images/generations",
+            headers={"Authorization": f"Bearer {AIRFORCE_API_KEY}", "Content-Type": "application/json"},
+            json=body
+        )
+
+    if resp.status_code != 200:
+        return JSONResponse(status_code=resp.status_code, content=resp.json())
 
-    video_content = await fetch_airforce_video(prompt, duration)
+    result = resp.json()
+    video_url = result.get("data", [{}])[0].get("url")
+    if not video_url:
+        raise HTTPException(500, "No video URL returned from Airforce")
 
+    video_resp = await client.get(video_url)
     return Response(
-        content=video_content,
+        content=video_resp.content,
         media_type="video/mp4",
-        headers={
-            "Content-Length": str(len(video_content)),
-            "Accept-Ranges": "bytes",
-        }
+        headers={"Content-Length": str(len(video_resp.content))}
     )