Update app.py

app.py

@@ -23,7 +23,8 @@ def main():
     if 'translated_text' not in st.session_state:
         st.session_state.translated_text = None
         st.session_state.audio_path = None
-        st.session_state.source_lang = "English"
+        st.session_state.source_lang = None
+        st.session_state.detected_options = []
 
     with tab1:
         # Text input
@@ -49,10 +50,25 @@ def main():
 
 # Handle input processing
 def handle_input(text, translation, lang_detect, audio_processor):
-    # Auto-detect source language
-    detected_lang = lang_detect.detect_language(text)
-    st.session_state.source_lang = detected_lang if detected_lang in translation.LANGUAGES else "English"
-    st.info(f"Detected Source Language: {st.session_state.source_lang}")
+    # 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
+    else:
+        detected_options = lang_detect.detect_language(text)
+    st.session_state.detected_options = detected_options
+
+    # 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")
+    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))
+
+    st.info(f"Selected Source Language: {st.session_state.source_lang}")
 
     # Target language selection
     target_lang = st.selectbox("Target Language", list(translation.LANGUAGES.keys()), index=1, key="target_lang")
@@ -79,6 +95,8 @@ def handle_input(text, translation, lang_detect, audio_processor):
                     st.session_state.audio_path = audio_processor.text_to_speech(st.session_state.translated_text, target_lang)
                     if st.session_state.audio_path:
                         st.audio(st.session_state.audio_path)
+                    else:
+                        st.error("Failed to generate audio. Please try again or check the input.")
                 
                 # Show footer after translation
                 st.markdown("""