Update app.py

app.py

@@ -27,18 +27,18 @@ def main():
             detected_options = lang_detect.detect_language(st.session_state.get("input_text", "")) if st.session_state.get("input_text", "").strip() else [("Auto-detect", 1.0, "Auto-detect")]
             source_lang = detected_options[0][0] if detected_options[0][0] != "Auto-detect" else "Auto-detect"
             source_lang_code = next((k for k, v in LANGUAGES.items() if v[0] == source_lang), "hi") if source_lang != "Auto-detect" else "auto"
-            source_options = ["Auto-detect"] + [v[0] for v in LANGUAGES.values()]
-            st.selectbox("Source", options=source_options, index=0 if source_lang == "Auto-detect" else source_options.index(source_lang), key="source_lang")
+            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"{source_lang} ({LANGUAGES[next((k for k, v in LANGUAGES.items() if v[0] == source_lang), 'hi')][1]})"), key="source_lang")
             input_type = st.radio("", ["Text", "File"], horizontal=True, label_visibility="hidden")
             if input_type == "Text":
                 st.text_area("", height=200, key="input_text", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,), label_visibility="hidden")
             else:
-                uploaded_file = st.file_uploader("", type=["txt", "docx", "pdf"], key="file_input", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,), label_visibility="hidden")
-                if uploaded_file:
+                uploaded_file = st.file_uploader("", type=["txt", "docx", "pdf"], key="file_input", on_change=trigger_translation, args=(translation, lang_detect, audio_processor,), label_visibility="hidden", max_size=1024*1024)  # 1 MB limit
+                if uploaded_file and uploaded_file.size <= 1024*1024:
                     st.session_state.input_text = uploaded_file.read().decode("utf-8").strip()
             st.button("Translate", key="translate_btn", on_click=trigger_translation, args=(translation, lang_detect, audio_processor,))
         with col2:
-            st.selectbox("Target", options=[v[0] for v in LANGUAGES.values()], index=list(LANGUAGES.values()).index(LANGUAGES["en"]), key="target_lang")
+            st.selectbox("Target", options=[f"{v[0]} ({v[1]})" for v in LANGUAGES.values()], index=list(LANGUAGES.values()).index(LANGUAGES["en"]), key="target_lang")
             if "translated_text" in st.session_state:
                 st.text_area("", value=st.session_state.translated_text, height=200, key="output_text", disabled=True, label_visibility="hidden")
                 st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor,), help="Play audio", use_container_width=False)
@@ -50,8 +50,8 @@ def main():
 def trigger_translation(translation, lang_detect, audio_processor):
     text = st.session_state.get("input_text", "").strip()
     if text:
-        source_lang = st.session_state.source_lang
-        target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang), "en")
+        source_lang = st.session_state.source_lang.split(" (")[0] if " (" in st.session_state.source_lang else st.session_state.source_lang
+        target_lang = next((k for k, v in LANGUAGES.items() if v[0] == st.session_state.target_lang.split(" (")[0]), "en")
         if source_lang == "Auto-detect":
             detected_options = lang_detect.detect_language(text)
             source_lang_code = next((k for k, v in LANGUAGES.items() if v[1] == detected_options[0][0]), "hi")
@@ -61,7 +61,7 @@ def trigger_translation(translation, lang_detect, audio_processor):
 
 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), "en")
+        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")