import streamlit as st
from transformers import pipeline

# Function to load the translation pipeline based on the target language
@st.cache_resource
def load_translation_pipeline(target_language):
    if target_language == "French":
        model_name = "Helsinki-NLP/opus-mt-en-fr"
    elif target_language == "Spanish":
        model_name = "Helsinki-NLP/opus-mt-en-es"
    elif target_language == "German":
        model_name = "Helsinki-NLP/opus-mt-en-de"
    else:
        st.error("Target language not supported!")
        return None
    return pipeline("translation", model=model_name)

# Streamlit app layout
st.title("Language Translator")

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

# Select target language
target_language = st.selectbox(
    "Select target language:",
    ["French", "Spanish", "German"]  # Add more languages if needed
)

# Translate button
if st.button("Translate"):
    if text:
        # Load the translation pipeline based on selected language
        translation_pipeline = load_translation_pipeline(target_language)
        if translation_pipeline:
            # Perform translation
            translation = translation_pipeline(text)
            translated_text = translation[0]['translation_text']
            st.write(f"**Translated text in {target_language}:**")
            st.write(translated_text)
    else:
        st.error("Please enter text to translate.")