better translation and dropdown settings

app.py

@@ -151,6 +151,13 @@ def translate_text(input_text, source_lang="en", target_lang="en"):
         print(f"Attempted to use language code: {target_lang}")
         return input_text
 
+def detect_language(text):
+    """Detect the language of the input text using Google Translator."""
+    try:
+        return GoogleTranslator().detect(text).lower()
+    except:
+        return "en"  # Default to English if detection fails
+
 def download_and_extract_model(url, destination):
     """Download and extract the model files."""
     print(f"Downloading from URL: {url}")
@@ -422,9 +429,6 @@ def tts_interface(selected_model, text, translate_enabled, status_output):
             
         model_info = models[model_id]
         
-        # Check if this is an MMS model
-        is_mms = 'mms' in model_id.lower()
-        
         # Get the language code and check if translation is needed
         lang_code = get_language_code(model_info)
         translate_code = get_translate_code(lang_code)
@@ -434,24 +438,16 @@ def tts_interface(selected_model, text, translate_enabled, status_output):
         translated_text = None
         was_translated = False
         
-        # For MMS models, we always need to translate
-        if is_mms:
-            if not translate_code:
-                return None, f"Cannot determine translation target language from code: {lang_code}"
-            print(f"MMS model detected, translating to {translate_code}")
-            translated_text = translate_text(text, "en", translate_code)
-            if translated_text != text:  # Only mark as translated if text actually changed
-                was_translated = True
-            text = translated_text
-        # For other models, check if translation is enabled and needed
-        elif translate_enabled and translate_code and translate_code != "en":
+        # Only translate if translation is enabled and needed
+        if translate_enabled and translate_code and translate_code != "en":
             if not translate_code:
                 return None, f"Cannot determine translation target language from code: {lang_code}"
-            print(f"Will translate to {translate_code} (from ISO code {lang_code})")
+                
+            print(f"Translating to {translate_code}")
             translated_text = translate_text(text, "en", translate_code)
             if translated_text != text:  # Only mark as translated if text actually changed
                 was_translated = True
-            text = translated_text
+                text = translated_text
         
         try:
             # Update status with language info
@@ -471,7 +467,6 @@ def tts_interface(selected_model, text, translate_enabled, status_output):
                 final_status += f"\nText: '{text}'"
             
             return (sample_rate, audio_data), final_status
-            
         except ValueError as e:
             # Handle known errors with user-friendly messages
             error_msg = str(e)
@@ -517,29 +512,26 @@ with gr.Blocks() as app:
                 interactive=False
             )
     
-    # Handle model selection to update translate checkbox
-    def update_translate_checkbox(selected_model):
-        """Update visibility of translate checkbox based on selected model's language."""
-        try:
-            # Find the model info for the selected model
-            for lang_group in models_by_lang.values():
-                for display_name, model_id in lang_group:
-                    if display_name == selected_model:
-                        model_info = models[model_id]
-                        lang_info = model_info.get('language', [{}])[0]
-                        lang_code = lang_info.get('lang_code', '')
-                        return {"visible": lang_code != 'en'}
-            return {"visible": False}
-        except Exception as e:
-            print(f"Error updating translate checkbox: {str(e)}")
-            return {"visible": False}
+    def on_model_change(model_name):
+        # Get model info
+        model_id = models_by_display.get(model_name)
+        if not model_id or model_id not in models:
+            return False
+            
+        model_info = models[model_id]
+        lang_code = get_language_code(model_info)
+        
+        # Auto-check translation for non-English models
+        should_translate = lang_code and lang_code.lower() != "en"
+        return should_translate
 
+    # Update translation checkbox when model changes
     model_dropdown.change(
-        fn=update_translate_checkbox,
+        fn=on_model_change,
         inputs=[model_dropdown],
         outputs=[translate_checkbox]
     )
-            
+
     # Set up event handlers
     gen_event = generate_btn.click(
         fn=tts_interface,