Update app.py

app.py

@@ -18,79 +18,72 @@ def main():
     st.markdown("<p style='text-align: center; color: #666;'>Translate text like Google Translate</p>", unsafe_allow_html=True)
 
     # Language Controls
-    col1, col2 = st.columns([1, 1])
+    col1, col2 = st.columns(2)
     with col1:
-        detected_options = lang_detect.detect_language(st.session_state.get("user_text", "")) if st.session_state.get("user_text", "").strip() and len(st.session_state.get("user_text", "").strip()) >= 10 else [("English", 1.0, "English")]
+        detected_options = lang_detect.detect_language(st.session_state.get("input_text", "")) if st.session_state.get("input_text", "").strip() and len(st.session_state.get("input_text", "").strip()) >= 10 else [("English", 1.0, "English")]
         source_lang = detected_options[0][0] if detected_options else "English"
         source_lang_code = next((k for k, v in LANGUAGES.items() if v[1] == source_lang), "en")
-        source_lang_display = st.selectbox("Source Language", options=[LANGUAGES[source_lang_code][0]] + [v[0] for v in LANGUAGES.values()], index=0, key="source_lang", help="Auto-detected, override if needed", label_visibility="hidden")
+        source_lang_display = st.selectbox("Source", options=[LANGUAGES[source_lang_code][0]] + [v[0] for v in LANGUAGES.values()], index=0, key="source_lang")
     with col2:
-        target_lang_display = st.selectbox("Target Language", options=[v[0] for v in LANGUAGES.values()], index=list(LANGUAGES.values()).index(LANGUAGES["hi"]), key="target_lang", label_visibility="hidden")
+        target_lang_display = st.selectbox("Target", options=[v[0] for v in LANGUAGES.values()], index=list(LANGUAGES.values()).index(LANGUAGES["hi"]), key="target_lang")
         target_lang = next((k for k, v in LANGUAGES.items() if v[0] == target_lang_display), "hi")
 
     # Input Section
-    if "user_text" not in st.session_state:
-        st.session_state.user_text = ""
-    col_input = st.container()
-    with col_input:
-        input_type = st.radio("Input Method", ["Text", "File Upload"], horizontal=True, key="input_type", label_visibility="hidden")
+    if "input_text" not in st.session_state:
+        st.session_state.input_text = ""
+    with st.container():
+        input_type = st.radio("Input", ["Text", "File"], horizontal=True)
         if input_type == "Text":
-            user_text = st.text_area("Enter or paste text", height=200, key="user_text", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,), label_visibility="hidden")
+            input_text = st.text_area("Text to Translate", height=150, key="input_text", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,))
         else:
-            user_text = st.file_uploader("Upload File", type=["txt", "docx", "pdf"], key="doc_input", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,), label_visibility="hidden")
-            if user_text:
-                st.session_state.user_text = user_text.read().decode("utf-8").strip()
-        st.markdown(f"<small style='color: grey;'>Characters: {len(st.session_state.user_text)}/1000</small>", unsafe_allow_html=True)
+            input_text = st.file_uploader("Upload File", type=["txt", "docx", "pdf"], key="file_input", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,))
+            if input_text:
+                st.session_state.input_text = input_text.read().decode("utf-8").strip()
+        st.markdown(f"<small style='color: grey;'>Chars: {len(st.session_state.input_text)}/1000</small>", unsafe_allow_html=True)
         st.button("Translate", key="translate_btn", on_click=trigger_translation, args=(translation, lang_detect, audio_processor,))
 
     # Output Section
-    col_output = st.container()
-    with col_output:
+    with st.container():
         if "translated_text" in st.session_state and st.session_state.translated_text:
-            st.text_area("Translation", value=st.session_state.translated_text, height=200, key="output_area", disabled=True, label_visibility="hidden")
-            col_a = st.columns([1])[0]  # Single column for speaker icon
-            if st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor,), help="Listen to translation", label_visibility="hidden"):
-                pass  # Handled by play_audio function
+            st.text_area("Translation", value=st.session_state.translated_text, height=150, key="output_text", disabled=True)
+            if st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor,)):
+                pass  # Handled by play_audio
 
 def trigger_translation(translation, lang_detect, audio_processor):
-    user_text = st.session_state.user_text.strip()
-    if user_text:
-        spinner = st.empty()
-        spinner.info("Translating...")
-        start_time = time.time()
-        try:
-            source_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.source_lang), "en")
-            target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang), "hi")
-            if source_lang != "en" and target_lang != "en":
-                temp_translation = translation.translate(user_text, source_lang, "en")
-                if not temp_translation or len(temp_translation.split()) < 2:
-                    raise ValueError("Intermediate translation invalid")
-                translated_text = translation.translate(temp_translation, "en", target_lang)
-            else:
-                translated_text = translation.translate(user_text, source_lang, target_lang)
-            if not translated_text or len(translated_text.split()) < 2:
-                raise ValueError("Final translation invalid")
-            st.session_state.translated_text = translated_text
-        except Exception as e:
-            st.session_state.translated_text = user_text
-            st.warning(f"Translation issue: {str(e)}. Using input as fallback.")
-        finally:
-            spinner.empty()
+    text = st.session_state.input_text.strip()
+    if text:
+        with st.spinner("Translating..."):
+            try:
+                source_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.source_lang), "en")
+                target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang), "hi")
+                if source_lang != "en" and target_lang != "en":
+                    temp = translation.translate(text, source_lang, "en")
+                    if not temp or len(temp.split()) < 2:
+                        raise ValueError("Intermediate translation failed")
+                    translated = translation.translate(temp, "en", target_lang)
+                else:
+                    translated = translation.translate(text, source_lang, target_lang)
+                if not translated or len(translated.split()) < 2:
+                    raise ValueError("Translation failed")
+                st.session_state.translated_text = translated
+            except Exception as e:
+                st.session_state.translated_text = text
+                st.warning(f"Error: {str(e)}. Using input as fallback.")
 
 def play_audio(audio_processor):
     if "translated_text" in st.session_state and st.session_state.translated_text:
         target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang), "hi")
         audio = audio_processor.text_to_speech(st.session_state.translated_text, target_lang)
         if audio and audio.getbuffer().nbytes > 0:
-            st.audio(audio, format="audio/mp3", start_time=0)
-            st.success("Audio playing.")
+            st.audio(audio, format="audio/mp3")
+            st.success("Playing audio.")
         else:
             audio_fallback = audio_processor.text_to_speech(st.session_state.translated_text, "en")
             if audio_fallback and audio_fallback.getbuffer().nbytes > 0:
-                st.audio(audio_fallback, format="audio/mp3", start_time=0)
-                st.success("Fallback audio in English playing.")
+                st.audio(audio_fallback, format="audio/mp3")
+                st.success("Playing fallback audio in English.")
             else:
-                st.error("Audio generation failed. Try again later.")
+                st.error("Audio failed. Try again.")
 
 if __name__ == "__main__":
     main()