51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
import asyncio
|
|
import os
|
|
import logging
|
|
from tts_voxcpm import VoxCPMTTS
|
|
from livekit.agents import tts
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
async def test_tts():
|
|
# Use the URL from the user's curl command
|
|
url = "http://10.6.80.21:5002/tts-blackbox"
|
|
|
|
# Check if we have a real wav file to test with
|
|
# In the earlier find_by_name, we found tests/change-sophie.wav
|
|
prompt_wav = "/home/verachen/Music/voice/2food.wav"
|
|
if not os.path.exists(prompt_wav):
|
|
prompt_wav = "/home/verachen/Music/voice/2food.wav" # fallback to the one in curl
|
|
|
|
print(f"Testing VoxCPMTTS with URL: {url}")
|
|
print(f"Using prompt wav: {prompt_wav}")
|
|
|
|
vox_tts = VoxCPMTTS(
|
|
url=url,
|
|
prompt_wav_path=prompt_wav
|
|
)
|
|
|
|
text = "你好，这是一段测试文本"
|
|
print(f"Synthesizing text: {text}")
|
|
|
|
try:
|
|
stream = vox_tts.synthesize(text)
|
|
audio_frame = await stream.collect()
|
|
|
|
print(f"Successfully synthesized audio!")
|
|
print(f"Audio duration: {audio_frame.sample_rate * len(audio_frame.data) / (audio_frame.num_channels * 2)} samples?")
|
|
# Actually AudioFrame has duration or samples
|
|
print(f"Samples: {len(audio_frame.data) // 2}")
|
|
|
|
# Save to file for manual check if possible
|
|
with open("test_output.wav", "wb") as f:
|
|
# This won't be a valid WAV yet if it's just raw PCM,
|
|
# but if collect() returns combined frames, we can use to_wav_bytes()
|
|
f.write(audio_frame.to_wav_bytes())
|
|
print("Saved output to test_output.wav")
|
|
|
|
except Exception as e:
|
|
print(f"TTS test failed: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_tts())
|