Update app.py

app.py

@@ -2,25 +2,35 @@ import gradio as gr
 from transformers import pipeline
 from huggingface_hub import list_models
 import langcodes
+import re
 
-# Tự động lấy danh sách tất cả mô hình MMS-TTS
+# 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 = {}
 
-# Tạo bản đồ tên ngôn ngữ (title case) → mã ISO
+# Duyệt từng mô hình
 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.title()] = code  # Đảm bảo lưu tên ở dạng Title Case
+    repo = model.modelId  # ví dụ: facebook/mms-tts-eng
+    code = repo.split("-")[-1].lower()
+
+    # Tên ngôn ngữ từ metadata hoặc fallback
+    name = ""
+    if model.cardData and "language" in model.cardData:
+        name = model.cardData["language"]
+    else:
+        try:
+            name = langcodes.Language.make(language=code).display_name()
+        except:
+            name = code
+
+    # Chuẩn hóa tên ngôn ngữ (thường hóa + bỏ khoảng trắng thừa)
+    name_clean = re.sub(r"\s+", " ", name.strip().lower())
+    lang_map[name_clean] = code
 
 def tts_fn(text, language):
-    # Normalize input language: bỏ khoảng trắng, viết lại dạng Title Case
-    language_clean = language.strip().title()
-    code = lang_map.get(language_clean)
+    lang_input = re.sub(r"\s+", " ", language.strip().lower())
+    code = lang_map.get(lang_input)
     if not code:
         return f"❌ Unsupported language name: '{language}'", None
     model_id = f"facebook/mms-tts-{code}"
@@ -34,7 +44,7 @@ def tts_fn(text, language):
 with gr.Blocks() as demo:
     gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages (No ISO Code Needed)")
     with gr.Row():
-        language = gr.Textbox(label="Enter language name (e.g., Meadow Mari)")
+        language = gr.Textbox(label="Enter language name (e.g., English, Meadow Mari)")
         text = gr.Textbox(label="Enter your text", placeholder="Type something here...")
     with gr.Row():
         btn = gr.Button("Speak")