Spaces:
Sleeping
Sleeping
File size: 1,522 Bytes
d38c13b fa77629 d38c13b fa77629 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import streamlit as st
from utils import generate_summary
# Initialize session state variables
if "clicked" not in st.session_state:
st.session_state.clicked = False
if "input_text" not in st.session_state:
st.session_state.input_text = ""
if "generated_summary" not in st.session_state:
st.session_state.generated_summary = ""
st.title("Dialogue Text Summarization")
st.write("---")
height = 200
# Text area with session state
input_text = st.text_area("Dialogue", height=height, key="input_text")
# Submit button logic
if st.button("Submit"):
if st.session_state.input_text.strip() == "":
st.error("Please enter a dialogue!")
else:
st.write("---")
st.write("## Summary")
st_container = st.empty()
st_info_container = st.empty()
# Generate summary and store it in session state
st.session_state.generated_summary = generate_summary(
" ".join(st.session_state.input_text.split()),
st_container,
st_info_container
)
# Display the generated summary
if st.session_state.generated_summary:
st.write(st.session_state.generated_summary)
# Clear button logic
def clear_all():
st.session_state.clicked = True
st.session_state.input_text = "" # Clear input text
st.session_state.generated_summary = "" # Clear summary
st.button("Clear", on_click=clear_all)
# Logic for clearing display
if st.session_state.clicked:
st.session_state.clicked = False
st.experimental_rerun()
|