54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
import subprocess
|
||
import sys
|
||
import os
|
||
from pathlib import Path
|
||
import glob
|
||
import tempfile
|
||
import shutil
|
||
from datetime import datetime
|
||
|
||
class A2FService:
|
||
def __init__(self, a2f_url="192.168.1.39:52000"):
|
||
self.base_dir = Path(__file__).parent.parent.parent
|
||
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
|
||
|
||
def audio_to_csv(self, audio_path: str) -> tuple[str, str]:
|
||
# 使用时间戳创建独立的临时工作目录
|
||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S_%f')
|
||
temp_work_dir = tempfile.mkdtemp(prefix=f"a2f_work_{timestamp}_")
|
||
|
||
try:
|
||
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=temp_work_dir)
|
||
|
||
if result.returncode != 0:
|
||
raise RuntimeError(f"A2F inference failed: {result.stdout}")
|
||
|
||
# 在工作目录中查找输出
|
||
output_dirs = sorted(glob.glob(os.path.join(temp_work_dir, "output_*")))
|
||
if not output_dirs:
|
||
raise RuntimeError(f"No output directory found in {temp_work_dir}")
|
||
|
||
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}")
|
||
|
||
# 返回CSV路径和临时目录路径（用于后续清理）
|
||
return csv_path, temp_work_dir
|
||
except Exception as e:
|
||
# 出错时清理临时目录
|
||
shutil.rmtree(temp_work_dir, ignore_errors=True)
|
||
raise e
|