Update app.py

app.py

@@ -1,41 +1,8 @@
 import gradio as gr
 from transformers import pipeline
-from huggingface_hub import list_models
-import langcodes
 
-# Tự động lấy danh sách mô hình MMS-TTS
-all_models = list_models(filter="facebook/mms-tts", full=True)
-lang_map = {}
-
-manual_map = {
-    "mhr": "Meadow Mari",
-    "mrj": "Hill Mari",
-    "cmn": "Mandarin Chinese",
-    "arb": "Arabic",
-    # thêm nếu cần
-}
-
-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()
-    except:
-        name = code
-    # Nếu langcodes không giải mã đúng, dùng thủ công
-    if name.lower() == code.lower() and code in manual_map:
-        name = manual_map[code]
-    lang_map[name.lower()] = code  # dùng tên thường hết
-
-# In ra để kiểm tra
-print("🧠 Available language names:")
-for lang in sorted(lang_map.keys()):
-    print("-", lang)
-
-def tts_fn(text, language):
-    code = lang_map.get(language.lower())
-    if not code:
-        return f"❌ Unsupported language name: '{language}'", None
+def tts_fn(text, language_iso_code):
+    code = language_iso_code.strip().lower()
     model_id = f"facebook/mms-tts-{code}"
     try:
         tts = pipeline("text-to-speech", model=model_id)
@@ -45,10 +12,10 @@ def tts_fn(text, language):
         return f"❌ Error: {str(e)}", None
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages (No ISO Code Needed)")
+    gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages (ISO Code Input Only)")
     with gr.Row():
-        language = gr.Textbox(label="Enter language name (e.g., Meadow Mari)", placeholder="Type a language name...")
-        text = gr.Textbox(label="Enter your text", placeholder="Type something here...")
+        language = gr.Textbox(label="Language ISO Code (e.g. eng, mhr, bod)", placeholder="Enter ISO code here")
+        text = gr.Textbox(label="Enter your text", placeholder="Type something here…")
     with gr.Row():
         btn = gr.Button("Speak")
     with gr.Row():