Update app.py

app.py

@@ -8,12 +8,13 @@ from PIL import Image
 import io
 import base64
 import google.generativeai as genai
-from fpdf import FPDF
+#from fpdf import FPDF
 import markdown2
 import re
 import json
 import os
 from markdown_pdf import MarkdownPdf, Section
+import tempfile
 
 # Configure Gemini API
 gemini_api_key = os.environ.get('GOOGLE_API_KEY')
@@ -92,58 +93,7 @@ def create_dashboard(data):
                      title='Top 10 Products by Quantity Sold')
         st.plotly_chart(fig, use_container_width=True)
 
-# ... [Keep the md_to_pdf, generate_pdf, StreamLitResponse, and generateResponse functions same as before but update generateResponse]
-def md_to_pdf(md_text, pdf):
-    """Renders basic Markdown to PDF using fpdf text functions (limited formatting)."""
-    md = markdown2.markdown(md_text)  # Parse Markdown
-    lines = md.split('\n')  # Split into lines
-    pdf.set_font("Arial", "", 12)  # Set default font
 
-    for line in lines:
-        line = line.strip()
-
-        # Basic heading support (adjust as needed)
-        if line.startswith("# "):
-            pdf.set_font("Arial", "B", 18)
-            pdf.cell(0, 10, line[2:], ln=True)
-        elif line.startswith("## "):
-            pdf.set_font("Arial", "B", 16)
-            pdf.cell(0, 10, line[3:], ln=True)
-        elif line.startswith("### "):
-            pdf.set_font("Arial", "B", 14)
-            pdf.cell(0, 10, line[4:], ln=True)
-
-        # Basic bold text support (very limited)
-        elif "**" in line:
-            parts = line.split("**")
-            for i, part in enumerate(parts):
-                if i % 2 == 1:  # Bold text
-                    pdf.set_font("Arial", "B", 12)
-                    pdf.cell(0, 10, part, ln=False)  # Don't add newline for inline bold
-                else:
-                    pdf.set_font("Arial", "", 12)
-                    pdf.cell(0, 10, part, ln=False)
-            pdf.ln()  # Newline after the whole line
-
-        # Add other basic formatting as needed...
-
-        else:  # Normal text
-            pdf.set_font("Arial", "", 12)
-            pdf.multi_cell(0, 10, line)  # multi_cell for wrapping
-
-
-def generate_pdf(report_text):
-    """Generates PDF from report text."""
-    pdf = FPDF()
-    pdf.add_page()
-    try:
-        pdf.add_font('Arial', '', 'arial.ttf', uni=True) # Add unicode support
-    except:
-        st.warning("Arial font not found. Unicode might not work.")
-    pdf.set_font("Arial", "", 12)
-    md_to_pdf(report_text, pdf)
-    pdf_bytes = pdf.output(dest="S").encode("latin1")
-    return pdf_bytes
 
 # --- Chat Tab Functions ---
 class StreamLitResponse(ResponseParser):
@@ -345,26 +295,34 @@ def main():
                     """
                     response = model.generate_content(prompt)
                     report = response.text
+                    html_text = markdown2.markdown(markdown_text)
                     
-
+ 
                     
                     # PDF Generation and display
                     try:
-                        pdf = MarkdownPdf()
-                        pdf.meta["title"] = 'Surburb Business Report'
-                        pdf.add_section(Section(report, toc=False))
-                        pdf.save('report.pdf')
+                    # Create a temporary file to store the PDF
+                        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+                            pdf = MarkdownPdf()
+                            pdf.meta["title"] = 'Suburb Business Report'
+                            pdf.add_section(Section(report, toc=False))
+                            pdf.save(tmp_file.name)  # Save the PDF to the temporary file
+
+                        # Read the PDF bytes from the temporary file
+                        with open(tmp_file.name, "rb") as f:
+                            pdf_bytes = f.read()
+
+                    # Provide the PDF for download
                         st.download_button(
                             label="Download Report as PDF",
-                            data=pdf,
+                            data=pdf_bytes,
                             file_name="report.pdf",
                             mime="application/pdf"
-                        )
-                        st.markdown(report)
-              # Display the report below the download button
+                    )   
+                        st.write(html_text, unsafe_allow_html=True)  # Display the report below the download button
                     except Exception as e:
                         st.error(f"Error generating PDF: {e}")
-                        st.markdown(report)
+                        st.write(html_text, unsafe_allow_html=True)
                         
         else: 
             st.error("No data available for reports")