Update app.py

app.py

@@ -13,36 +13,39 @@ except ImportError as e:
     st.error(f"Failed to import translation module: {e}")
     st.stop()
 
-def extract_text_from_file(file):
+# Function to extract text from uploaded files (PDF, DOCX, TXT)
+def extract_text_from_file(uploaded_file):
     try:
-        if file.type == "application/pdf":
-            pdf_reader = PdfReader(file)
+        if uploaded_file.type == "application/pdf":
+            pdf_reader = PdfReader(uploaded_file)
             text = ""
             for page in pdf_reader.pages:
                 text += page.extract_text() or ""
             return text.encode().decode('utf-8', errors='ignore').strip()
-        elif file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-            doc = docx.Document(file)
+        elif uploaded_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+            doc = docx.Document(uploaded_file)
             text = "\n".join([para.text for para in doc.paragraphs])
             return text.encode().decode('utf-8', errors='ignore').strip()
-        elif file.type == "text/plain":
-            return file.read().decode('utf-8', errors='ignore').strip()
+        elif uploaded_file.type == "text/plain":
+            return uploaded_file.read().decode('utf-8', errors='ignore').strip()
         return ""
     except Exception:
         return ""
 
+# Callback to update input text when file is uploaded
 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)
+        st.session_state.user_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")
 
+# Main application function
 def main():
     try:
-        translation = importlib.import_module("translation")
-        lang_detect = importlib.import_module("lang_detect")
-        audio_processor = importlib.import_module("audio_processor")
+        translation_module = importlib.import_module("translation")
+        language_detector = importlib.import_module("lang_detect")
+        audio_processor_module = importlib.import_module("audio_processor")
 
         # Header
         st.markdown("<h1 style='text-align: center; color: #4285F4;'>Multilingual Translator</h1>", unsafe_allow_html=True)
@@ -61,36 +64,34 @@ def main():
                 font-size: 0.8em;
             }
             .stFileUploader label::after {
-                content: "Limit 1MB per file • TXT, DOCX, PDF";
+                content: 'Limit 1MB per file • TXT, DOCX, PDF';
             }
             </style>
             """,
             unsafe_allow_html=True
         )
 
-        # Language and Input/Output Layout
-        col1, col2 = st.columns([2, 1])  # Wider input column
-        with col1:
-            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][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"
+        # Language and Input/Output Layout with symmetric columns
+        left_col, right_col = st.columns([1, 1])  # Equal width for symmetric layout
+        with left_col:
+            detected_options = language_detector.detect_language(st.session_state.get("user_input_text", "")) if st.session_state.get("user_input_text", "").strip() else [("Auto-detect", 1.0, "Auto-detect")]
+            source_language = 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_language), "hi") if source_language != "Auto-detect" else "auto"
             source_options = ["Auto-detect"] + [f"{v[0]} ({v[1]})" for v in LANGUAGES.values()]
-            st.selectbox("Source Language", 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("Input Text", height=400, key="input_text", placeholder="Enter text here", label_visibility="hidden")
+            st.selectbox("Source Language", options=source_options, index=0 if source_language == "Auto-detect" else source_options.index(f"{LANGUAGES[source_lang_code][0]} ({source_language})"), key="source_lang")
+            user_input_text = st.text_area("Input Text", height=400, key="user_input_text", placeholder="Enter text here", label_visibility="hidden")
             input_type = st.radio("Input Type", ["Text", "File"], horizontal=True, label_visibility="hidden", key="input_type")
             if input_type == "File":
                 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))
-        with col2:
+            st.button("Translate", key="translate_btn", on_click=trigger_translation, args=(translation_module, language_detector, audio_processor_module))
+        with right_col:
             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 Language", 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")
             if "translated_text" in st.session_state:
                 st.text_area("Output Text", value=st.session_state.translated_text, height=400, 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):
-                    pass
         # Footer
         if "translated_text" in st.session_state:
             st.markdown("""
@@ -101,30 +102,35 @@ def main():
                     </a>
                 </p>
             """, unsafe_allow_html=True)
+            # Play audio button and playback below output
+            if st.button("🔊", key="audio_btn", on_click=play_audio, args=(audio_processor_module,), help="Play audio", use_container_width=False):
+                pass
 
     except Exception as e:
         st.error(f"App error: {e}")
 
-def trigger_translation(translation, lang_detect, audio_processor):
-    text = st.session_state.get("input_text", "").strip()
+# Function to trigger translation process
+def trigger_translation(translation_module, language_detector, audio_processor_module):
+    text = st.session_state.get("user_input_text", "").strip()
     if text:
         source_lang = st.session_state.source_lang.split(" (")[0] if " (" in st.session_state.source_lang else st.session_state.source_lang
         target_lang = st.session_state.target_lang.split(" (")[0] if " (" in st.session_state.target_lang else st.session_state.target_lang
         if source_lang == "Auto-detect":
-            detected_options = lang_detect.detect_language(text)
+            detected_options = language_detector.detect_language(text)
             source_lang_code = next((k for k, v in LANGUAGES.items() if v[1] == detected_options[0][0]), "hi")
         else:
             source_lang_code = next((k for k, v in LANGUAGES.items() if v[0] == source_lang), "hi")
         target_lang_code = next((k for k, v in LANGUAGES.items() if v[0] == target_lang), "en")
-        translated_text = translation.translate(text, source_lang_code, target_lang_code)
+        translated_text = translation_module.translate(text, source_lang_code, target_lang_code)
         st.session_state.translated_text = translated_text or text
 
-def play_audio(audio_processor):
+# Function to handle audio playback
+def play_audio(audio_processor_module):
     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.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")
+        audio_data = audio_processor_module.text_to_speech(st.session_state.translated_text, target_lang)
+        if audio_data and audio_data.getbuffer().nbytes > 0:
+            st.audio(audio_data, format="audio/mp3")
 
 if __name__ == "__main__":
     main()