Spaces:
Runtime error
Runtime error
File size: 3,194 Bytes
5bbd2a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
import requests, json, os
from string import Template
from urllib.parse import quote
import time
def send_request(endpoint, endpoint_data, text, cha_name, character_emotion, text_language, top_k=6, top_p=0.8, temperature=0.8):
urlencoded_text = requests.utils.quote(text)
# 使用Template填充变量
endpoint_template = Template(endpoint)
final_endpoint = endpoint_template.substitute(chaName=cha_name, speakText=urlencoded_text,textLanguage=text_language, topK=top_k, topP=top_p, temperature=temperature, characterEmotion=character_emotion)
endpoint_data_template = Template(endpoint_data)
filled_json_str = endpoint_data_template.substitute(chaName=cha_name, speakText=urlencoded_text,textLanguage=text_language, topK=top_k, topP=top_p, temperature=temperature, characterEmotion=character_emotion)
# 解析填充后的JSON字符串
request_data = json.loads(filled_json_str)
body = request_data["body"]
# 发送POST请求
response = requests.post(final_endpoint, json=body)
# 检查请求是否成功
if response.status_code == 200:
# 生成保存路径
save_path = f"tmp_audio/{cha_name}/{quote(character_emotion)}.wav"
# 检查保存路径是否存在
if not os.path.exists(f"tmp_audio/{cha_name}/"):
os.makedirs(f"tmp_audio/{cha_name}/")
# 保存音频文件到本地
with open(save_path, "wb") as f:
f.write(response.content)
else:
print(f"请求失败，状态码：{response.status_code}")
global models_path
models_path = r"D:\123pan\Downloads\准备重新封包"
def load_info_config(character_name):
emotion_options = ["default"]
try:
with open(f"{models_path}/{character_name}/infer_config.json", "r", encoding="utf-8") as f:
config = json.load(f)
emotion_list=config.get('emotion_list', None)
if emotion_list is not None:
emotion_options = []
for emotion, details in emotion_list.items():
emotion_options.append(emotion)
except:
pass
return emotion_options
default_endpoint = "http://127.0.0.1:5000/tts"
default_endpoint_data = """{
"method": "POST",
"body": {
"cha_name": "${chaName}",
"character_emotion": "${characterEmotion}",
"text": "${speakText}",
"text_language": "${textLanguage}",
"top_k": ${topK},
"top_p": ${topP},
"temperature": ${temperature}
}
}
"""
default_text="""
我可太激动了！跑近一看，居然是一群小动物组成的戏团！牵着火焰小马的猴子、指挥戏团的兔子、双脚站立不停跳舞的大猫…
它们唱唱跳跳地
带我走出了森林，还让我务必收下这个面具！它们说，这个就是他们看到我时的样子！但你看，这面具明明是只狐狸，怎么可能是我呢？"""
character_name = "花火"
emotion_options = load_info_config(character_name)
for emotion in emotion_options:
print(emotion)
send_request(default_endpoint, default_endpoint_data, default_text, character_name, emotion, "多语种混合")
time.sleep(20)
|