Update app.py

app.py

@@ -31,6 +31,13 @@ def extract_text_from_file(file):
     except Exception:
         return ""
 
+def on_file_upload():
+    uploaded_file = st.session_state.file_input
+    if uploaded_file and uploaded_file.size < 1024*1024:
+        st.session_state.input_text = extract_text_from_file(uploaded_file)
+    elif uploaded_file and uploaded_file.size >= 1024*1024:
+        st.error("File size must be less than 1 MB")
+
 def main():
     try:
         translation = importlib.import_module("translation")
@@ -48,23 +55,21 @@ def main():
             source_lang = detected_options[0][2] if detected_options[0][0] != "Auto-detect" else "Auto-detect"
             source_lang_code = next((k for k, v in LANGUAGES.items() if v[1] == source_lang), "hi") if source_lang != "Auto-detect" else "auto"
             source_options = ["Auto-detect"] + [f"{v[0]} ({v[1]})" for v in LANGUAGES.values()]
-            st.selectbox("Source", options=source_options, index=0 if source_lang == "Auto-detect" else source_options.index(f"{LANGUAGES[source_lang_code][0]} ({source_lang})"), key="source_lang")
-            input_text = st.text_area("", height=300, key="input_text", label_visibility="hidden")
-            input_type = st.radio("", ["Text", "File"], horizontal=True, label_visibility="hidden")
+            st.selectbox("Source", options=source_options, index=0 if source_lang == "Auto-detect" else source_options.index(f"{LANGUAGES[source_lang_code][0]} ({source_lang})"), key="source_lang", label="Source Language")
+            input_text = st.text_area("Input Text", height=300, key="input_text", label_visibility="hidden")
+            input_type = st.radio("Input Type", ["Text", "File"], horizontal=True, label_visibility="hidden", key="input_type")
             if input_type == "File":
-                uploaded_file = st.file_uploader("", type=["txt", "docx", "pdf"], key="file_input", label_visibility="hidden")
-                if uploaded_file and uploaded_file.size < 1024*1024:  # 1 MB limit
-                    st.session_state.input_text = extract_text_from_file(uploaded_file)
-                elif uploaded_file and uploaded_file.size >= 1024*1024:
+                st.file_uploader("Upload File", type=["txt", "docx", "pdf"], key="file_input", on_change=on_file_upload, label_visibility="hidden")
+                if st.session_state.get("file_input") and st.session_state.get("file_input").size >= 1024*1024:
                     st.error("File size must be less than 1 MB")
-            st.button("Translate", key="translate_btn", on_click=trigger_translation, args=(translation, lang_detect, audio_processor,))
+            st.button("Translate", key="translate_btn", on_click=trigger_translation, args=(translation, lang_detect, audio_processor), label="Translate Button")
         with col2:
             source_lang_display = st.session_state.source_lang.split(" (")[0] if " (" in st.session_state.source_lang else st.session_state.source_lang
             target_options = [f"{v[0]} ({v[1]})" for v in LANGUAGES.values() if v[0] != source_lang_display and v[1] != source_lang_display]
-            st.selectbox("Target", options=target_options, index=target_options.index(f"{LANGUAGES['en'][0]} ({LANGUAGES['en'][1]})") if "English" not in source_lang_display else 0, key="target_lang")
+            st.selectbox("Target", options=target_options, index=target_options.index(f"{LANGUAGES['en'][0]} ({LANGUAGES['en'][1]})") if "English" not in source_lang_display else 0, key="target_lang", label="Target Language")
             if "translated_text" in st.session_state:
-                st.text_area("", value=st.session_state.translated_text, height=300, key="output_text", disabled=True, label_visibility="hidden")
-                if st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor,), help="Play audio", use_container_width=False):
+                st.text_area("Output Text", value=st.session_state.translated_text, height=300, key="output_text", disabled=True, label_visibility="hidden")
+                if st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor,), help="Play audio", use_container_width=False, label="Play Audio"):
                     pass
         # Footer
         if "translated_text" in st.session_state:
@@ -92,7 +97,7 @@ def play_audio(audio_processor):
         target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang.split(" (")[0]), "en")
         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")
+            st.audio(audio, format="audio/mp3", key=f"audio_{id(st.session_state)}")
 
 if __name__ == "__main__":
     main()