Update app.py

app.py

@@ -1,25 +1,20 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
-# Define available languages with MarianMT models
-LANGUAGES = {
-    'Spanish': 'es',
-    'French': 'fr',
-    'German': 'de',
-    'Chinese': 'zh',
-    'Hindi': 'hi',
-    'Arabic': 'ar',
-    'Japanese': 'ja',
-    'Russian': 'ru',
-    'Italian': 'it',
-    'Portuguese': 'pt',
-    # Add more languages if needed
+# Define available target languages and their Hugging Face model names
+language_options = {
+    "French": "Helsinki-NLP/opus-mt-en-fr",
+    "German": "Helsinki-NLP/opus-mt-en-de",
+    "Spanish": "Helsinki-NLP/opus-mt-en-es",
+    "Italian": "Helsinki-NLP/opus-mt-en-it",
+    "Dutch": "Helsinki-NLP/opus-mt-en-nl",
+    "Urdu": "Helsinki-NLP/opus-mt-en-ur",  # Urdu language model
+    # Add more languages as needed
 }
 
 # Function to load the model based on the selected language
 @st.cache_resource
-def load_model(src_lang='en', tgt_lang='es'):
-    model_name = f'Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}'
+def load_model(model_name):
     model = MarianMTModel.from_pretrained(model_name)
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     return model, tokenizer
@@ -37,7 +32,7 @@ st.markdown("<p style='text-align: center; color: #333; font-size: 18px;'>Transl
 
 # Sidebar for language selection and instructions
 st.sidebar.title("Language Options")
-language = st.sidebar.selectbox("Choose target language", list(LANGUAGES.keys()))
+language = st.sidebar.selectbox("Choose target language", list(language_options.keys()))
 
 st.sidebar.markdown("### How to use")
 st.sidebar.write("1. Enter English text in the box below.")
@@ -56,36 +51,17 @@ if st.button("Translate"):
         # Show a spinner during the translation process
         with st.spinner('Translating...'):
             # Load model and tokenizer based on selected language
-            tgt_lang = LANGUAGES[language]
-            model, tokenizer = load_model('en', tgt_lang)
+            model_name = language_options[language]
+            model, tokenizer = load_model(model_name)
             
             # Perform translation
             translated_text = translate_text(model, tokenizer, text)
         
         # Display the translation
-        st.markdown("<h3 style='color: #4B9CD3;'>Translated Text:</h3>", unsafe_allow_html=True)
+        st.markdown(f"<h3 style='color: #4B9CD3;'>Translated Text ({language}):</h3>", unsafe_allow_html=True)
         st.success(translated_text)
     else:
         st.error("Please enter text to translate.")
 
-# Footer with styling
-st.markdown(
-    """
-    <style>
-    footer {visibility: hidden;}
-    .footer-text {
-        position: fixed;
-        bottom: 0;
-        left: 0;
-        right: 0;
-        background-color: #f9f9f9;
-        padding: 10px;
-        text-align: center;
-        color: #4B9CD3;
-        font-weight: bold;
-    }
-    </style>
-    <div class="footer-text">Powered by Hugging Face Transformers</div>
-    """,
     unsafe_allow_html=True
 )