import streamlit as st
from transformers import pipeline

# Load the model from Hugging Face
@st.cache_resource
def load_model():
    summarization_pipeline = pipeline("summarization", model="tareeb23/summarizer_bart")
    return summarization_pipeline

summarization_pipeline = load_model()

# Streamlit app
st.title("Dialogue Summarizer")
st.write("Enter a dialogue to get a summarized version.")

# User input
dialogue = st.text_area("Enter Dialogue", "")

if st.button("Summarize"):
    if dialogue:
        summary = summarization_pipeline(dialogue)
        st.write("### Summary:")
        st.write(summary[0]['summary_text'])
    else:
        st.write("Please enter a dialogue to summarize.")