Fixed html

app.py

@@ -1234,29 +1234,38 @@ def run_data_analysis(dataset_url, llm_provider, api_key, missing_strategy, crea
 
         # Get the final report
         report_html = "<h2>No report generated</h2>"
-        if "report_result" in [msg["message"]["message_type"] for msg in compute_messages if msg["type"] == "received"]:
-            # Find the report message
-            for msg in compute_messages:
-                if msg["type"] == "received" and msg["message"]["message_type"] == "report_result":
-                    report_content = msg["message"]["content"]["report"]
-                    if "content" in report_content:
-                        # LLM-generated report
-                        report_html = f"<h2>{report_content['title']}</h2>"
-                        newline_replaced = report_content['content'].replace("\n", "<br>")
-                        report_html += f"<div>{newline_replaced}</div>"
-                    elif "sections" in report_content:
-                        # Template-based report
-                        report_html = f"<h2>{report_content['title']}</h2>"
-                        for section in report_content["sections"]:
-                            report_html += f"<h3>{section['title']}</h3>"
-                            report_html += f"<p>{section['content']}</p>"
-                            if "insights" in section:
-                                report_html += "<ul>"
-                                for insight in section["insights"]:
-                                    report_html += f"<li>{insight}</li>"
-                                report_html += "</ul>"
-                            if "data" in section:
-                                report_html += "<pre>" + format_json(section["data"]) + "</pre>"
+
+        # Check if report result exists
+        report_found = False
+        for msg in compute_messages:
+            if msg["type"] == "received" and msg["message"]["message_type"] == "report_result":
+                report_found = True
+                try:
+                    if "content" in msg["message"] and "report" in msg["message"]["content"]:
+                        report_content = msg["message"]["content"]["report"]
+                        if "content" in report_content:
+                            # LLM-generated report
+                            report_html = f"<h2>{report_content['title']}</h2>"
+                            newline_replaced = report_content['content'].replace("\n", "<br>")
+                            report_html += f"<div>{newline_replaced}</div>"
+                        elif "sections" in report_content:
+                            # Template-based report
+                            report_html = f"<h2>{report_content['title']}</h2>"
+                            for section in report_content["sections"]:
+                                report_html += f"<h3>{section['title']}</h3>"
+                                report_html += f"<p>{section['content']}</p>"
+                                if "insights" in section:
+                                    report_html += "<ul>"
+                                    for insight in section["insights"]:
+                                        report_html += f"<li>{insight}</li>"
+                                    report_html += "</ul>"
+                                if "data" in section:
+                                    report_html += "<pre>" + format_json(section["data"]) + "</pre>"
+                except Exception as e:
+                    report_html = f"<h2>Error generating report: {str(e)}</h2>"
+
+        if not report_found:
+            report_html = "<h2>No report message received</h2>"
 
         # Return all results
         return {
@@ -1268,11 +1277,12 @@ def run_data_analysis(dataset_url, llm_provider, api_key, missing_strategy, crea
 
     except Exception as e:
         import traceback
+        error_traceback = traceback.format_exc()
         return {
-            'mcp_messages': f"Error: {str(e)}\n\n{traceback.format_exc()}",
+            'mcp_messages': f"Error: {str(e)}\n\n{error_traceback}",
             'llm_logs': "",
             'visualizations': "",
-            'final_report': ""
+            'final_report': f"<h2>Error: {str(e)}</h2><pre>{error_traceback}</pre>"
         }
 
 # Define the Gradio interface