63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
数字人处理脚本（对齐 zhenqianba digital_human_process.py）
|
||
调用 Gradio API：音频 + 形象视频 → 口播视频
|
||
"""
|
||
|
||
import json
|
||
import sys
|
||
|
||
|
||
def main():
|
||
if len(sys.argv) < 4:
|
||
print(
|
||
json.dumps(
|
||
{
|
||
"success": False,
|
||
"error": "参数不足: 需要 api_url, audio_file, video_file",
|
||
}
|
||
)
|
||
)
|
||
sys.exit(1)
|
||
|
||
api_url = sys.argv[1]
|
||
audio_file = sys.argv[2]
|
||
video_file = sys.argv[3]
|
||
|
||
try:
|
||
from gradio_client import Client, handle_file
|
||
|
||
client = Client(api_url)
|
||
result = client.predict(
|
||
audio_file=handle_file(audio_file),
|
||
video_file={"video": handle_file(video_file)},
|
||
api_name="/process_single",
|
||
)
|
||
|
||
if isinstance(result, dict) and result.get("video"):
|
||
print(
|
||
json.dumps(
|
||
{
|
||
"success": True,
|
||
"videoPath": result["video"],
|
||
"subtitles": result.get("subtitles"),
|
||
}
|
||
)
|
||
)
|
||
sys.exit(0)
|
||
|
||
print(
|
||
json.dumps(
|
||
{"success": False, "error": f"未获取到视频结果: {str(result)[:500]}"}
|
||
)
|
||
)
|
||
sys.exit(1)
|
||
|
||
except Exception as e:
|
||
print(json.dumps({"success": False, "error": str(e)}))
|
||
sys.exit(1)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|