update

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import PyPDF2
 from extractive_model import summarize_pdf_with_textrank
 
 # Set page to wide mode
@@ -9,42 +10,39 @@ 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()
+        pdf_text += pdf_reader.pages[page_num].extract_text() or ""
     return pdf_text
 
 # Main app
 def main():
-
     st.title("Terms of Service Summarizer")
 
     # Layout: 3 columns
     col1, col2, col3 = st.columns([1, 3, 2], gap="large")
 
-    # Left column: Dropdown menu
+    # Left column: Radio buttons for summarizer choice
     with col1:
-        # Options for the radio buttons
         radio_options = ['Abstractive', 'Extractive']
-
-        # Create radio buttons and get the selected option
         radio_selection = st.radio("Choose type of summarizer:", radio_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
+            summary = ""
             if uploaded_file is not None:
+                # Extract text from PDF
                 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
+                if radio_selection == "Extractive":
+                    # Perform extractive summarization
+                    summary = summarize_pdf_with_textrank(file_content)
+                else:
+                    None
             else:
-                st.wirte("Upload a PDF or put in your text!")
+                st.write("Please upload a PDF or enter some text to summarize.")
+
             st.session_state.summary = summary
 
     # Right column: Displaying text after pressing 'Summarize'

extractive_model.py

@@ -9,7 +9,6 @@ from pysummarization.abstractabledoc.top_n_rank_abstractor import TopNRankAbstra
 from sumy.nlp.stemmers import Stemmer
 from sumy.utils import get_stop_words"""
 
-import PyPDF2
 from sumy.parsers.plaintext import PlaintextParser
 from sumy.nlp.tokenizers import Tokenizer
 from sumy.summarizers.text_rank import TextRankSummarizer