Update app.py

app.py

@@ -1,18 +1,13 @@
-streamlit 
-transformers 
-torch
-sentencepiece
-sacremoses
-
 import streamlit as st
 from transformers import pipeline
 
-# Initialize the translation pipeline
+# Function to load translation pipeline
 @st.cache_resource
-def load_translator():
-    return pipeline("translation", model="Helsinki-NLP/opus-mt-en-{target}")
+def load_translator(target_language):
+    model_name = f"Helsinki-NLP/opus-mt-en-{target_language}"
+    return pipeline("translation", model=model_name)
 
-# Supported languages (ISO 639-1 codes mapped to language names)
+# Supported languages
 supported_languages = {
     "fr": "French",
     "es": "Spanish",
@@ -27,27 +22,29 @@ supported_languages = {
 }
 
 # Streamlit App
-st.title("Language Translator App")
-st.write("Translate text from English to a selected target language using Hugging Face models.")
+st.title("🌍 Language Translator App")
+st.write("Translate text from English to any supported language using Hugging Face models.")
 
-# Input text from user
+# User input for text
 input_text = st.text_area("Enter text in English:", placeholder="Type here...")
 
-# Language selection
+# Dropdown to select target language
 target_language = st.selectbox(
     "Select target language:",
     options=list(supported_languages.keys()),
-    format_func=lambda lang: supported_languages[lang],
+    format_func=lambda lang: f"{supported_languages[lang]} ({lang})",
 )
 
 # Translate button
 if st.button("Translate"):
-    if input_text.strip() == "":
-        st.error("Please enter text to translate.")
+    if not input_text.strip():
+        st.error("❌ Please enter some text to translate.")
     else:
-        translator = load_translator()
-        # Replace `{target}` with the user-selected language in the model
-        translator = pipeline("translation", model=f"Helsinki-NLP/opus-mt-en-{target_language}")
-        translation = translator(input_text)[0]["translation_text"]
-        st.success("Translated Text:")
-        st.write(translation)
+        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)}")