Update app.py

app.py

@@ -7,9 +7,6 @@ from io import BytesIO
 
 # Define the language options
 languages = {
-    'es': 'Spanish',
-    'fr': 'French',
-    'de': 'German',
     'ur': 'Urdu',
     'hi': 'Hindi',
     'bn': 'Bengali'
@@ -19,21 +16,20 @@ languages = {
 st.title("Real-Time Language Translator and TTS")
 
 # Text input
-text_to_translate = st.text_area("Enter text to translate:", "Hello, how are you?")
+text_to_translate = st.text_area("Enter text to translate:", "My Name is Meer Hassan Farhad, I live in Islamabad")
 
 # Language selection
 target_language = st.selectbox("Select target language:", list(languages.keys()))
 
 def translate_text(text, target_lang):
-    # Update the model and tokenizer based on the selected language
+    # Use a more specific model for translation
     model_name = f"Helsinki-NLP/opus-mt-en-{target_lang}"
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
     
     # Translate text
     encoded_text = tokenizer.encode(text, return_tensors="pt")
-    forced_bos_token_id = model.config.forced_bos_token_id
-    translated = model.generate(encoded_text, forced_bos_token_id=forced_bos_token_id)
+    translated = model.generate(encoded_text)
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
@@ -65,8 +61,11 @@ if st.button("Translate and Synthesize Speech"):
     st.write(f"Translated text ({languages[target_language]}): {translated_text}")
     
     # Perform text-to-speech
-    audio_bytes = synthesize_speech(translated_text)
-    st.audio(audio_bytes, format="audio/wav")
+    try:
+        audio_bytes = synthesize_speech(translated_text)
+        st.audio(audio_bytes, format="audio/wav")
+    except Exception as e:
+        st.error(f"Error generating audio: {e}")