import streamlit as st
from transformers import pipeline

# Load translation pipelines
models = {
    "French": "Helsinki-NLP/opus-mt-en-fr",
    "Spanish": "Helsinki-NLP/opus-mt-en-es",
    "German": "Helsinki-NLP/opus-mt-en-de",
    "Urdu": "Helsinki-NLP/opus-mt-en-ur"
}

translators = {lang: pipeline("translation", model=model) for lang, model in models.items()}

# UI Layout
st.title("Translex Multi-Language Translator")
st.write("Translate text from English to multiple languages")

# Input text
text = st.text_area("Enter text in English:")

# Select target language
target_language = st.selectbox("Select the target language:", list(models.keys()))

if st.button("Translate"):
    if text and target_language:
        # Perform translation
        translation = translators[target_language](text)[0]['translation_text']
        st.write(f"Translated text in {target_language}:")
        st.write(translation)
    else:
        st.write("Please enter text and select a target language.")