import streamlit as st
from transformers import MarianMTModel, MarianTokenizer

@st.cache_resource
def load_model():
    model_name = "Helsinki-NLP/opus-mt-en-grk"
    tokenizer = MarianTokenizer.from_pretrained(model_name)
    model = MarianMTModel.from_pretrained(model_name)
    return tokenizer, model

tokenizer, model = load_model()

st.title("English to Greek Translator")
st.write("Enter English text to translate it into Greek.")

text = st.text_area("English Text", height=200)
if st.button("Translate"):
    if text.strip():
        # Add the required language token for Greek
        src_text = [f">>ell<< {text.strip()}"]
        inputs = tokenizer(src_text, return_tensors="pt", padding=True)
        translated = model.generate(**inputs)
        greek_translation = tokenizer.decode(translated[0], skip_special_tokens=True)
        st.subheader("Greek Translation")
        st.write(greek_translation)
    else:
        st.warning("Please enter some text to translate.")