Update app.py

app.py

@@ -1,50 +1,50 @@
 import streamlit as st
-from transformers import pipeline
-
-# Function to load translation pipeline
-@st.cache_resource
-def load_translator(target_language):
-    model_name = f"Helsinki-NLP/opus-mt-en-{target_language}"
-    return pipeline("translation", model=model_name)
-
-# Supported languages
-supported_languages = {
-    "fr": "French",
-    "es": "Spanish",
-    "de": "German",
-    "zh": "Chinese",
-    "hi": "Hindi",
-    "ar": "Arabic",
-    "ru": "Russian",
-    "ja": "Japanese",
-    "ko": "Korean",
-    "it": "Italian",
+from transformers import MarianMTModel, MarianTokenizer
+
+# Define a dictionary to map language names to model identifiers
+models = {
+    'French': 'Helsinki-NLP/opus-mt-en-fr',
+    'Spanish': 'Helsinki-NLP/opus-mt-en-es',
+    'German': 'Helsinki-NLP/opus-mt-en-de',
+    'Italian': 'Helsinki-NLP/opus-mt-en-it',
+    'Urdu': 'Helsinki-NLP/opus-mt-en-ur',
+    'Arabic': 'Helsinki-NLP/opus-mt-en-ar', 
+    # Add more language models if needed
 }
 
-# Streamlit App
-st.title("🌍 Language Translator App")
-st.write("Translate text from English to any supported language using Hugging Face models.")
-
-# User input for text
-input_text = st.text_area("Enter text in English:", placeholder="Type here...")
-
-# Dropdown to select target language
-target_language = st.selectbox(
-    "Select target language:",
-    options=list(supported_languages.keys()),
-    format_func=lambda lang: f"{supported_languages[lang]} ({lang})",
-)
-
-# Translate button
-if st.button("Translate"):
-    if not input_text.strip():
-        st.error("❌ Please enter some text to translate.")
-    else:
-        try:
-            st.info("⏳ Translating...")
-            translator = load_translator(target_language)
-            translation = translator(input_text)[0]["translation_text"]
-            st.success("✅ Translated Text:")
-            st.write(translation)
-        except Exception as e:
-            st.error(f"⚠️ An error occurred: {str(e)}")
+def load_model(model_name):
+    """Load the model and tokenizer based on the selected model name."""
+    model = MarianMTModel.from_pretrained(model_name)
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    return model, tokenizer
+
+def translate_text(text, model, tokenizer):
+    """Translate text using the provided model and tokenizer."""
+    inputs = tokenizer.encode(text, return_tensors="pt")
+    translated = model.generate(inputs)
+    translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
+    return translated_text
+
+def main():
+    st.title("Multilingual Translator")
+
+    # User input for English text
+    text_to_translate = st.text_area("Enter text in English:")
+
+    # Language options
+    selected_language = st.selectbox("Select target language:", list(models.keys()))
+
+    if st.button("Translate"):
+        if text_to_translate:
+            # Load the selected model
+            model_name = models[selected_language]
+            model, tokenizer = load_model(model_name)
+            
+            translated_text = translate_text(text_to_translate, model, tokenizer)
+            st.write(f"**Translation in {selected_language}:**")
+            st.write(translated_text)
+        else:
+            st.warning("Please enter text to translate.")
+
+if __name__ == "__main__":
+    main()