Update storage/embeddings_routers.py

storage/embeddings_routers.py

@@ -19,6 +19,46 @@ file_manager = FileManager(EMBEDDINGS_ROOT)
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 
+@router.get("/list_embeddings", tags=["Embeddings Manager"])
+def list_all_embeddings(
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    List all embeddings stored under /data/embeddings.
+
+    For each video hash folder, returns:
+    - video: folder name (hash)
+    - faces: true/false depending on whether faces/embeddings.json exists
+    - voices: true/false depending on whether voices/embeddings.json exists
+
+    Notes:
+    - A video folder may contain only faces, only voices, or neither.
+    - Missing folders are treated as false.
+    """
+    validate_token(token)
+
+    results = []
+
+    # If embeddings root does not exist, return empty list
+    if not EMBEDDINGS_ROOT.exists():
+        return []
+
+    for video_dir in EMBEDDINGS_ROOT.iterdir():
+        if not video_dir.is_dir():
+            continue  # Skip anything that is not a folder
+
+        faces_path = video_dir / "faces" / "embeddings.json"
+        voices_path = video_dir / "voices" / "embeddings.json"
+
+        results.append({
+            "video": video_dir.name,
+            "faces": faces_path.exists(),
+            "voices": voices_path.exists()
+        })
+
+    return results
+
+
 @router.post("/upload_embeddings", tags=["Embeddings Manager"])
 async def upload_embeddings(
     file: UploadFile = File(...),