41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import subprocess
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
import glob
|
|
|
|
class A2FService:
|
|
def __init__(self, a2f_url="192.168.1.39:52000"):
|
|
self.base_dir = Path(__file__).parent.parent.parent
|
|
self.output_dir = self.base_dir / "data" / "output"
|
|
self.a2f_script = self.base_dir / "external" / "Audio2Face-3D-Samples" / "scripts" / "audio2face_3d_microservices_interaction_app" / "a2f_3d.py"
|
|
self.config_file = self.base_dir / "external" / "Audio2Face-3D-Samples" / "scripts" / "audio2face_3d_microservices_interaction_app" / "config" / "config_james.yml"
|
|
self.a2f_url = a2f_url
|
|
os.makedirs(self.output_dir, exist_ok=True)
|
|
|
|
def audio_to_csv(self, audio_path: str) -> str:
|
|
cmd = [
|
|
sys.executable,
|
|
str(self.a2f_script),
|
|
"run_inference",
|
|
audio_path,
|
|
str(self.config_file),
|
|
"--url",
|
|
self.a2f_url
|
|
]
|
|
|
|
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, cwd=str(self.output_dir))
|
|
|
|
if result.returncode != 0:
|
|
raise RuntimeError(f"A2F inference failed: {result.stdout}")
|
|
|
|
output_dirs = sorted(glob.glob(str(self.output_dir / "output_*")))
|
|
if not output_dirs:
|
|
raise RuntimeError("No output directory found")
|
|
|
|
csv_path = os.path.join(output_dirs[-1], "animation_frames.csv")
|
|
if not os.path.exists(csv_path):
|
|
raise RuntimeError(f"CSV file not found: {csv_path}")
|
|
|
|
return csv_path
|