Update app.py

app.py

@@ -71,6 +71,12 @@ except: # Handle blank file error
     st.error('Please select a valid file')
 
 
+def check_page_count(pdf):
+    pdf_content = PdfReader(pdf)
+    pages =pdf_content.pages
+    page_count = len(pages)
+    
+    return page_count
     
 
 # Processs to trigger summary
@@ -78,7 +84,10 @@ if st.button('Summarize pdf content'):
     with st.spinner('Extracting text from PDF...'):
         pdf_input = extract_text(uploaded_pdf)
         st.success('Text extracted')
- 
+
+    num_of_pages = check_page_count(uploaded_pdf)
+    st.success(f'NUmber of pages is {num_of_pages}.')
+    
     with st.spinner('Summarizing extracted text...'):
         pdf_output = []
     
@@ -103,11 +112,14 @@ if st.button('Summarize pdf content'):
 # Summary text to PDF function
     def summary_download():
         # summary_pdf = pdfkit.from_string(pdf_summary, 'Summary.pdf')
-        st.download_button(label='Download summary PDF', data=pdf_summary, file_name='summary.pdf', mime='application/pdf')   
-
+        download_button = st.download_button(label='Download summary PDF', data=pdf_summary, file_name='summary.pdf', mime='application/pdf')   
+        return download_button
+    
+    
     if st.button('Generate pdf download link'):
         summary_download()
-    
+
+  
 
 st.write('')
 st.write('')