Update app.py

app.py

@@ -1,27 +1,41 @@
 import streamlit as st
-from transformers import pipeline
-
-# Initialize the translation pipeline from Hugging Face
-@st.cache_resource
-def load_translation_model():
-    return pipeline("translation_en_to_ur", model="Helsinki-NLP/opus-mt-en-ur")
-
-translator = load_translation_model()
-
-# Streamlit app layout
-st.title("English to Urdu Translator")
-
-# Input text in English
-english_text = st.text_area("Enter English text", placeholder="Type something in English...", height=150)
-
-# Translate button
-if st.button("Translate to Urdu"):
-    if english_text.strip() != "":
-        translation = translator(english_text)
-        urdu_text = translation[0]['translation_text']
-        st.write("### Translated Text in Urdu:")
-        st.write(urdu_text)
-    else:
-        st.warning("Please enter some text to translate!")
+from transformers import MarianMTModel, MarianTokenizer
+
+# Supported languages
+supported_languages = {
+    'English to Urdu': 'Helsinki-NLP/opus-mt-en-ur',
+    'English to French': 'Helsinki-NLP/opus-mt-en-fr',
+    'English to Spanish': 'Helsinki-NLP/opus-mt-en-es',
+    'French to English': 'Helsinki-NLP/opus-mt-fr-en',
+    'Spanish to English': 'Helsinki-NLP/opus-mt-es-en',
+    'Urdu to English': 'Helsinki-NLP/opus-mt-ur-en'
+}
+
+# Streamlit App Title
+st.title("Multi-Language Translator")
+
+# Input Text
+input_text = st.text_area("Enter the text you want to translate:")
+
+# Language selection dropdown
+language_pair = st.selectbox("Choose the language pair:", list(supported_languages.keys()))
+
+# Load the model and tokenizer for the selected language pair
+if input_text:
+    model_name = supported_languages[language_pair]
+    model = MarianMTModel.from_pretrained(model_name)
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+
+    # Tokenize and translate
+    translated = model.generate(**tokenizer(input_text, return_tensors="pt", padding=True))
+    translated_text = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
+
+    # Display the translated text
+    st.subheader("Translated Text:")
+    st.write(translated_text[0])
+
+# Footer
+st.markdown("Powered by Helsinki-NLP/Opus-MT and Hugging Face")
+