Update reports/export_pdf.py

reports/export_pdf.py

@@ -40,7 +40,7 @@ def create_ei_heatmap(data):
     plt.close(fig)
     return tmp.name
 
-# 3. PDF Generation
+# 3. Main report function
 def generate_report():
     st.subheader("📄 Export Your Weekly Report")
 
@@ -66,6 +66,7 @@ def generate_report():
         pdf.add_page()
         pdf.set_auto_page_break(auto=True, margin=15)
 
+        # Title and intro
         pdf.set_font("Arial", 'B', 14)
         pdf.cell(0, 10, f"MindMentor Report - {name}", ln=True, align='C')
         pdf.set_font("Arial", '', 11)
@@ -114,7 +115,7 @@ def generate_report():
         pdf.image(heatmap_path, w=100)
         os.unlink(heatmap_path)
 
-        # Section 4: Final Summary
+        # Final Summary
         pdf.set_font("Arial", 'B', 12)
         pdf.cell(0, 10, "Overall Summary", ln=True)
         pdf.set_font("Arial", '', 11)
@@ -124,16 +125,16 @@ def generate_report():
             "Keep up your cognitive training and continue practicing daily EI drills for balanced growth."
         )
 
-        # Create download button using in-memory buffer
-        pdf_buffer = io.BytesIO()
-        pdf.output(pdf_buffer)
-        pdf_data = pdf_buffer.getvalue()
+        # Convert to BytesIO for download
+        pdf_output = pdf.output(dest='S').encode('latin1')
+        pdf_buffer = io.BytesIO(pdf_output)
 
         st.download_button(
             label="📄 Download Your MindMentor PDF Report",
-            data=pdf_data,
+            data=pdf_buffer,
             file_name=f"{name}_MindMentor_Report.pdf",
             mime="application/pdf"
         )
 
 
+