Update app.py

app.py

@@ -524,6 +524,7 @@ async def generate_text(request: Request):
 
     raise HTTPException(500, "Unknown provider routing error")
 
+@app.head("/gen/sfx")
 @app.get("/gen/sfx")
 @app.post("/gen/sfx")
 async def gensfx(request: Request, prompt: str = None):
@@ -537,4 +538,52 @@ async def gensfx(request: Request, prompt: str = None):
     return Response(
         response.content,
         media_type="audio/mpeg"
-    )
+    )
+
+@app.get("/gen/video")
+@app.post("/gen/video")
+async def genvideo(request: Request, prompt: str = None):
+    client_ip = request.client.host
+    check_audio_rate_limit(client_ip)
+
+    if prompt is None:
+        prompt = (await request.json()).prompt
+
+    url = f"https://gen.pollinations.ai/image/{prompt}?model=grok-video&key={PKEY}"
+
+    async def fetch_with_retry(url, max_retries=6):
+        async with httpx.AsyncClient(timeout=None) as client:
+            for attempt in range(max_retries):
+                response = await client.get(url)
+
+                if response.status_code == 200:
+                    return response
+
+                body_text = ""
+                try:
+                    body_text = response.text
+                except:
+                    pass
+
+                if response.status_code == 429 and "api.airforce" in body_text:
+                    wait = 0.5 * (2 ** attempt)
+                    print(f"[VIDEO RETRY] 429 from api.airforce, retrying in {wait:.2f}s...")
+                    await asyncio.sleep(wait)
+                    continue
+
+                raise HTTPException(
+                    status_code=500,
+                    detail=f"Pollinations error: {response.status_code}"
+                )
+
+            raise HTTPException(
+                status_code=503,
+                detail="Pollinations video generation overloaded (api.airforce). Try again later."
+            )
+
+    response = await fetch_with_retry(url)
+
+    return Response(
+        content=response.content,
+        media_type="video/mp4"
+    )