Spaces:
Sleeping
Sleeping
import streamlit as st | |
import pandas as pd | |
import PyPDF2 | |
import base64 | |
from summarizer import bert_summarizer, simple_summarizer | |
summarizer = bert_summarizer | |
# Set page to wide mode | |
st.set_page_config(layout="wide") | |
# Function to handle file upload and return its content | |
def load_pdf(file): | |
pdf_reader = PyPDF2.PdfReader(file) | |
pdf_text = "" | |
for page_num in range(len(pdf_reader.pages)): | |
pdf_text += pdf_reader.pages[page_num].extract_text() | |
return pdf_text | |
# Main app | |
def main(): | |
st.title("Streamlit App") | |
# Layout: 3 columns | |
col1, col2, col3 = st.columns([1, 3, 2], gap="large") | |
# Left column: Dropdown menu | |
with col1: | |
dropdown_options = ['Abstractive', 'Extractive'] | |
dropdown_selection = st.selectbox("Choose type of summerizer:", dropdown_options) | |
# Middle column: Text input and File uploader | |
with col2: | |
user_input = st.text_input("Enter your text here:") | |
uploaded_file = st.file_uploader("Upload a PDF", type="pdf") | |
if st.button("Summarize"): | |
# Handling file upload | |
if uploaded_file is not None: | |
file_content = load_pdf(uploaded_file) | |
st.write("PDF uploaded successfully.") | |
# summary = summarizer(file_content) | |
summary = file_content | |
elif user_input is not None: | |
# summary = summarizer(user_input) | |
summary = user_input | |
else: | |
st.wirte("Upload a PDF or put in your text!") | |
st.session_state.summary = summary | |
# Right column: Displaying text after pressing 'Summarize' | |
with col3: | |
st.write("Output:") | |
if 'summary' in st.session_state: | |
st.write(st.session_state.summary) | |
if __name__ == "__main__": | |
main() |