Update app.py

app.py

@@ -2,64 +2,42 @@ import gradio as gr
 from transformers import pipeline
 from huggingface_hub import list_models
 import langcodes
-import difflib
 
-# Map tên ngôn ngữ (không phân biệt hoa/thường) → mã ISO
+# Tự động lấy danh sách mô hình MMS-TTS
+all_models = list_models(filter="facebook/mms-tts", full=True)
 lang_map = {}
 
-models = list_models(filter="facebook/mms-tts", full=True)
-
-for model in models:
-    model_id = model.modelId  # ví dụ: facebook/mms-tts-kaz
-    code = model_id.split("-")[-1].lower()
-    
-    # Lấy tên ngôn ngữ từ metadata hoặc từ mã ISO
-    if model.cardData and "language" in model.cardData:
-        name = model.cardData["language"]
-    else:
-        try:
-            name = langcodes.Language.make(code).display_name()
-        except:
-            name = code.upper()
-
-    # Đưa vào bản đồ, key là lowercase để không phân biệt hoa/thường
-    lang_map[name.strip().lower()] = code
+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):
-    key = language.strip().lower()
-    code = lang_map.get(key)
-
+    code = lang_map.get(language.lower())
     if not code:
-        # 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"\nDid you mean: {', '.join(suggestions)}?"
-        return msg, None
-
+        return f"❌ Unsupported language name: '{language}'", None
     model_id = f"facebook/mms-tts-{code}"
-
     try:
         tts = pipeline("text-to-speech", model=model_id)
         out = tts(text)
-        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 (1100+ Languages)\nJust type the language name, no ISO needed!")
-
-    with gr.Row():
-        lang_input = gr.Textbox(label="Language name (e.g. English, Meadow Mari)")
-        text_input = gr.Textbox(label="Text")
-
-    with gr.Row():
-        speak_button = gr.Button("🔊 Speak")
-
-    with gr.Row():
-        result_text = gr.Textbox(label="Info")
-        result_audio = gr.Audio(label="Audio", type="filepath")
-
-    speak_button.click(fn=tts_fn, inputs=[text_input, lang_input], outputs=[result_text, result_audio])
-
-demo.launch()
+        return f"✅ Model: {model_id}", (out["audio"], "output.wav