Input updated for app.py

app.py

@@ -1,45 +1,105 @@
 import streamlit as st
-from translation import translate, LANGUAGES
+from translation import translate
+from lang_detect import detect_language
+from audio_processor import transcribe_audio, text_to_speech
+import os
 
-# Main function for the app
+# Set page config for better UI
+st.set_page_config(page_title="Multilingual Translator", page_icon="🌐", layout="centered")
+
+# Main app function
 def main():
-    # Set page title and description
-    st.title("Multilingual Translation App")
-    st.write("Translate text for multilingual support (e.g., chatbots or customer service).")
+    # 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)
+
+    # Tabs for input methods
+    tab1, tab2, tab3 = st.tabs(["Text Input", "Audio Input", "Document Upload"])
+
+    # Initialize session state for results
+    if 'translated_text' not in st.session_state:
+        st.session_state.translated_text = None
+        st.session_state.audio_path = None
+        st.session_state.source_lang = "English"
+
+    with tab1:
+        # 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)
 
-    # Dropdowns for selecting languages
-    source_lang = st.selectbox("Source Language", list(LANGUAGES.keys()), index=0)
-    target_lang = st.selectbox("Target Language", list(LANGUAGES.keys()), index=1)
-    
-    # Text input area
-    user_input = st.text_area("Input Text", placeholder="Enter text to translate...", height=150)
+    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)
+            st.write(f"Transcribed Text: {user_text}")
+            handle_input(user_text)
+
+    with tab3:
+        # Document input
+        doc_file = st.file_uploader("Upload Document (TXT)", type=["txt"], key="doc_input")
+        if doc_file:
+            user_text = doc_file.read().decode("utf-8")
+            st.write(f"Document Text: {user_text}")
+            handle_input(user_text)
+
+# Handle input processing
+def handle_input(text):
+    # Auto-detect source language
+    detected_lang = detect_language(text)
+    st.session_state.source_lang = detected_lang if detected_lang in 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")
 
     # Translate button
-    if st.button("Translate"):
-        if user_input:
-            # Show a spinner while translating
-            with st.spinner("Translating..."):
-                try:
-                    # Call the translate function from translation.py
-                    result = translate(user_input, source_lang, target_lang)
+    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)
+                
+                # Display results in a styled container
+                st.markdown("<h3 style='color: #2E86C1;'>Translation Result</h3>", unsafe_allow_html=True)
+                
+                # Output options: Text and Audio
+                output_option = st.radio("Output Format", ["Text", "Audio"], key="output_option")
+                
+                if output_option == "Text":
                     st.success("Translated Text:")
-                    st.write(result)
-                    
-                    # Show footer only after translation
-                    st.markdown("""
-                        <p style="font-size: small; color: grey; text-align: center;">
-                            Developed By: Krishna Prakash
-                            <a href="https://www.linkedin.com/in/krishnaprakash-profile/" target="_blank">
-                                <img src="https://img.icons8.com/ios-filled/30/0077b5/linkedin.png" alt="LinkedIn" style="vertical-align: middle; margin: 0 5px;"/>
-                            </a>
-                        </p>
-                    """, unsafe_allow_html=True)
-                except Exception as e:
-                    # Show error if translation fails
-                    st.error(f"Translation failed: {str(e)}")
-        else:
-            # Warn if no text is entered
-            st.warning("Please enter text to translate!")
+                    st.write(st.session_state.translated_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.audio(st.session_state.audio_path)
+                
+                # Show footer after translation
+                st.markdown("""
+                    <p style='font-size: small; color: grey; text-align: center; margin-top: 20px;'>
+                        Developed By: Krishna Prakash
+                        <a href='https://www.linkedin.com/in/krishnaprakash-profile/' target='_blank'>
+                            <img src='https://img.icons8.com/ios-filled/30/0077b5/linkedin.png' alt='LinkedIn' style='vertical-align: middle; margin: 0 5px;'/>
+                        </a>
+                    </p>
+                """, unsafe_allow_html=True)
+            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__":