Update app.py

app.py

@@ -432,7 +432,7 @@ def generate_documentation_page():
 
         # Button to generate PDF
         if st.button("Generate PDF"):
-            pdf = fpdf.FPDF()
+            pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
             pdf.set_font("Arial", size=12)
@@ -448,9 +448,16 @@ def generate_documentation_page():
                     pdf.set_font("Arial", size=12)
                 pdf.multi_cell(0, 10, line)
 
-            pdf_file_path = os.path.join(user_folder, f"{st.session_state.current_project}_documentation.pdf")
-            pdf.output(pdf_file_path)
-            st.success(f"PDF file generated at: {pdf_file_path}")
+            # Save and allow download of the PDF
+            pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+            pdf.output(pdf_file.name)
+            st.download_button(
+                label="Download PDF",
+                data=open(pdf_file.name, "rb").read(),
+                file_name="documentation.pdf",
+                mime="application/pdf",
+            )
+            os.unlink(pdf_file.name)
 
         # Button to generate Markdown file
         if st.button("Generate Markdown File"):