Update app.py

app.py

@@ -903,7 +903,8 @@ class ReportGenerator:
         html_content = html_content.replace("<h2>Conclusion</h2>", "<div class='page-break'></div><h2>Conclusion</h2>")
         html_content = html_content.replace("<h2>References</h2>", "<div class='page-break'></div><h2>References</h2>")
         html_content = html_content.replace("<h2>Surprise-Me Extension Report</h2>", "<div class='page-break'></div><h2>Surprise-Me Extension Report</h2>")
-        
+
+        # Add header
         date_str = datetime.now().strftime("%Y-%m-%d")
         header = ""
         if metadata:
@@ -911,13 +912,18 @@ class ReportGenerator:
 <p>Author: {metadata.get('User name', 'N/A')}</p>
 <p>Date: {metadata.get('Date', date_str)}</p>
 <hr/>"""
-
-        logging.info(f"ReportGenerator: HTML report generated successfully:\n{html_content}")
-        return html_content
+        soup = BeautifulSoup(html_content, "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: HTML report generated successfully:\n{updated_html}")
+        return updated_html
 
     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, metadata)
+        html_report = self.generate_report_html(solution_content)
         
         # Parse the HTML
         soup = BeautifulSoup(html_report, "html.parser")
@@ -1002,6 +1008,10 @@ class ReportGenerator:
             return cleaned_string
         cleaned_string = remove_img_tags(final_html)
         logging.info(f"ReportGenerator: Final HTML for PDF conversion:\n{cleaned_string}")
+
+        #Post-processing tests
+        final_html = final_html.replace("<h1>Abstract</h1>","<h1>Abst<br>ract</h1>")
+
         
         # Generate the final PDF from final_html using xhtml2pdf (A4 layout)        
         pdf_buffer = io.BytesIO()