Create routers/videos.py

routers/videos.py

@@ -0,0 +1,52 @@
+# routers/videos.py
+from fastapi import APIRouter, Depends, HTTPException, status
+from starlette.concurrency import run_in_threadpool
+import models
+import config
+from services import generation, security
+import logging
+
+logger = logging.getLogger(__name__)
+router = APIRouter(
+    prefix="/videos",
+    tags=["Video Generation"],
+    dependencies=[Depends(security.get_api_key)]
+)
+
+@router.post("/image-to-video", response_model=models.VideoResponse, status_code=status.HTTP_201_CREATED)
+async def create_video_from_image(request: models.VideoRequest):
+    """
+    Generates a short video clip animating a source image.
+    """
+    if not generation.model_cache.get("video_pipeline"):
+         raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="Video generation model is not available.")
+
+    try:
+        logger.info(f"Received image-to-video request.")
+        # Video generation can be VERY slow, especially on CPU/basic GPU
+        video_base64, actual_format = await run_in_threadpool(
+            generation.generate_video_sync,
+            image_base64=request.image_base64,
+            prompt=request.prompt,
+            motion_bucket_id=request.motion_bucket_id,
+            noise_aug_strength=request.noise_aug_strength,
+            num_frames=request.num_frames,
+            fps=request.fps,
+            num_inference_steps=request.num_inference_steps,
+            guidance_scale=request.guidance_scale
+        )
+        logger.info(f"Successfully generated video.")
+        return models.VideoResponse(
+            video_base64=video_base64,
+            model_name=config.VIDEO_MODEL_NAME,
+            format=actual_format
+        )
+    except ValueError as e: # Catch specific errors like invalid base64 or frame issues
+        logger.error(f"Value error during video generation: {e}", exc_info=True)
+        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
+    except RuntimeError as e:
+         logger.error(f"Model runtime error during video generation: {e}", exc_info=True)
+         raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail=f"Model inference failed: {e}")
+    except Exception as e:
+        logger.error(f"Unexpected error during video generation: {e}", exc_info=True)
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"An unexpected error occurred: {e}")