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())