Update app.py

app.py

@@ -3,43 +3,46 @@ from transformers import pipeline
 from huggingface_hub import list_models
 import langcodes
 
-# 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)
+# Tải danh sách mô hình MMS-TTS
 lang_map = {}
 
-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()
-    if name not in lang_map:  # Tránh trùng tên
-        lang_map[name] = code
-
-def tts_fn(text, language):
-    code = lang_map.get(language)
+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)
     if not code:
-        return "Unsupported language", None
+        return f"❌ Unsupported language name: '{language_name}'", 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")
+        return f"✅ Model: {model_id}", (out["audio"], "output.wav")
     except Exception as e:
-        return f"Error: {str(e)}", None
+        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 – Type Language Name")
+
     with gr.Row():
-        language = gr.Dropdown(
-            choices=sorted(lang_map.keys()),
-            label="Choose a language",
-            interactive=True
-        )
+        language = gr.Textbox(label="Language name", placeholder="e.g., Kazakh, Bashkir, French...")
         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")