import streamlit as st
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM

translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")

def translate_text(text):
    if text:
        result = translator(text)
        return result[0]['translation_text']
    return ""

st.title("Text Translation App")
st.subheader("Enter text to translate:")

user_input = st.text_area("Your text here:", height=200)

if st.button("Translate"):
    translation = translate_text(user_input)
    st.subheader("Translated Text:")
    st.write(translation)
else:
    st.info("Enter text and click 'Translate' to see the result.")

if __name__ == '__main__':
    import streamlit.web.cli as stcli
    stcli.main()