Update app.py

app.py

@@ -52,7 +52,7 @@ def main():
 def handle_input(text, translation, lang_detect, audio_processor):
     # Auto-detect source language with options
     if len(text) < 10:  # Minimum length for reliable detection
-        detected_options = [("English", 1.0)]  # Default to English with full confidence
+        detected_options = [("English", 1.0, "English")]
     else:
         detected_options = lang_detect.detect_language(text)
     st.session_state.detected_options = detected_options
@@ -60,18 +60,36 @@ def handle_input(text, translation, lang_detect, audio_processor):
     # Set initial source language to the top detected option
     st.session_state.source_lang = detected_options[0][0] if detected_options else "English"
 
-    # Manual language override with detected options
-    override_options = ["Auto"] + [lang for lang, _ in detected_options] + list(translation.LANGUAGES.keys())
-    override_lang = st.selectbox("Override Detected Language", override_options, index=0, key="override_lang")
+    # Display native language names for detected options
+    native_lang_map = {
+        "English": "English",
+        "French": "Français",
+        "Spanish": "Español",
+        "German": "Deutsch",
+        "Hindi": "हिन्दी",
+        "Chinese": "中文",
+        "Arabic": "العربية",
+        "Russian": "Русский",
+        "Japanese": "日本語",
+    }
+    display_options = ["Auto"] + [f"{native_lang_map.get(lang, lang)} ({conf:.2f})" for lang, conf, _ in detected_options] + list(translation.LANGUAGES.keys())
+    override_lang = st.selectbox("Override Detected Language", display_options, index=0, key="override_lang")
+
+    # Map override selection to supported language
     if override_lang != "Auto":
-        st.session_state.source_lang = override_lang
-    elif len(detected_options) > 1:
-        st.write("Multiple detections found (confidence > 0.7):", ", ".join(f"{lang} ({conf:.2f})" for lang, conf in detected_options))
+        selected_lang = next((lang for lang, _, _ in detected_options if native_lang_map.get(lang, lang) in override_lang), override_lang)
+        st.session_state.source_lang = selected_lang
+        if selected_lang not in translation.LANGUAGES:
+            st.session_state.source_lang = "English"
+            st.warning(f"{native_lang_map.get(selected_lang, selected_lang)} is not supported by the translation model. Using English instead.")
 
-    st.info(f"Selected Source Language: {st.session_state.source_lang}")
+    st.info(f"Selected Source Language: {native_lang_map.get(st.session_state.source_lang, st.session_state.source_lang)}")
 
-    # Target language selection
-    target_lang = st.selectbox("Target Language", list(translation.LANGUAGES.keys()), index=1, key="target_lang")
+    # Target language selection with native names
+    target_options = {lang: native_lang_map.get(lang, lang) for lang in translation.LANGUAGES}
+    target_lang = st.selectbox("Target Language", [target_options[lang] for lang in translation.LANGUAGES],
+                              index=1, key="target_lang", format_func=lambda x: x)
+    target_lang = next(key for key, value in target_options.items() if value == target_lang)
 
     # Translate button
     if st.button("Translate", key="translate_button"):