cat-shark
63 lines
1.4 KiB
Python
63 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
数字人处理脚本
|
|
用于调用 Gradio API 生成数字人视频
|
|
"""
|
|
|
|
import sys
|
|
import json
|
|
|
|
|
|
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()
|