Update api.py

api.py

@@ -55,7 +55,7 @@ class JobStatus(str, Enum):
 jobs: Dict[str, dict] = {}
 
 
-@app.post("/upload_video", tags=["Media Upload"])
+@app.post("/upload_video", tags=["Media Manager"])
 async def upload_video(video: UploadFile = File(...)):
     """
     Saves an uploaded video by hashing it with SHA1 and placing it under:
@@ -111,6 +111,73 @@ async def upload_video(video: UploadFile = File(...)):
         }
     )
 
+
+@app.get("/download_video/{sha1}", tags=["Media Manager"])
+def download_video(sha1: str):
+    """
+    Download a stored video by its SHA-1 directory name.
+
+    This endpoint looks for a video stored under the path:
+        /data/media/<sha1>/clip/
+    and returns the first MP4 file found in that folder.
+
+    The method performs the following steps:
+    - Checks if the SHA-1 folder exists inside the media root.
+    - Validates that the "clip" subfolder exists.
+    - Searches for the first .mp4 file inside the clip folder.
+    - Uses the FileManager.get_file method to ensure the file is accessible.
+    - Returns the video directly as a FileResponse.
+
+    Parameters
+    ----------
+    sha1 : str
+        The SHA-1 hash corresponding to the directory where the video is stored.
+
+    Returns
+    -------
+    FileResponse
+        A streaming response containing the MP4 video.
+
+    Raises
+    ------
+    HTTPException
+        - 404 if the SHA-1 folder does not exist.
+        - 404 if the clip folder is missing.
+        - 404 if no MP4 files are found.
+        - 404 if the file cannot be retrieved using FileManager.
+    """
+    sha1_folder = MEDIA_ROOT / sha1
+    clip_folder = sha1_folder / "clip"
+
+    if not sha1_folder.exists() or not sha1_folder.is_dir():
+        raise HTTPException(status_code=404, detail="SHA1 folder not found")
+
+    if not clip_folder.exists() or not clip_folder.is_dir():
+        raise HTTPException(status_code=404, detail="Clip folder not found")
+
+    # Find first MP4 file
+    mp4_files = list(clip_folder.glob("*.mp4"))
+    if not mp4_files:
+        raise HTTPException(status_code=404, detail="No MP4 files found")
+
+    video_path = mp4_files[0]
+
+    # Convert to relative path for FileManager
+    relative_path = video_path.relative_to(MEDIA_ROOT)
+
+    handler = file_manager.get_file(relative_path)
+    if handler is None:
+        raise HTTPException(status_code=404, detail="Video not accessible")
+
+    handler.close()
+
+    return FileResponse(
+        path=video_path,
+        media_type="video/mp4",
+        filename=video_path.name
+    )
+
+
 def describe_image_with_svision(image_path: str, is_face: bool = True) -> tuple[str, str]:
     """
     Llama al space svision para describir una imagen (usado en generación de AD).