Update functions

app.py

@@ -1,6 +1,7 @@
 import streamlit as st 
-from PyPDF2 import PdfReader
+import pdfkit
 
+from PyPDF2 import PdfReader
 from transformers import pipeline
 
 summarizer = pipeline(task="summarization")
@@ -69,13 +70,10 @@ try:
 except: # Handle blank file error
     st.error('Please select a valid file')
 
-#  Prepare output 
-
-
-# summary_pdf = pdfkit.from_string(pdf_output, 'Summary.pdf')
-
 
+    
 
+# Processs to trigger summary
 if st.button('Summarize pdf content'):
     with st.spinner('Extracting text from PDF...'):
         pdf_input = extract_text(uploaded_pdf)
@@ -94,17 +92,26 @@ if st.button('Summarize pdf content'):
         
     st.markdown(f'''
             <div style="background-color: black; color: white; font-weight: bold; padding: 1rem; border-radius: 10px;">
-             <h4>Summary</h4>
+             <h4>Summary </h4>
                 <p>{pdf_summary}</p>
             </div>
                 ''', unsafe_allow_html=True)
-    # st.write('Download summary pdf here')
-    # download_button = st.download_button(summary_pdf, label='Download summary')
+    
+    generate_download_button = st.button('Generate')
     st.success('PDF page summarized :)', icon="✅")
+    
+    
+# 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')   
 
+if generate_download_button:
+    summary_download()
+    
 
 st.write('')
-st.write(''
+st.write('')
 
 
 st.markdown("<hr style='border: 1px dashed #ddd; margin: 2rem;'>", unsafe_allow_html=True) #Horizontal line
@@ -121,5 +128,3 @@ st.markdown("""
     </div>
 """,
 unsafe_allow_html=True)
-
-# Peace Out :)