Update app.py

app.py

@@ -905,9 +905,23 @@ class ReportGenerator:
     def generate_report_pdf(self, solution_content: str, metadata: dict = None) -> bytes:
         # Generate the full HTML report (including text, focus placeholders, and mermaid visuals as iframes)
         html_report = self.generate_report_html(solution_content)
+
+        # Add header
+        date_str = datetime.now().strftime("%Y-%m-%d")
+        header = ""
+        if metadata:
+            header = f"""
+<h1>Search Query: {metadata.get('Query name', 'N/A')}</h1>
+<p>Author: {metadata.get('User name', 'N/A')}</p>
+<p>Date: {metadata.get('Date', date_str)}</p>
+<hr/>"""
+        soup = BeautifulSoup(html_report, "html.parser")
+        body_tag = soup.body
+        if body_tag:
+            body_tag.insert(0, BeautifulSoup(header, "html.parser"))
+        updated_html = str(soup)
         
         # Parse the HTML
-        soup = BeautifulSoup(html_report, "html.parser")
         logging.info(f"ReportGenerator: soup report generated:\n{soup}")
         
         # Find all mermaid visual iframes (assumed to have class "mermaid-frame")
@@ -989,29 +1003,14 @@ class ReportGenerator:
             return cleaned_string
         cleaned_string = remove_img_tags(final_html)
 
-        # Add header
-        date_str = datetime.now().strftime("%Y-%m-%d")
-        header = ""
-        if metadata:
-            header = f"""
-<h1>Search Query: {metadata.get('Query name', 'N/A')}</h1>
-<p>Author: {metadata.get('User name', 'N/A')}</p>
-<p>Date: {metadata.get('Date', date_str)}</p>
-<hr/>"""
-        soup = BeautifulSoup(cleaned_string, "html.parser")
-        body_tag = soup.body
-        if body_tag:
-            body_tag.insert(0, BeautifulSoup(header, "html.parser"))
-        updated_html = str(soup)
-
-        logging.info(f"ReportGenerator: Final HTML for PDF conversion:\n{updated_html}")
+        logging.info(f"ReportGenerator: Final HTML for PDF conversion:\n{cleaned_string}")
         
         #Post-processing tests
-        updated_html = updated_html.replace("<h1>Abstract</h1>","<h1>Abs ;lkj sdf;tract</h1>")
+        final_html = final_html.replace("<h1>Abstract</h1>","<h1>Abs ;lkj sdf;tract</h1>")
 
         # Generate the final PDF from final_html using xhtml2pdf (A4 layout)        
         pdf_buffer = io.BytesIO()
-        pisa_status = pisa.CreatePDF(updated_html, dest=pdf_buffer, 
+        pisa_status = pisa.CreatePDF(final_html, dest=pdf_buffer, 
                                      link_callback=lambda uri, rel: uri,
                                      default_css="""
 @page {