Update app.py #0

app.py

@@ -1,28 +1,32 @@
-import streamlit as str 
+import streamlit as st 
+import pdfkit
 from PyPDF2 import PdfReader
 
 from transformers import pipeline
 
 summarizer = pipeline(task="summarization")
 
-str.set_page_config(
+# Basic text summary
+st.set_page_config(
     page_title='Text Summarizer'
 )
 
-str.title('Text Summarization')
+st.title('Text Summarization')
+
+# Text summary function
 
 def summarize_text(text):
     summary = summarizer(text)
     summary = summary[0]['summary_text']
     return summary
 
-input = str.text_area('Enter long text')
+input = st.text_area('Enter long text')
 
 output = summarize_text(input)
 
 
-if str.button('Summarize text'):
-    str.markdown(f'''
+if st.button('Summarize text'):
+    st.markdown(f'''
             <div style="background-color: black; color: white; font-weight: bold; padding: 1rem; border-radius: 10px;">
             <h4>Results</h4>
                 <p>
@@ -30,46 +34,63 @@ if str.button('Summarize text'):
                 </p>
             </div>
                 ''', unsafe_allow_html=True)
-    str.success('Done')
+    st.success('Done')
     
     
 #####
 
 # PDF summary section
-str.subheader('PDF summary')
+
+st.subheader('PDF summary')
 
 try:
-    uploaded_pdf = str.file_uploader('Choose a pdf file', type=['pdf'])
+    # Upload file
+    uploaded_pdf = st.file_uploader('Choose a pdf file', type=['pdf'])
 
     if uploaded_pdf is not None:
-        str.success('Succesfully uploaded')
+        st.success('Succesfully uploaded')
         
+    # Extract PDF content    
     def extract_text(pdf_file):
         pdf_content = PdfReader(pdf_file)
-        pages =  pdf_content.pages
+        pages =pdf_content.pages
         # page_count = len(pages)
-        page_text = pages[17].extract_text()
-
-        return page_text
+        
+        page_text_stack = []
 
+        for page in pages:
+            page_text = page.extract_text()
+            page_text_stack.append(page_text)
+        
+        pages_stack = []
+        
+        for text_stack in page_text_stack:
+            pages_stack.append(text_stack)
 
-    pdf_input = extract_text(uploaded_pdf)
+        return page_text
+    
+    
 
-    pdf_output = summarize_text(pdf_input)
 
-except:
-    str.error('Please select a valid file')  
+except: # Handle blank file error
+    st.error('Please select a valid file')
 
+#  Prepare output 
+pdf_input = extract_text(uploaded_pdf)
+pdf_output = summarize_text(pdf_input)
+summary_pdf = pdfkit.from_sting(pdf_input, 'Summary.pdf')
 
 
 
-if str.button('Summarize pdf page'):
-    str.markdown(f'''
+if st.button('Summarize pdf page'):
+    st.markdown(f'''
             <div style="background-color: black; color: white; font-weight: bold; padding: 1rem; border-radius: 10px;">
-            <h4>Results</h4>
+            <h4>Download the summary here </h4>
                 <p>
                     {pdf_output}
                 </p>
             </div>
                 ''', unsafe_allow_html=True)
-    str.success('PDF page summarized :)', icon="✅")
+    st.write('Download summary pdf here')
+    download_button = st.download_button(summary_pdf, label='Download summary')
+    st.success('PDF page summarized :)', icon="✅")