Update app.py

app.py

@@ -28,7 +28,7 @@ def main():
 
     with tab1:
         # Text input
-        user_text = st.text_area("Enter Text", placeholder="Type or paste your text here...", height=150, key="text_input")
+        user_text = st.text_area("Enter Text", placeholder="Type or paste your text here...", height=150, key="text_input").strip()
         if user_text:
             handle_input(user_text, "text", translation, lang_detect, audio_processor)
 
@@ -36,7 +36,7 @@ def main():
         # Audio input
         audio_file = st.file_uploader("Upload Audio (MP3/WAV)", type=["mp3", "wav"], key="audio_input")
         if audio_file:
-            user_text = audio_processor.transcribe_audio(audio_file)
+            user_text = audio_processor.transcribe_audio(audio_file).strip()
             st.write(f"Transcribed Text: {user_text}")
             handle_input(user_text, "audio", translation, lang_detect, audio_processor)
 
@@ -44,8 +44,8 @@ def main():
         # Document input
         doc_file = st.file_uploader("Upload Document (TXT)", type=["txt"], key="doc_input")
         if doc_file:
-            user_text = doc_file.read().decode("utf-8")
-            st.write(f"Document Text: {user_text}")
+            user_text = doc_file.read().decode("utf-8").strip()  # Explicit decoding and stripping
+            st.write(f"Raw Document Text: {user_text}")  # Debug raw input
             handle_input(user_text, "doc", translation, lang_detect, audio_processor)
 
 # Handle input processing
@@ -79,14 +79,14 @@ def handle_input(text, input_type, translation, lang_detect, audio_processor):
     if override_lang != "Auto":
         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
-        # No strict pair check, just warn if model fails later
 
     st.info(f"Selected Source Language: {native_lang_map.get(st.session_state.source_lang, st.session_state.source_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=f"target_lang_{input_type}", format_func=lambda x: x)
+                              index=list(translation.LANGUAGES.keys()).index("Hindi") if "Hindi" in translation.LANGUAGES else 1,
+                              key=f"target_lang_{input_type}", format_func=lambda x: x)
     target_lang = next(key for key, value in target_options.items() if value == target_lang)
 
     # Translate button
@@ -95,7 +95,7 @@ def handle_input(text, input_type, translation, lang_detect, audio_processor):
             try:
                 # Translate the text
                 st.session_state.translated_text = translation.translate(text, st.session_state.source_lang, target_lang)
-                
+                st.write(f"Translated Text (Debug): {st.session_state.translated_text}")  # Debug output
                 # Display results in a styled container
                 st.markdown("<h3 style='color: #2E86C1;'>Translation Result</h3>", unsafe_allow_html=True)