Update agents/report_generation_agent.py

agents/report_generation_agent.py

@@ -373,33 +373,67 @@ class ReportGeneratorAgent:
             
         elif format == "html":
             # Format as HTML
-            html = fr"""
-            <div class="report">
-                <h1>Report: {report.get('topic', 'Unknown Topic')}</h1>
-                <p class="timestamp"><em>Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</em></p>
+            # html = fr"""
+            # <div class="report">
+            #     <h1>Report: {report.get('topic', 'Unknown Topic')}</h1>
+            #     <p class="timestamp"><em>Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</em></p>
                 
-                <h2>Executive Summary</h2>
-                <div class="summary">
-                    <p>{report.get('executive_summary', 'No summary available.')}</p>
-                </div>
+            #     <h2>Executive Summary</h2>
+            #     <div class="summary">
+            #         <p>{report.get('executive_summary', 'No summary available.')}</p>
+            #     </div>
                 
-                <div class="confidence">
-                    <p><strong>Confidence Level: {report.get('confidence_level', 'unknown').title()}</strong></p>
-                    <p><em>{self.generate_confidence_statement(report.get('confidence_level', 'low'))}</em></p>
-                </div>
+            #     <div class="confidence">
+            #         <p><strong>Confidence Level: {report.get('confidence_level', 'unknown').title()}</strong></p>
+            #         <p><em>{self.generate_confidence_statement(report.get('confidence_level', 'low'))}</em></p>
+            #     </div>
                 
-                <h2>Detailed Report</h2>
-                <div class="detailed-report">
-                    {report.get('detailed_report', 'No detailed report available.').replace('\\n', '<br>')}
-                </div>
+            #     <h2>Detailed Report</h2>
+            #     <div class="detailed-report">
+            #         {report.get('detailed_report', 'No detailed report available.').replace('\\n', '<br>')}
+            #     </div>
                 
-                <h2>Sources</h2>
-                <ul>
-                    <li>Text Documents: {report.get('sources', {}).get('text_documents', 0)}</li>
-                    <li>Images: {report.get('sources', {}).get('images', 0)}</li>
-                </ul>
-            </div>
-            """
+            #     <h2>Sources</h2>
+            #     <ul>
+            #         <li>Text Documents: {report.get('sources', {}).get('text_documents', 0)}</li>
+            #         <li>Images: {report.get('sources', {}).get('images', 0)}</li>
+            #     </ul>
+            # </div>
+            # """
+            # Start with the opening tags
+            html = "<div class=\"report\">\n"
+            html += f"    <h1>Report: {report.get('topic', 'Unknown Topic')}</h1>\n"
+            html += f"    <p class=\"timestamp\"><em>Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}</em></p>\n"
+                
+            # Executive Summary section
+            html += "    <h2>Executive Summary</h2>\n"
+            html += "    <div class=\"summary\">\n"
+            html += f"        <p>{report.get('executive_summary', 'No summary available.')}</p>\n"
+            html += "    </div>\n"
+                
+            # Confidence section
+            html += "    <div class=\"confidence\">\n"
+            html += f"        <p><strong>Confidence Level: {report.get('confidence_level', 'unknown').title()}</strong></p>\n"
+            html += f"        <p><em>{self.generate_confidence_statement(report.get('confidence_level', 'low'))}</em></p>\n"
+            html += "    </div>\n"
+                
+            # Detailed Report section
+            html += "    <h2>Detailed Report</h2>\n"
+            html += "    <div class=\"detailed-report\">\n"
+            detailed_report = report.get('detailed_report', 'No detailed report available.').replace('\n', '<br>')
+            html += f"        {detailed_report}\n"
+            html += "    </div>\n"
+                
+            # Sources section
+            html += "    <h2>Sources</h2>\n"
+            html += "    <ul>\n"
+            html += f"        <li>Text Documents: {report.get('sources', {}).get('text_documents', 0)}</li>\n"
+            html += f"        <li>Images: {report.get('sources', {}).get('images', 0)}</li>\n"
+            html += "    </ul>\n"
+            
+            # Close the main div
+            html += "</div>\n"
+
             return html
         
         else: