Complete cloud storage: return cloud URL when HF upload successful, redirect download

modules/story_reels/router.py

@@ -86,7 +86,21 @@ async def get_story_status(video_id: str):
     }
 )
 async def download_story_reel(video_id: str):
-    """Download/stream a story reel"""
+    """Download/stream a story reel - supports both local and cloud storage"""
+    # Get job status to check storage type
+    status = story_creator.get_status(video_id)
+    
+    if status.get("status") == "not_found":
+        raise HTTPException(status_code=404, detail="Video not found")
+    
+    video_url = status.get("video_url")
+    
+    # If video is in cloud (HF Hub), redirect to cloud URL
+    if video_url and video_url.startswith("https://"):
+        from fastapi.responses import RedirectResponse
+        return RedirectResponse(url=video_url, status_code=302)
+    
+    # Otherwise serve from local storage
     video_path = story_creator.get_video_path(video_id)
     
     if not video_path.exists():

modules/video_creator/router.py

@@ -72,7 +72,18 @@ async def get_video_status(video_id: str):
     }
 )
 async def get_video(video_id: str):
-    """Download/stream a video"""
+    """Download/stream a video - supports both local and cloud storage"""
+    from config import config
+    
+    # Check if video is in cloud (has .cloud metadata file)
+    cloud_meta_path = config.videos_dir_path / f"{video_id}.cloud"
+    if cloud_meta_path.exists():
+        cloud_url = cloud_meta_path.read_text().strip()
+        if cloud_url:
+            from fastapi.responses import RedirectResponse
+            return RedirectResponse(url=cloud_url, status_code=302)
+    
+    # Otherwise serve from local storage
     video_path = short_creator.get_video_path(video_id)
     
     if not video_path.exists():

modules/video_creator/services/short_creator.py

@@ -339,8 +339,10 @@ class ShortCreator:
                             folder="short_video"  # Module-specific folder
                         )
                         if cloud_url:
-                            # Store cloud URL for status response
-                            # Note: video_creator uses file-based status, so we just log
+                            # Save cloud URL to metadata file for download redirect
+                            cloud_meta_path = self.config.videos_dir_path / f"{video_id}.cloud"
+                            cloud_meta_path.write_text(cloud_url)
+                            # Delete local video file to save space
                             output_path.unlink()
                             logger.info(f"Video {video_id} uploaded to HF Hub, local file deleted")
                 except Exception as e: