import streamlit as st
from transformers import pipeline

# Title of the app
st.title("Translator App")
st.write("Translate text between multiple languages using AI!")

# Load translation model
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")  # English to French by default

# Select source and target languages
language_pairs = {
    "English → French": "Helsinki-NLP/opus-mt-en-fr",
    "French → English": "Helsinki-NLP/opus-mt-fr-en",
    "English → German": "Helsinki-NLP/opus-mt-en-de",
    "German → English": "Helsinki-NLP/opus-mt-de-en",
}

option = st.selectbox("Choose Language Pair:", list(language_pairs.keys()))
translator = pipeline("translation", model=language_pairs[option])

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

# Translate button
if st.button("Translate"):
    if text:
        translated_text = translator(text)[0]["translation_text"]
        st.success(f"Translated Text: {translated_text}")
    else:
        st.warning("Please enter some text to translate!")

# Footer
st.caption("Powered by 🤗 Hugging Face & Streamlit")