Update api.py

api.py

@@ -23,6 +23,8 @@ from character_detection import detect_characters_from_video
 
 from pipelines.audiodescription import generate as ad_generate
 
+from storage.files.file_manager import FileManager
+
 app = FastAPI(title="Veureu Engine API", version="0.2.0")
 app.add_middleware(
     CORSMiddleware,
@@ -40,6 +42,8 @@ VIDEOS_ROOT = Path("/tmp/data/videos")
 VIDEOS_ROOT.mkdir(parents=True, exist_ok=True)
 IDENTITIES_ROOT = Path("/tmp/characters")
 IDENTITIES_ROOT.mkdir(parents=True, exist_ok=True)
+MEDIA_ROOT = Path("/data/media")
+file_manager = FileManager(MEDIA_ROOT)
 
 # Sistema de jobs asíncronos
 class JobStatus(str, Enum):
@@ -50,6 +54,63 @@ class JobStatus(str, Enum):
 
 jobs: Dict[str, dict] = {}
 
+
+@app.post("/upload_video", tags=["Media Upload"])
+async def upload_video(video: UploadFile = File(...)):
+    """
+    Saves an uploaded video by hashing it with SHA1 and placing it under:
+    /data/media/<sha1>/clip/<original_filename>
+
+    Steps:
+    - Compute SHA1 of the uploaded video.
+    - Ensure /data/media exists.
+    - Create folder /data/media/<sha1> if missing.
+    - Create folder /data/media/<sha1>/clip if missing.
+    - Save the video inside /data/media/<sha1>/clip/.
+    """
+
+    # Read content into memory (needed to compute hash twice)
+    file_bytes = await video.read()
+
+    # Create an in-memory file handler
+    import io
+    file_handler = io.BytesIO(file_bytes)
+
+    # Compute SHA1 using your FileManager method
+    try:
+        sha1 = file_manager.compute_sha1(file_handler)
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"SHA1 computation failed: {exc}")
+
+    # Ensure /data/media exists
+    MEDIA_ROOT.mkdir(parents=True, exist_ok=True)
+
+    # Path: /data/media/<sha1>
+    video_root = MEDIA_ROOT / sha1
+    video_root.mkdir(parents=True, exist_ok=True)
+
+    # Path: /data/media/<sha1>/clip
+    clip_dir = video_root / "clip"
+    clip_dir.mkdir(parents=True, exist_ok=True)
+
+    # Final file path
+    final_path = clip_dir / video.filename
+
+    # Save file using your FileManager.upload_file
+    save_result = file_manager.upload_file(io.BytesIO(file_bytes), final_path)
+
+    if not save_result["operation_success"]:
+        raise HTTPException(status_code=500, detail=save_result["error"])
+
+    return JSONResponse(
+        status_code=200,
+        content={
+            "status": "ok",
+            "sha1": sha1,
+            "saved_to": str(final_path)
+        }
+    )
+
 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).