Update app.py

app.py

@@ -1,14 +1,20 @@
 import streamlit as st
 from transformers import pipeline
 
-
 st.title("Multilingual Translator Chatbot")
 
 # Choose the translation model from Hugging Face
-translation_model = st.selectbox("Select translation model", ["Helsinki-NLP/opus-mt-en-de", "Helsinki-NLP/opus-mt-en-fr"])
+translation_models = {
+    "English to German": "Helsinki-NLP/opus-mt-en-de",
+    "English to French": "Helsinki-NLP/opus-mt-en-fr",
+    "English to Urdu": "Helsinki-NLP/opus-mt-en-ur",
+    # Add more language pairs as needed
+}
+
+selected_translation = st.selectbox("Select translation model", list(translation_models.keys()))
 
 # Load the translation pipeline
-translator = pipeline(task="translation", model=translation_model)
+translator = pipeline(task="translation", model=translation_models[selected_translation])
 
 # User input for translation
 user_input = st.text_area("Enter text for translation:", "")
@@ -30,3 +36,4 @@ st.write(
 st.write(
     "Select a translation model from the dropdown, enter text, and click 'Translate' to see the translation."
 )
+