import streamlit as st
from transformers import pipeline


translator = pipeline("translation", model="sarahai/nllb-uzbek-cyrillic-to-russian")
summarizer = pipeline("summarization", model="sarahai/ruT5-base-summarizer", max_length=512, truncation=True)  

def translate_and_summarize(uzbek_text):
    """Translates Uzbek text to Russian and summarizes the translation."""

    
    if not uzbek_text:
        st.warning("Пожалуйста введите текст для перевода")
        return

    try:
        
        russian_text = translator(uzbek_text)[0]["translation_text"]

        
        summary = summarizer(russian_text, max_length=200, truncation=True)[0]["summary_text"]

        
        st.success("Перевод:")
        st.write(russian_text)
        st.success("Краткое содержание:")
        st.write(summary)
    except Exception as e:
        st.error(f"Произошла ошибка: {e}")

st.title("Перевод узбекских текстов и суммаризация русских")
st.subheader("Переведите текст с узбекского на русский язык и кратко его суммируйте с помощью моделей Hugging Face.")

uzbek_text = st.text_area("Enter Uzbek Text", height=200)

if st.button("Перевод и суммаризация "):
    translate_and_summarize(uzbek_text)