Update app.py

app.py

@@ -1,50 +1,45 @@
 import streamlit as st
-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
+from transformers import pipeline
+
+# Title of the app
+st.title("🌐 Language Translator")
+
+# Load translation pipeline
+@st.cache_resource
+def load_translation_pipeline(language_code):
+    model_name = f"Helsinki-NLP/opus-mt-en-{language_code}"
+    return pipeline("translation", model=model_name)
+
+# Input text
+text_to_translate = st.text_area("Enter text in English to translate:")
+
+# Target language selection
+languages = {
+    "German": "de",
+    "French": "fr",
+    "Spanish": "es",
+    "Italian": "it",
+    "Chinese": "zh",
+    "Hindi": "hi",
 }
-
-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()
+target_language = st.selectbox("Select target language:", options=list(languages.keys()))
+
+# Translation action
+if st.button("Translate"):
+    if not text_to_translate.strip():
+        st.warning("Please enter some text to translate.")
+    else:
+        # Get the language code for the selected target language
+        language_code = languages[target_language]
+        
+        # Load the translation model
+        translator = load_translation_pipeline(language_code)
+        
+        # Perform translation
+        with st.spinner("Translating..."):
+            try:
+                result = translator(text_to_translate)[0]['translation_text']
+                st.success("Translated Text:")
+                st.write(result)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")