import streamlit as st
from transformers import pipeline

# Streamlit app title
st.title("Translator App 🈹")
st.write("Translate text between different languages using AI.")

# Language model options
language_models = {
    "English → French": "Helsinki-NLP/opus-mt-en-fr",
    "French → English": "Helsinki-NLP/opus-mt-fr-en",
    "English → Spanish": "Helsinki-NLP/opus-mt-en-es",
    "Spanish → English": "Helsinki-NLP/opus-mt-es-en",
}

# Language selection
selected_model = st.selectbox("Select Language Pair", list(language_models.keys()))

# Load the translation model
translator = pipeline("translation", model=language_models[selected_model])

# Input text area
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 text to translate!")

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