import streamlit as st
from transformers import pipeline

# Load the translation pipeline with the specified model
pipe = pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")

# Define language codes
language_codes = {
    "Bengali": "bn_IN",
    "Gujarati": "gu_IN",
    "Kannada": "kn_IN",
    "Telugu": "te_IN",
    "English": "en_XX"
}

# Streamlit app
st.title("Multilingual Translation App")

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

# Language selection
target_language = st.selectbox("Select target language:", options=list(language_codes.keys()))

# Translation button
if st.button("Translate"):
    if input_text:
        # Get the target language code
        tgt_lang = language_codes[target_language]
        
        # Perform translation
        translated_text = pipe(input_text, src_lang="en_XX", tgt_lang=tgt_lang)[0]['translation_text']
        
        # Display translated text
        st.subheader("Translated Text:")
        st.write(translated_text)
    else:
        st.write("Please enter text to translate.")