Update app.py

app.py

@@ -1,9 +1,16 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
-from datasets import load_dataset
 from gtts import gTTS
 import os
 
+# Install SentencePiece if not already installed
+try:
+    import sentencepiece
+except ImportError:
+    st.warning("The SentencePiece library is required but not installed. Installing now...")
+    os.system("pip install sentencepiece")
+    import sentencepiece
+
 # Streamlit app
 st.title("Text Translator with Voice")
 
@@ -31,11 +38,16 @@ language_names = {
 }
 
 # Function to load the translation model and tokenizer
+@st.cache_resource
 def load_translation_model(language_pair):
-    model_name = f'Helsinki-NLP/opus-mt-{language_pair}'
-    model = MarianMTModel.from_pretrained(model_name)
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    return model, tokenizer
+    try:
+        model_name = f'Helsinki-NLP/opus-mt-{language_pair}'
+        model = MarianMTModel.from_pretrained(model_name)
+        tokenizer = MarianTokenizer.from_pretrained(model_name)
+        return model, tokenizer
+    except Exception as e:
+        st.error(f"Failed to load model for language pair {language_pair}: {str(e)}")
+        return None, None
 
 # Function to translate text
 def translate_text(text, target_language_code):
@@ -44,6 +56,9 @@ def translate_text(text, target_language_code):
         language_pair = f'en-{target_language_code}'
         model, tokenizer = load_translation_model(language_pair)
         
+        if model is None or tokenizer is None:
+            return "Translation failed: Model or tokenizer not loaded."
+        
         # Tokenize the input text
         inputs = tokenizer(text, return_tensors="pt", padding=True)
         
@@ -63,17 +78,20 @@ selected_target_language = st.selectbox("Select a target language:", list(langua
 if user_text:
     translated_text = translate_text(user_text, language_names[selected_target_language])
     
-    # 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")
-    
-    # Clean up the TTS audio file
-    os.remove(tts_file_path)
+    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")
+        
+        # Clean up the TTS audio file
+        os.remove(tts_file_path)
+    else:
+        st.error(translated_text)