Update app.py

app.py

@@ -1,17 +1,44 @@
+import gradio as gr
 from transformers import pipeline
+from huggingface_hub import list_models
+import langcodes
 
-def main():
-    # Dùng model Facebook MMS TTS đa ngôn ngữ
-    model_name = "facebook/mms-tts-1b"
-    tts = pipeline("text-to-speech", model=model_name)
+# Tự động lấy danh sách tất cả mô hình MMS-TTS
+all_models = list_models(filter="facebook/mms-tts", full=True)
+lang_map = {}
 
-    text = "Hello world, this is a multi-language TTS test."
-    audio = tts(text)
+for model in all_models:
+    repo = model.modelId  # ví dụ: facebook/mms-tts-kaz
+    code = repo.split("-")[-1]
+    try:
+        name = langcodes.Language.make(language=code).display_name().title()
+    except:
+        name = code.upper()
+    lang_map[name] = code
 
-    with open("output.wav", "wb") as f:
-        f.write(audio["wav"])
+def tts_fn(text, language):
+    code = lang_map.get(language)
+    if not code:
+        return "Unsupported language", None
+    model_id = f"facebook/mms-tts-{code}"
+    try:
+        tts = pipeline("text-to-speech", model=model_id)
+        out = tts(text)
+        return f"✅ Model: {model_id}", (out["audio"], "output.wav")
+    except Exception as e:
+        return f"❌ Error: {str(e)}", None
 
-    print("TTS thành công, file output.wav đã lưu")
+with gr.Blocks() as demo:
+    gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages (No ISO Code Needed)")
+    with gr.Row():
+        language = gr.Dropdown(choices=sorted(lang_map.keys()), label="Choose a language")
+        text = gr.Textbox(label="Enter your text", placeholder="Type something here...")
+    with gr.Row():
+        btn = gr.Button("Speak")
+    with gr.Row():
+        out_text = gr.Textbox(label="Info")
+        out_audio = gr.Audio(label="Audio", type="filepath")
 
-if __name__ == "__main__":
-    main()
+    btn.click(fn=tts_fn, inputs=[text, language], outputs=[out_text, out_audio])
+
+demo.launch()