Unify Story Reels API with Video Creator pattern: /story-reel endpoints

modules/story_reels/router.py

@@ -1,5 +1,6 @@
 """
 Story Reels Router - API Endpoints
+Consistent with Video Creator API pattern
 """
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import FileResponse
@@ -9,8 +10,8 @@ from .schemas import (
     GenerateVideoRequest,
     GenerateVideoResponse,
     VideoStatusResponse,
-    PreviewResponse,
-    JobStatus
+    JobStatus,
+    StyleEnum
 )
 from .services.story_creator import StoryCreator
 
@@ -29,82 +30,64 @@ def set_story_creator(creator: StoryCreator):
 router = APIRouter()
 
 
-@router.post("/generate",
+@router.post("/story-reel",
     response_model=GenerateVideoResponse,
     status_code=201,
-    summary="Generate story video",
-    description="Generate a character-consistent story video from script"
+    summary="Create a new story reel",
+    description="Create a new AI-generated story video. Returns video_id to track progress."
 )
-async def generate_video(request: GenerateVideoRequest):
+async def create_story_reel(request: GenerateVideoRequest):
     """
-    Main video generation endpoint.
+    Create a new story reel from topic.
     
+    - AI generates script from topic
     - Converts script to speech (TTS)
     - Generates captions (Whisper)
-    - Creates character-consistent images (Cloudflare)
+    - Creates AI images (NVIDIA/Cloudflare)
     - Composes final video (MoviePy)
     """
     try:
-        logger.info(f"Generating video for topic: {request.topic}")
+        logger.info(f"Creating story reel for topic: {request.topic}")
         
-        job_id = story_creator.add_to_queue(
+        video_id = story_creator.add_to_queue(
             topic=request.topic,
             image_style=request.image_style.value,
             voice=request.voice
         )
         
         return GenerateVideoResponse(
-            job_id=job_id,
+            job_id=video_id,
             status=JobStatus.queued,
             message="Video generation started"
         )
         
     except Exception as e:
-        logger.error(f"Error starting generation: {e}", exc_info=True)
+        logger.error(f"Error creating story reel: {e}", exc_info=True)
         raise HTTPException(status_code=400, detail=str(e))
 
 
-@router.get("/status/{job_id}",
+@router.get("/story-reel/{video_id}/status",
     response_model=VideoStatusResponse,
-    summary="Get job status",
-    description="Check the processing status of a video generation job"
+    summary="Get video status",
+    description="Check the processing status of a story reel (queued, processing, ready, or failed)"
 )
-async def get_status(job_id: str):
-    """Get video generation status"""
-    status = story_creator.get_status(job_id)
+async def get_story_status(video_id: str):
+    """Get the status of a story reel"""
+    status = story_creator.get_status(video_id)
     return VideoStatusResponse(**status)
 
 
-@router.get("/preview/{job_id}/{scene_id}",
-    response_model=PreviewResponse,
-    summary="Get scene preview",
-    description="Get preview of a generated scene image"
-)
-async def get_preview(job_id: str, scene_id: int):
-    """Get scene preview"""
-    scene = story_creator.get_preview(job_id, scene_id)
-    
-    if not scene:
-        raise HTTPException(status_code=404, detail="Scene not found")
-    
-    return PreviewResponse(
-        scene_id=scene["scene_id"],
-        image_url=scene["image_path"],
-        prompt=scene["prompt"]
-    )
-
-
-@router.get("/download/{job_id}",
+@router.get("/story-reel/{video_id}",
     summary="Download video",
-    description="Download the generated video file",
+    description="Download the generated story reel video file (MP4 format)",
     responses={
         200: {"description": "Video file", "content": {"video/mp4": {}}},
         404: {"description": "Video not found"}
     }
 )
-async def download_video(job_id: str):
-    """Download generated video"""
-    video_path = story_creator.get_video_path(job_id)
+async def download_story_reel(video_id: str):
+    """Download/stream a story reel"""
+    video_path = story_creator.get_video_path(video_id)
     
     if not video_path or not video_path.exists():
         raise HTTPException(status_code=404, detail="Video not found")
@@ -112,5 +95,52 @@ async def download_video(job_id: str):
     return FileResponse(
         video_path,
         media_type="video/mp4",
-        filename=f"story_{job_id}.mp4"
+        filename=f"story_{video_id}.mp4"
     )
+
+
+@router.get("/story-reels",
+    summary="List all story reels",
+    description="Get a list of all story reels with their current status"
+)
+async def list_story_reels():
+    """List all story reels"""
+    videos = story_creator.list_all_videos()
+    return {"videos": videos}
+
+
+@router.delete("/story-reel/{video_id}",
+    summary="Delete story reel",
+    description="Delete a story reel video by its ID"
+)
+async def delete_story_reel(video_id: str):
+    """Delete a story reel"""
+    try:
+        story_creator.delete_video(video_id)
+        return {"success": True}
+    except Exception as e:
+        logger.error(f"Error deleting video: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get("/styles",
+    summary="List image styles",
+    description="Get all available image generation styles"
+)
+async def get_styles():
+    """List available image styles"""
+    return [{"value": s.value, "name": s.name.replace("_", " ").title()} for s in StyleEnum]
+
+
+@router.get("/voices",
+    summary="List TTS voices",
+    description="Get all available text-to-speech voice options"
+)
+async def get_voices():
+    """List available TTS voices"""
+    return [
+        {"value": "af_heart", "name": "Female - Heart"},
+        {"value": "am_adam", "name": "Male - Adam"},
+        {"value": "af_bella", "name": "Female - Bella"},
+        {"value": "am_michael", "name": "Male - Michael"}
+    ]

modules/story_reels/services/story_creator.py

@@ -95,6 +95,71 @@ class StoryCreator:
             
         return job_id
     
+    def get_status(self, job_id: str) -> Dict:
+        """Get job status by ID"""
+        job = self.jobs.get(job_id)
+        if not job:
+            return {
+                "job_id": job_id,
+                "status": "not_found",
+                "progress": 0,
+                "error": "Job not found"
+            }
+        
+        return {
+            "job_id": job["id"],
+            "status": job["status"],
+            "progress": job.get("progress", 0),
+            "video_url": job.get("video_url"),
+            "duration": job.get("duration"),
+            "error": job.get("error")
+        }
+    
+    def get_video_path(self, job_id: str) -> Path:
+        """Get video file path for download"""
+        video_path = self.config.videos_dir_path / f"{job_id}.mp4"
+        if video_path.exists():
+            return video_path
+        return None
+    
+    def get_preview(self, job_id: str, scene_id: int) -> Dict:
+        """Get scene preview"""
+        job = self.jobs.get(job_id)
+        if not job or not job.get("scenes"):
+            return None
+        
+        for scene in job["scenes"]:
+            if scene.get("scene_id") == scene_id:
+                return scene
+        return None
+    
+    def list_all_videos(self) -> List[Dict]:
+        """List all videos with their status"""
+        videos = []
+        for job_id, job in self.jobs.items():
+            videos.append({
+                "video_id": job_id,
+                "topic": job.get("topic", ""),
+                "status": job.get("status", "unknown"),
+                "progress": job.get("progress", 0),
+                "video_url": job.get("video_url"),
+                "created_at": job.get("created_at")
+            })
+        return videos
+    
+    def delete_video(self, job_id: str) -> bool:
+        """Delete a video by ID"""
+        # Remove from jobs dict
+        if job_id in self.jobs:
+            del self.jobs[job_id]
+        
+        # Delete video file
+        video_path = self.config.videos_dir_path / f"{job_id}.mp4"
+        if video_path.exists():
+            video_path.unlink()
+        
+        return True
+    
     async def process_queue(self):
         """Process jobs in queue"""
         if self.processing:

static/index.html

@@ -395,7 +395,7 @@
             };
 
             try {
-                const res = await fetch('/api/story/generate', {
+                const res = await fetch('/api/story/story-reel', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify(data)
@@ -415,11 +415,11 @@
         // Poll status
         async function pollStatus(jobId, type) {
             const status = document.getElementById(type + 'Status');
-            const endpoint = type === 'story' ? '/api/story/status/' : '/api/video/short-video/';
+            const endpoint = type === 'story' ? '/api/story/story-reel/' : '/api/video/short-video/';
 
             const check = async () => {
                 try {
-                    const res = await fetch(endpoint + jobId + (type === 'video' ? '/status' : ''));
+                    const res = await fetch(endpoint + jobId + '/status');
                     const data = await res.json();
 
                     const progress = data.progress || 0;
@@ -430,7 +430,7 @@
 
                     if (data.status === 'ready') {
                         status.className = 'status success';
-                        const downloadUrl = type === 'story' ? `/api/story/download/${jobId}` : `/api/video/short-video/${jobId}`;
+                        const downloadUrl = type === 'story' ? `/api/story/story-reel/${jobId}` : `/api/video/short-video/${jobId}`;
                         status.innerHTML += `<br><a href="${downloadUrl}" class="btn btn-primary" style="margin-top: 1rem; display: inline-block;">📥 Download Video</a>`;
                     } else if (data.status === 'failed') {
                         status.className = 'status error';