Add image retry logic (NVIDIA) and fix video playback mounting

app.py

@@ -104,6 +104,12 @@ static_dir = Path(__file__).parent / "static"
 if static_dir.exists():
     app.mount("/static", StaticFiles(directory=str(static_dir)), name="static")
 
+# Mount videos directory for direct access and streaming
+videos_dir = config.videos_dir_path
+if not videos_dir.exists():
+    videos_dir.mkdir(parents=True, exist_ok=True)
+app.mount("/videos", StaticFiles(directory=str(videos_dir)), name="videos")
+
 
 if __name__ == "__main__":
     import uvicorn

modules/story_reels/services/story_creator.py

@@ -361,8 +361,39 @@ class StoryCreator:
             else:
                 raise Exception("No image generation client available!")
             
+            # ====================
+            # Retry Logic for Failed Images
+            # ====================
+            successful_ids = {r["id"] for r in batch_results if r.get("path")}
+            missing_ids = [chunk['chunk_id'] for chunk in image_chunks if chunk['chunk_id'] not in successful_ids]
+            
+            if missing_ids and has_nvidia:
+                logger.warning(f"[{job_id}] Retrying {len(missing_ids)} failed images with NVIDIA...")
+                
+                # Filter prompts for missing IDs
+                retry_prompts = [p for p in prompts_list if p[0] in missing_ids]  # p is (idx, prompt_text)
+                
+                # Retry with NVIDIA (more robust)
+                retry_results = self.nvidia.generate_batch(
+                    prompts=retry_prompts,
+                    output_dir=temp_dir,
+                    seed=seed + 1,  # Change seed slightly
+                    batch_size=5,
+                    delay_seconds=2.0  # Slower delay for safety
+                )
+                
+                # Add successful retries
+                for res in retry_results:
+                    if res.get("path"):
+                        batch_results.append(res)
+                        logger.info(f"[{job_id}] Recovered image {res['id']}")
+
             # Build generated_scenes from batch results
             generated_scenes = []
+            
+            # Sort all results by ID to ensure correct order
+            batch_results.sort(key=lambda x: x["id"])
+            
             for result in batch_results:
                 if result.get("path"):
                     temp_files.append(Path(result["path"]))
@@ -381,7 +412,7 @@ class StoryCreator:
                         "duration": scene_duration
                     })
             
-            logger.info(f"[{job_id}] Generated {len(generated_scenes)}/{len(ai_prompts)} images")
+            logger.info(f"[{job_id}] Final count: {len(generated_scenes)}/{len(ai_prompts)} images")
             
             job["scenes"] = generated_scenes
             job["progress"] = 80
@@ -400,7 +431,7 @@ class StoryCreator:
                 output_path=output_path
             )
             
-            job["video_url"] = str(output_path)
+            job["video_url"] = f"/videos/{job_id}.mp4"
             job["duration"] = audio_duration
             job["progress"] = 100