Update app.py

app.py

@@ -3,28 +3,26 @@ from transformers import pipeline
 from huggingface_hub import list_models
 import langcodes
 
-# Tải danh sách mô hình MMS-TTS
+# 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 = {}
 
-def load_languages():
-    global lang_map
-    all_models = list_models(filter="facebook/mms-tts", full=True)
-    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.lower()] = code  # key là tên thường để dễ so sánh
-
-load_languages()
-
-def tts_fn(text, language_name):
-    lang_key = language_name.strip().lower()
-    code = lang_map.get(lang_key)
+# Tạo bản đồ tên ngôn ngữ (title case) → mã ISO
+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
+
+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)
     if not code:
-        return f"❌ Unsupported language name: '{language_name}'", None
+        return f"❌ Unsupported language name: '{language}'", None
     model_id = f"facebook/mms-tts-{code}"
     try:
         tts = pipeline("text-to-speech", model=model_id)
@@ -34,15 +32,12 @@ def tts_fn(text, language_name):
         return f"❌ Error: {str(e)}", None
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages – Type Language Name")
-
+    gr.Markdown("# 🌐 MMS-TTS: 1100+ Languages (No ISO Code Needed)")
     with gr.Row():
-        language = gr.Textbox(label="Language name", placeholder="e.g., Kazakh, Bashkir, French...")
+        language = gr.Textbox(label="Enter language name (e.g., Meadow Mari)")
         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")