Update app.py

app.py

@@ -565,6 +565,20 @@ class ReportGenerator:
         logging.info("ReportGenerator: HTML report generated successfully.")
         return full_html
 
+    def generate_report_pdf(self, solution_content: str, metadata: dict = None) -> bytes:
+        """
+        Generate a PDF report by first creating the HTML report and converting it with xhtml2pdf.
+        """
+        # Create the HTML content
+        html_report = self.generate_report_html(solution_content, metadata)
+        # Convert the HTML to PDF using xhtml2pdf (pisa)
+        pdf_buffer = io.BytesIO()
+        pisa_status = pisa.CreatePDF(html_report, dest=pdf_buffer)
+        if pisa_status.err:
+            raise Exception("Error converting HTML to PDF")
+        logging.info("ReportGenerator: PDF report generated successfully from HTML.")
+        return pdf_buffer.getvalue()
+
 def handle_generate_report(query_name: str, user_name: str, final_report: str):
     try:
         report_generator = ReportGenerator()
@@ -576,12 +590,13 @@ def handle_generate_report(query_name: str, user_name: str, final_report: str):
         }
         if len(final_report) > MAX_MESSAGE_LENGTH:
             final_report = compress_text(final_report, MAX_MESSAGE_LENGTH)
-        html_report = report_generator.generate_report_html(solution_content=final_report, metadata=metadata)
-        # Optionally, you could write the HTML to a temporary file for download as .html
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.html') as tmp_file:
-            tmp_file.write(html_report.encode("utf-8"))
+        # Get the PDF content from the HTML report
+        pdf_bytes = report_generator.generate_report_pdf(solution_content=final_report, metadata=metadata)
+        # Write the PDF content to a temporary file for download.
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file:
+            tmp_file.write(pdf_bytes)
             tmp_path = tmp_file.name
-        logging.info(f"handle_generate_report: Report generated at {tmp_path}")
+        logging.info(f"handle_generate_report: PDF report generated at {tmp_path}")
         return "Report generated successfully.", gr.update(value=tmp_path, visible=True)
     except Exception as e:
         logging.error(f"handle_generate_report error: {e}", exc_info=True)