import asyncio
import logging
import os

from tts import BlackboxTTS

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 BlackboxTTS with URL: {url}")
    print(f"Using prompt wav: {prompt_wav}")

    blackbox_tts = BlackboxTTS(
        url=url,
        model_name="voxcpmtts",
        prompt_wav_path=prompt_wav,
        params={
            "streaming": "false",
            "prompt_text": "澳门有乜嘢好食嘅",
            "cfg_value": "2.0",
            "inference_timesteps": "10",
            "do_normalize": "true",
            "denoise": "true",
            "retry_badcase": "true",
            "retry_badcase_max_times": "3",
            "retry_badcase_ratio_threshold": "6.0",
        },
    )

    text = "你好，这是一段测试文本"
    print(f"Synthesizing text: {text}")

    try:
        stream = blackbox_tts.synthesize(text)
        audio_frame = await stream.collect()

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