Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
-from gtts import gTTS
-import os
+import pyttsx3
+import time
 
-# Install SentencePiece if not already installed
+# Install required libraries if not already installed
 try:
     import sentencepiece
 except ImportError:
@@ -11,11 +11,8 @@ except ImportError:
     os.system("pip install sentencepiece")
     import sentencepiece
 
-# Streamlit app
-st.title("Text Translator with Voice")
-
-# Input text
-user_text = st.text_input("Enter the text you want to translate:")
+# Streamlit app title
+st.title("🌍 Advanced Text Translator with Voice �")
 
 # Define a dictionary of language codes and their full names
 language_names = {
@@ -70,28 +67,43 @@ def translate_text(text, target_language_code):
     except Exception as e:
         return f"Translation failed: {str(e)}"
 
-# Target language selection
+# Function to initialize the TTS engine
+def init_tts_engine():
+    engine = pyttsx3.init()
+    engine.setProperty('rate', 150)  # Speed of speech
+    engine.setProperty('volume', 1.0)  # Volume level (0.0 to 1.0)
+    return engine
+
+# Streamlit UI
+st.subheader("Enter the text you want to translate:")
+user_text = st.text_input("Input text:", placeholder="Type your text here...")
+
 st.subheader("Select the target language for translation:")
-selected_target_language = st.selectbox("Select a target language:", list(language_names.keys()))
+selected_target_language = st.selectbox("Target language:", list(language_names.keys()))
 
-# Display the translated text and speak it
+# Translate and speak the text
 if user_text:
-    translated_text = translate_text(user_text, language_names[selected_target_language])
+    with st.spinner("Translating text..."):
+        translated_text = translate_text(user_text, language_names[selected_target_language])
     
     if "Translation failed" not in translated_text:
-        # Apply advanced CSS for better display in a bordered box
-        styled_text = f'<div style="border: 2px solid #0072B5; padding: 10px; background-color: #E5E5E5; border-radius: 5px; color: black;">{translated_text}</div>'
-        st.markdown(styled_text, unsafe_allow_html=True)
-        
-        # Text-to-Speech (TTS) conversion and playback
-        tts = gTTS(translated_text, lang=language_names[selected_target_language])
-        tts_file_path = "translated_audio.mp3"
-        tts.save(tts_file_path)
-        
-        # Play the TTS audio
-        st.audio(tts_file_path, format="audio/mp3")
+        # Display translated text in a styled box
+        st.success("Translation successful! 🎉")
+        st.markdown(
+            f'<div style="border: 2px solid #0072B5; padding: 10px; background-color: #E5E5E5; border-radius: 5px; color: black;">'
+            f'<b>Translated Text:</b> {translated_text}'
+            f'</div>',
+            unsafe_allow_html=True
+        )
         
-        # Clean up the TTS audio file
-        os.remove(tts_file_path)
+        # Convert translated text to speech
+        with st.spinner("Generating audio..."):
+            try:
+                engine = init_tts_engine()
+                engine.say(translated_text)
+                engine.runAndWait()
+                st.success("Audio playback complete! 🔊")
+            except Exception as e:
+                st.error(f"Text-to-speech failed: {str(e)}")
     else:
         st.error(translated_text)