Update storage/embeddings_routers.py

storage/embeddings_routers.py

@@ -0,0 +1,75 @@
+import os
+import io
+import shutil
+
+import sqlite3
+
+from pathlib import Path
+
+from fastapi import APIRouter, UploadFile, File, Query, HTTPException
+from fastapi.responses import FileResponse, JSONResponse
+
+
+from storage.files.file_manager import FileManager
+from storage.common import validate_token
+
+router = APIRouter(prefix="/embeddings", tags=["Embeddings Manager"])
+EMBEDDINGS_ROOT = Path("/data/embeddings")
+file_manager = FileManager(EMBEDDINGS_ROOT)
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+
+@router.post("/upload_embeddings", tags=["Embeddings Manager"])
+async def upload_embeddings(
+    file: UploadFile = File(...),
+    embedding_type: str = Query(..., description="faces or voices"),
+    video_hash: str = Query(..., description="Hash of the video"),
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Upload embeddings JSON for a given video and type (faces or voices).
+
+    Behavior:
+    - Validate the token.
+    - Validate embedding_type.
+    - Ensure directory structure: /data/embeddings/<video_hash>/<embedding_type>/
+    - Delete any existing .json file inside that folder.
+    - Save the uploaded embeddings as embeddings.json.
+    """
+    validate_token(token)
+
+    # Validación del tipo
+    if embedding_type not in ("faces", "voices"):
+        raise HTTPException(status_code=400, detail="embedding_type must be 'faces' or 'voices'")
+
+    # Rutas objetivo
+    video_path = EMBEDDINGS_ROOT / video_hash
+    type_path = video_path / embedding_type
+
+    # Crear carpetas si no existen
+    type_path.mkdir(parents=True, exist_ok=True)
+
+    # Eliminar JSONs previos
+    for existing in type_path.glob("*.json"):
+        try:
+            existing.unlink()
+        except Exception as exc:
+            raise HTTPException(status_code=500, detail=f"Failed to delete old embeddings: {exc}")
+
+    # Guardar como embeddings.json
+    final_path = type_path / "embeddings.json"
+
+    try:
+        file_bytes = await file.read()
+        with open(final_path, "wb") as f:
+            f.write(file_bytes)
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"Failed to save embeddings: {exc}")
+
+    return JSONResponse(
+        status_code=200,
+        content={
+            "status": "ok",
+            "saved_to": str(final_path)
+        }
+    )