import gradio as gr
from TTS.api import TTS

# Chọn model Kani-TTS tiếng Việt từ HuggingFace
model_name = "pnnbao-ump/kani-tts-370m-vie"

# Load model
tts = TTS(model_name=model_name, progress_bar=True, gpu=False)

def generate_audio(text):
    output_file = "output.wav"
    tts.tts_to_file(text=text, file_path=output_file)
    return output_file

ui = gr.Interface(
    fn=generate_audio,
    inputs=gr.Textbox(label="Nhập văn bản tiếng Việt"),
    outputs=gr.Audio(label="Kết quả giọng nói"),
    title="Text-to-Speech Tiếng Việt",
    description="Sử dụng model Kani-TTS tiếng Việt (CPU, miễn phí)"
)

ui.launch()