import streamlit as st
from transformers import MarianMTModel, MarianTokenizer

# Load the model and tokenizer
model_name = "Helsinki-NLP/opus-mt-ro-en"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

# 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.")