Update app.py

app.py

@@ -1,14 +1,17 @@
 import streamlit as st
-from translation import translate
-from lang_detect import detect_language
-from audio_processor import transcribe_audio, text_to_speech
 import os
+import importlib
 
-# Set page config as the first Streamlit command
+# Set page config as the absolute first Streamlit command
 st.set_page_config(page_title="Multilingual Translator", page_icon="🌐", layout="centered")
 
 # Main app function
 def main():
+    # Lazy import modules to avoid Streamlit initialization during import
+    translation = importlib.import_module("translation")
+    lang_detect = importlib.import_module("lang_detect")
+    audio_processor = importlib.import_module("audio_processor")
+
     # Title and header with styling
     st.markdown("<h1 style='text-align: center; color: #2E86C1;'>Multilingual Translator</h1>", unsafe_allow_html=True)
     st.markdown("<p style='text-align: center; color: #666;'>Translate text or audio for multilingual support.</p>", unsafe_allow_html=True)
@@ -26,15 +29,15 @@ def main():
         # Text input
         user_text = st.text_area("Enter Text", placeholder="Type or paste your text here...", height=150, key="text_input")
         if user_text:
-            handle_input(user_text)
+            handle_input(user_text, translation, lang_detect, audio_processor)
 
     with tab2:
         # Audio input
         audio_file = st.file_uploader("Upload Audio (MP3/WAV)", type=["mp3", "wav"], key="audio_input")
         if audio_file:
-            user_text = transcribe_audio(audio_file)
+            user_text = audio_processor.transcribe_audio(audio_file)
             st.write(f"Transcribed Text: {user_text}")
-            handle_input(user_text)
+            handle_input(user_text, translation, lang_detect, audio_processor)
 
     with tab3:
         # Document input
@@ -42,24 +45,24 @@ def main():
         if doc_file:
             user_text = doc_file.read().decode("utf-8")
             st.write(f"Document Text: {user_text}")
-            handle_input(user_text)
+            handle_input(user_text, translation, lang_detect, audio_processor)
 
 # Handle input processing
-def handle_input(text):
+def handle_input(text, translation, lang_detect, audio_processor):
     # Auto-detect source language
-    detected_lang = detect_language(text)
-    st.session_state.source_lang = detected_lang if detected_lang in LANGUAGES else "English"
+    detected_lang = lang_detect.detect_language(text)
+    st.session_state.source_lang = detected_lang if detected_lang in translation.LANGUAGES else "English"
     st.info(f"Detected Source Language: {st.session_state.source_lang}")
 
     # Target language selection
-    target_lang = st.selectbox("Target Language", list(LANGUAGES.keys()), index=1, key="target_lang")
+    target_lang = st.selectbox("Target Language", list(translation.LANGUAGES.keys()), index=1, key="target_lang")
 
     # Translate button
     if st.button("Translate", key="translate_button"):
         with st.spinner("Translating..."):
             try:
                 # Translate the text
-                st.session_state.translated_text = translate(text, st.session_state.source_lang, target_lang)
+                st.session_state.translated_text = translation.translate(text, st.session_state.source_lang, target_lang)
                 
                 # Display results in a styled container
                 st.markdown("<h3 style='color: #2E86C1;'>Translation Result</h3>", unsafe_allow_html=True)
@@ -73,7 +76,7 @@ def handle_input(text):
                 
                 elif output_option == "Audio":
                     st.success("Translated Audio:")
-                    st.session_state.audio_path = text_to_speech(st.session_state.translated_text, target_lang)
+                    st.session_state.audio_path = audio_processor.text_to_speech(st.session_state.translated_text, target_lang)
                     if st.session_state.audio_path:
                         st.audio(st.session_state.audio_path)
                 
@@ -89,19 +92,6 @@ def handle_input(text):
             except Exception as e:
                 st.error(f"Translation failed: {str(e)}")
 
-# Language dictionary for UI
-LANGUAGES = {
-    "English": "en",
-    "French": "fr",
-    "Spanish": "es",
-    "German": "de",
-    "Chinese": "zh",
-    "Arabic": "ar",
-    "Russian": "ru",
-    "Hindi": "hi",
-    "Japanese": "ja"
-}
-
 # Run the app
 if __name__ == "__main__":
     main()