Update app.py

app.py

@@ -145,7 +145,7 @@ st.markdown(st.session_state.current_text)
 def render_markdown_to_html(markdown_str):
     return markdown.markdown(markdown_str)
 
-def create_pdf_from_markdown_strings(markdown_strings, output_file):
+def create_pdf_from_markdown_strings(markdown_strings):
     html_pages = [render_markdown_to_html(md) for md in markdown_strings]
 
     # Combine HTML content with page breaks and add a style section for font size and margins
@@ -173,8 +173,31 @@ def create_pdf_from_markdown_strings(markdown_strings, output_file):
         'orientation': 'Landscape'
     }
 
+    return html_pages
+
+def html_to_pdf(html_content):
+    # Create a temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+        pdfkit.from_string(html_content, tmp_file.name)
+        return tmp_file.name
+
     # Convert combined HTML to PDF directly into a file
     pdfkit.from_string(combined_html, output_file, options=options)
 
-create_pdf_from_markdown_strings(st.session_state.summ_text, '/tmp/output.pdf')
+html_content = create_pdf_from_markdown_strings(st.session_state.summ_text)
+
+if st.button("Download PDF"):
+    if html_content:
+        pdf_path = html_to_pdf(html_content)
+        
+        with open(pdf_path, "rb") as pdf_file:
+            st.download_button(
+                label="Download PDF",
+                data=pdf_file,
+                file_name="converted.pdf",
+                mime="application/pdf"
+            )
+
+        # Remove the temporary file
+        os.remove(pdf_path)