Enhance video generation API: add polling URL, video URLs, and error handling; implement polling status endpoint with tests

Co-authored-by: Copilot <copilot@github.com>

backend/app/models/ai.py

@@ -87,6 +87,9 @@ class VideoFromImageRequest(BaseModel):
 class VideoResponse(BaseModel):
     id: str
     model: str
-    status: str  # "queued" | "processing" | "completed"
-    video_url: str | None = None
+    status: str  # "queued" | "processing" | "completed" | "failed"
+    polling_url: str | None = None
+    video_urls: list[str] | None = None
+    video_url: str | None = None  # first entry of video_urls for convenience
+    error: str | None = None
     metadata: dict[str, Any] | None = None

backend/app/routers/generate.py

@@ -105,11 +105,15 @@ async def generate_video(
         raise HTTPException(
             status_code=status.HTTP_502_BAD_GATEWAY, detail=f"OpenRouter error: {exc}")
 
+    urls = result.get("unsigned_urls") or result.get("video_urls")
     return VideoResponse(
         id=result.get("id", ""),
-        model=result.get("model", body.model),
+        model=body.model,
         status=result.get("status", "queued"),
-        video_url=result.get("video_url"),
+        polling_url=result.get("polling_url"),
+        video_urls=urls,
+        video_url=(urls or [None])[0],
+        error=result.get("error"),
         metadata=result.get("metadata"),
     )
 
@@ -132,10 +136,39 @@ async def generate_video_from_image(
         raise HTTPException(
             status_code=status.HTTP_502_BAD_GATEWAY, detail=f"OpenRouter error: {exc}")
 
+    urls = result.get("unsigned_urls") or result.get("video_urls")
     return VideoResponse(
         id=result.get("id", ""),
-        model=result.get("model", body.model),
+        model=body.model,
         status=result.get("status", "queued"),
-        video_url=result.get("video_url"),
+        polling_url=result.get("polling_url"),
+        video_urls=urls,
+        video_url=(urls or [None])[0],
+        error=result.get("error"),
+        metadata=result.get("metadata"),
+    )
+
+
+@router.get("/video/status", response_model=VideoResponse)
+async def poll_video_status(
+    polling_url: str,
+    _: dict = Depends(get_current_user),
+) -> VideoResponse:
+    """Poll the status of a video generation job via its polling_url."""
+    try:
+        result = await openrouter.poll_video_status(polling_url)
+    except Exception as exc:
+        raise HTTPException(
+            status_code=status.HTTP_502_BAD_GATEWAY, detail=f"OpenRouter error: {exc}")
+
+    urls = result.get("unsigned_urls") or result.get("video_urls")
+    return VideoResponse(
+        id=result.get("id", ""),
+        model=result.get("model", ""),
+        status=result.get("status", "processing"),
+        polling_url=result.get("polling_url"),
+        video_urls=urls,
+        video_url=(urls or [None])[0],
+        error=result.get("error"),
         metadata=result.get("metadata"),
     )

backend/app/services/openrouter.py

@@ -50,6 +50,9 @@ async def chat_completion(
         "max_tokens": max_tokens,
     }
     async with httpx.AsyncClient(timeout=60) as client:
+        resp = client.build_request(
+            "POST", f"{base_url}/chat/completions", headers=_headers(), json=payload
+        )
         response = await client.send(resp)
         response.raise_for_status()
         return response.json()
@@ -90,7 +93,7 @@ async def generate_video(
         payload["duration_seconds"] = duration_seconds
     async with httpx.AsyncClient(timeout=120) as client:
         resp = client.build_request(
-            "POST", f"{base_url}/video/generations", headers=_headers(), json=payload
+            "POST", f"{base_url}/videos", headers=_headers(), json=payload
         )
         response = await client.send(resp)
         response.raise_for_status()
@@ -116,8 +119,17 @@ async def generate_video_from_image(
         payload["duration_seconds"] = duration_seconds
     async with httpx.AsyncClient(timeout=120) as client:
         resp = client.build_request(
-            "POST", f"{base_url}/video/generations/from-image", headers=_headers(), json=payload
+            "POST", f"{base_url}/videos", headers=_headers(), json=payload
         )
         response = await client.send(resp)
         response.raise_for_status()
         return response.json()
+
+
+async def poll_video_status(polling_url: str) -> dict[str, Any]:
+    """Check the status of a video generation job via its polling_url."""
+    async with httpx.AsyncClient(timeout=15) as client:
+        resp = client.build_request("GET", polling_url, headers=_headers())
+        response = await client.send(resp)
+        response.raise_for_status()
+        return response.json()