Update app.py

app.py

@@ -4,59 +4,62 @@ from huggingface_hub import list_models
 import langcodes
 import difflib
 
-# Tạo bản đồ ngôn ngữ → mã ISO
+# Map tên ngôn ngữ (không phân biệt hoa/thường) → mã ISO
 lang_map = {}
+
 models = list_models(filter="facebook/mms-tts", full=True)
 
 for model in models:
-    repo_id = model.modelId  # ví dụ: facebook/mms-tts-tur
-    code = repo_id.split("-")[-1].lower()
+    model_id = model.modelId  # ví dụ: facebook/mms-tts-kaz
+    code = model_id.split("-")[-1].lower()
     
-    # Ưu tiên lấy tên từ metadata
+    # Lấy tên ngôn ngữ từ metadata hoặc từ mã ISO
     if model.cardData and "language" in model.cardData:
-        lang_name = model.cardData["language"]
+        name = model.cardData["language"]
     else:
         try:
-            lang_name = langcodes.Language.make(code).display_name()
+            name = langcodes.Language.make(code).display_name()
         except:
-            lang_name = code
+            name = code.upper()
 
-    lang_map[lang_name.strip().lower()] = code
+    # Đưa vào bản đồ, key là lowercase để không phân biệt hoa/thường
+    lang_map[name.strip().lower()] = code
 
-def tts_fn(text, lang_input):
-    lang_key = lang_input.strip().lower()
-    code = lang_map.get(lang_key)
+def tts_fn(text, language):
+    key = language.strip().lower()
+    code = lang_map.get(key)
 
     if not code:
-        # Gợi ý
-        suggestions = difflib.get_close_matches(lang_key, lang_map.keys(), n=5)
-        msg = f"❌ Unsupported language name: '{lang_input}'"
+        # Nếu không tìm thấy → gợi ý tên gần giống
+        suggestions = difflib.get_close_matches(key, lang_map.keys(), n=5)
+        msg = f"❌ Unsupported language name: '{language}'"
         if suggestions:
-            msg += f"\n🔍 Did you mean: {', '.join(suggestions)}?"
+            msg += f"\nDid you mean: {', '.join(suggestions)}?"
         return msg, 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"✅ Success with model {model_id}", (out["audio"], "output.wav")
     except Exception as e:
         return f"❌ Pipeline error: {str(e)}", None
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🌍 MMS-TTS: Speak over 1100 Languages\n(Just type the name, no ISO code needed!)")
+    gr.Markdown("## 🌍 MMS-TTS (1100+ Languages)\nJust type the language name, no ISO needed!")
 
     with gr.Row():
-        lang_box = gr.Textbox(label="Language (e.g., English, Meadow Mari)")
-        text_box = gr.Textbox(label="Text")
+        lang_input = gr.Textbox(label="Language name (e.g. English, Meadow Mari)")
+        text_input = gr.Textbox(label="Text")
 
     with gr.Row():
-        btn = gr.Button("🔊 Speak")
+        speak_button = gr.Button("🔊 Speak")
 
     with gr.Row():
-        out_msg = gr.Textbox(label="Output Message")
-        out_audio = gr.Audio(label="Audio", type="filepath")
+        result_text = gr.Textbox(label="Info")
+        result_audio = gr.Audio(label="Audio", type="filepath")
 
-    btn.click(fn=tts_fn, inputs=[text_box, lang_box], outputs=[out_msg, out_audio])
+    speak_button.click(fn=tts_fn, inputs=[text_input, lang_input], outputs=[result_text, result_audio])
 
 demo.launch()