Update storage/embeddings_routers.py

storage/embeddings_routers.py

@@ -1,5 +1,6 @@
 import os
 import io
+import json
 import shutil
 
 import sqlite3
@@ -113,3 +114,49 @@ async def upload_embeddings(
             "saved_to": str(final_path)
         }
     )
+
+def get_embeddings_json(video_hash: str, embedding_type: str):
+    """
+    Returns the parsed embeddings.json for a given video and type.
+
+    Behavior:
+    - Validate embedding_type.
+    - Build the file path: /data/embeddings/<video_hash>/<embedding_type>/embeddings.json
+    - Raise HTTPException if missing.
+    - Load and return parsed JSON.
+    """
+
+    if embedding_type not in ("faces", "voices"):
+        raise HTTPException(status_code=400, detail="embedding_type must be 'faces' or 'voices'")
+
+    target_file = EMBEDDINGS_ROOT / video_hash / embedding_type / "embeddings.json"
+
+    if not target_file.exists():
+        raise HTTPException(
+            status_code=404,
+            detail=f"embeddings.json not found for video={video_hash}, type={embedding_type}"
+        )
+
+    try:
+        with open(target_file, "r", encoding="utf-8") as f:
+            data = json.load(f)
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"Failed to read embeddings: {exc}")
+
+    return data
+
+
+@router.get("/get_embedding", tags=["Embeddings Manager"])
+def get_embeddings(
+    video_hash: str = Query(..., description="Hash of the video"),
+    embedding_type: str = Query(..., description="faces or voices"),
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Endpoint to retrieve embeddings.json for a given video hash and type.
+    """
+    validate_token(token)
+
+    data = get_embeddings_json(video_hash, embedding_type)
+
+    return data