Update app.py

app.py

@@ -13,6 +13,7 @@ import markdown2
 import re
 import json
 import os
+from markdown_pdf import MarkdownPdf, Section
 
 # Configure Gemini API
 gemini_api_key = os.environ.get('GOOGLE_API_KEY')
@@ -335,7 +336,7 @@ def main():
                     
                     # Generate report
                     prompt = f"""
-                        Analyze this business data and generate a comprehensive report in plain text format.  Do not include any json.
+                        Analyze this business data and generate a comprehensive report in plain text format. Use markdown for headings and structure. Do not include any json.
 
                         Data:
                         {json.dumps(json_data, indent=2)}
@@ -345,8 +346,24 @@ def main():
                     response = model.generate_content(prompt)
                     report = response.text
                     st.markdown(report)
+
+                    
                     # PDF Generation and display
-                
+                   try:
+                        pdf = MarkdownPdf()
+                        pdf.meta["title"] = 'Surburb Business Report'
+                        pdf.add_section(Section(Report, toc=False))
+                        pdf.save('report.pdf')
+                        st.download_button(
+                            label="Download Report as PDF",
+                            data=pdf_bytes,
+                            file_name="report.pdf",
+                            mime="application/pdf"
+                        )
+                        st.markdown(report)  # Display the report below the download button
+                    except Exception as e:
+                        st.error(f"Error generating PDF: {e}")
+                        st.markdown(report) 
         else:
             st.error("No data available for reports")