119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
import requests
|
||
import base64
|
||
import json
|
||
import io
|
||
from IPython.display import Audio
|
||
import soundfile as sf
|
||
|
||
# 假设您的FastAPI服务在本地运行
|
||
BASE_URL = "http://feng-arch.cn:31006"
|
||
|
||
def tts_file_response(text, voice="af_heart", speed=1.0, lang_code="z"):
|
||
"""
|
||
发送请求并直接获取音频文件
|
||
"""
|
||
url = f"{BASE_URL}/tts"
|
||
payload = {
|
||
"text": text,
|
||
"voice": voice,
|
||
"speed": speed,
|
||
"return_type": "file"
|
||
}
|
||
|
||
response = requests.post(url, json=payload)
|
||
|
||
if response.status_code == 200:
|
||
# 保存音频文件
|
||
with open("received_audio.wav", "wb") as f:
|
||
f.write(response.content)
|
||
print("音频已保存为 received_audio.wav")
|
||
|
||
# 如果在Jupyter Notebook中，可以直接播放
|
||
return Audio(data=response.content, rate=24000)
|
||
else:
|
||
print(f"错误: {response.status_code}")
|
||
print(response.text)
|
||
return None
|
||
|
||
def tts_base64_response(text, voice="af_heart", speed=1.0, lang_code="z"):
|
||
"""
|
||
发送请求并获取base64编码的音频数据
|
||
"""
|
||
url = f"{BASE_URL}/tts"
|
||
payload = {
|
||
"text": text,
|
||
"voice": voice,
|
||
"speed": speed,
|
||
"return_type": "base64"
|
||
}
|
||
|
||
response = requests.post(url, json=payload)
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
# 获取base64编码的音频
|
||
audio_base64 = data.get("audio_base64")
|
||
|
||
# 解码base64数据
|
||
audio_data = base64.b64decode(audio_base64)
|
||
|
||
# 保存音频文件
|
||
with open("received_audio.wav", "wb") as f:
|
||
f.write(audio_data)
|
||
print("音频已保存为 received_audio.wav")
|
||
|
||
# 如果在Jupyter Notebook中，可以直接播放
|
||
return Audio(data=audio_data, rate=24000)
|
||
else:
|
||
print(f"错误: {response.status_code}")
|
||
print(response.text)
|
||
return None
|
||
|
||
def get_available_voices(lang_code="z"):
|
||
"""
|
||
获取指定语言的可用声音列表
|
||
Name Traits Target Quality Training Duration Overall Grade SHA256
|
||
af_heart 🚺❤️ A 0ab5709b
|
||
af_alloy 🚺 B MM minutes C 6d877149
|
||
af_aoede 🚺 B H hours C+ c03bd1a4
|
||
af_bella 🚺🔥 A HH hours A- 8cb64e02
|
||
af_jessica 🚺 C MM minutes D cdfdccb8
|
||
af_kore 🚺 B H hours C+ 8bfbc512
|
||
af_nicole 🚺🎧 B HH hours B- c5561808
|
||
af_nova 🚺 B MM minutes C e0233676
|
||
af_river 🚺 C MM minutes D e149459b
|
||
af_sarah 🚺 B H hours C+ 49bd364e
|
||
af_sky 🚺 B M minutes 🤏 C- c799548a
|
||
am_adam 🚹 D H hours F+ ced7e284
|
||
am_echo 🚹 C MM minutes D 8bcfdc85
|
||
am_eric 🚹 C MM minutes D ada66f0e
|
||
am_fenrir 🚹 B H hours C+ 98e507ec
|
||
am_liam 🚹 C MM minutes D c8255075
|
||
am_michael 🚹 B H hours C+ 9a443b79
|
||
am_onyx 🚹 C MM minutes D e8452be1
|
||
am_puck 🚹 B H hours C+ dd1d8973
|
||
am_santa 🚹 C M minutes 🤏 D- 7f2f7582
|
||
|
||
Name Traits Target Quality Training Duration Overall Grade SHA256
|
||
zf_xiaobei 🚺 C MM minutes D 9b76be63
|
||
zf_xiaoni 🚺 C MM minutes D 95b49f16
|
||
zf_xiaoxiao 🚺 C MM minutes D cfaf6f2d
|
||
zf_xiaoyi 🚺 C MM minutes D b5235dba
|
||
zm_yunjian 🚹 C MM minutes D 76cbf8ba
|
||
zm_yunxi 🚹 C MM minutes D dbe6e1ce
|
||
zm_yunxia 🚹 C MM minutes D bb2b03b0
|
||
zm_yunyang 🚹 C MM minutes D 5238ac22
|
||
"""
|
||
|
||
|
||
# 示例使用
|
||
if __name__ == "__main__":
|
||
text = "你能解决什么问题"
|
||
|
||
# 获取音频文件
|
||
audio = tts_file_response(text, voice="zf_xiaoxiao")
|
||
|
||
# 或者获取base64编码的音频
|
||
# audio = tts_base64_response(text)
|
||
|