from tts_voice import tts_order_voice
import edge_tts
import gradio as gr
import tempfile
import anyio

language_dict = tts_order_voice

async def text_to_speech_edge(text, language_code):
    voice = language_dict[language_code]
    communicate = edge_tts.Communicate(text, voice)
    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
        tmp_path = tmp_file.name

    await communicate.save(tmp_path)

    return "语音合成完成：{}".format(text), tmp_path

input_text = gr.components.Textbox(lines=5, label="输入文本")
output_audio = gr.components.Audio(type="filepath", label="导出文件")
default_language = list(language_dict.keys())[0]
language_choices = list(language_dict.keys())
language_choices.remove(default_language)
language_choices.insert(0, default_language)
language = gr.components.Dropdown(choices=language_choices, label="语言")

interface = gr.Interface(
    fn=text_to_speech_edge, 
    inputs=[input_text, language], 
    outputs=[output_audio], 
    title="Edge TTS 文字转语音"
)

if __name__ == "__main__":
    anyio.run(interface.launch, backend="asyncio")