import streamlit as st
from transformers import pipeline


st.title("Text Summarization App PEGASUS-Large) ")
st.subheader('Input text below to be summarised', divider='rainbow')
# Create a text input widget
text_input = st.text_area(label="Input Text", height=200)
generated_summary = ""
# Define a function to generate the summary
def generate_summary(text):
    summarizer = pipeline("summarization", model="sabre-code/pegasus-large-cnn-dailymail")
    generated_summary = summarizer(text)
    
    
    # Return the generated summary
    return generated_summary

# Add a button to trigger the generation of the summary
generate_button = st.button(label="Generate Summary")
if generate_button:
    # Call the generate_summary function when the button is clicked
    generated_summary = generate_summary(text_input)
    st.success("Summary Generated!")


# Display the generated summary
st.markdown("## Summary")
st.code(generated_summary)