from transformers import MarianTokenizer, MarianMTModel
import streamlit as st

tokenizer = MarianTokenizer.from_pretrained("BlackKakapo/opus-mt-ro-en")
model = MarianMTModel.from_pretrained("BlackKakapo/opus-mt-ro-en")

# Streamlit app
st.title("Romanian to English Translator")

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

# Translate button
if st.button("Translate"):
    if input_text:
        # Tokenize the input text
        inputs = tokenizer(input_text, return_tensors="pt", padding=True)
        # Perform translation
        translated_tokens = model.generate(**inputs)
        translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
        st.write("Translation:")
        st.write(translation)
    else:
        st.write("Please enter some text to translate.")