39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
import math
|
|
import os
|
|
import requests
|
|
from pprint import pprint
|
|
import argparse
|
|
import soundfile
|
|
from a2f_api import A2F
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("audio_path")
|
|
parser.add_argument("--host", type=str,
|
|
default="https://a2fdemo.piedata.ai/")
|
|
args = parser.parse_args()
|
|
|
|
|
|
a2f = A2F(args.host)
|
|
print(f"Uploading {args.audio_path}...")
|
|
server_audio_path = a2f.upload(args.audio_path)
|
|
fname = os.path.basename(server_audio_path)
|
|
print("Status: ", a2f.status())
|
|
print("EmotionNames: ", a2f.get_emotion_names())
|
|
print("Scene Objects: ", a2f.get_scene_objects())
|
|
print("Scene Players: ", a2f.get_players())
|
|
print("Preprocessing settings: ", a2f.get_pre_settings())
|
|
print("Postprocessing settings: ", a2f.get_post_settings())
|
|
print("Setting player root: ", a2f.set_player_root("/home/ubuntu/results"))
|
|
print("Player root: ", a2f.get_player_root())
|
|
print("Setting audio: ", a2f.set_audio(os.path.basename(server_audio_path)))
|
|
print("Audio Range: ", a2f.get_audio_range())
|
|
print("Running: ", a2f.run())
|
|
print("NumKeys: ", a2f.get_number_of_keys())
|
|
print("Keys: ", a2f.get_generated_keys())
|
|
# print("BlendShape solvers: ", a2f.get_blendshape_solvers())
|
|
print("Exporting: ", a2f.export_json("/home/ubuntu/results",
|
|
filename=os.path.splitext(fname)[0]))
|
|
print(f"Pulling to ./{fname}...")
|
|
a2f.pull(fname)
|