Update preprocessing_router.py

preprocessing_router.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 from fastapi import APIRouter, UploadFile, File, Form, BackgroundTasks, HTTPException, Body
 from fastapi.responses import FileResponse
-from pathlib import Path
 from datetime import datetime
 from enum import Enum
 from typing import Dict, Any, List
@@ -12,6 +11,7 @@ import uuid
 import numpy as np
 import cv2
 import tempfile
+from pathlib import Path
 
 from casting_loader import ensure_chroma, build_faces_index, build_voices_index
 from llm_router import load_yaml, LLMRouter
@@ -212,6 +212,16 @@ async def load_casting(
     return {"ok": True, "faces": n_faces, "voices": n_voices}
 
 
+from pathlib import Path
+def find_video_hash(filename: str, media_root) -> str | None:
+    for hash_dir in media_root.iterdir():
+        if hash_dir.is_dir():
+            clips_dir = hash_dir / "clips"
+            video_path = clips_dir / filename
+            if video_path.exists():
+                return hash_dir.name
+    return None
+
 @router.post("/finalize_casting")
 async def finalize_casting(
     payload: dict = Body(...),
@@ -225,6 +235,11 @@ async def finalize_casting(
     video_hash = payload.get("video_hash") or "empty"
     voice_clusters = payload.get("voice_clusters", []) or []
 
+    MEDIA_ROOT = _P("/data/media")
+    video_hash = find_video_hash(video_name,MEDIA_ROOT)
+    print(video_name)
+    print(characters)
+    print(voice_clusters)
     if not video_name or not base_dir:
         raise HTTPException(status_code=400, detail="Missing video_name or base_dir")