|
|
|
|
|
|
|
|
|
|
|
import requests
|
|
|
|
|
|
class VeureuEngineClient:
|
|
|
def __init__(self, base_url: str):
|
|
|
self.base = base_url.rstrip("/")
|
|
|
|
|
|
def process_video(self, video_path: str, **kwargs):
|
|
|
with open(video_path, "rb") as f:
|
|
|
files = {"video_file": (Path(video_path).name, f, "video/mp4")}
|
|
|
data = {"config_path": kwargs.get("config_path", "config_veureu.yaml"),
|
|
|
"out_root": kwargs.get("out_root", "results"),
|
|
|
"db_dir": kwargs.get("db_dir", "chroma_db")}
|
|
|
r = requests.post(f"{self.base}/process_video", files=files, data=data, timeout=3600)
|
|
|
r.raise_for_status()
|
|
|
return r.json()
|
|
|
|
|
|
def load_casting(self, faces_dir: str, voices_dir: str, db_dir: str = "chroma_db", drop_collections: bool = False):
|
|
|
data = {"faces_dir": faces_dir, "voices_dir": voices_dir, "db_dir": db_dir, "drop_collections": str(drop_collections)}
|
|
|
r = requests.post(f"{self.base}/load_casting", data=data, timeout=600)
|
|
|
r.raise_for_status(); return r.json()
|
|
|
|
|
|
def refine_narration(self, dialogues_srt: str, frame_descriptions: list, model_url: str, une_guidelines_path: str):
|
|
|
data = {
|
|
|
"dialogues_srt": dialogues_srt,
|
|
|
"frame_descriptions_json": json.dumps(frame_descriptions, ensure_ascii=False),
|
|
|
"model_url": model_url,
|
|
|
"une_guidelines_path": une_guidelines_path,
|
|
|
}
|
|
|
r = requests.post(f"{self.base}/refine_narration", data=data, timeout=600)
|
|
|
r.raise_for_status(); return r.json() |
