Update app.py

app.py

@@ -103,18 +103,31 @@ def format_observations(observations):
     ])
 
 def format_insights(insights, visuals):
+    if isinstance(insights, dict):
+        # Old format (dict of key: text)
+        insight_items = list(insights.items())
+    elif isinstance(insights, list):
+        # New format (list of dicts with "insight" and optional "category")
+        insight_items = [(item.get("category", f"Insight {idx+1}"), item["insight"]) for idx, item in enumerate(insights)]
+    else:
+        return "<p>No insights available or incorrect format.</p>"
+
     return '\n'.join([
         f"""
         <div style="margin: 20px 0; padding: 20px; background: white; border-radius: 8px; box-shadow: 0 2px 4px rgba(0,0,0,0.05);">
             <div style="display: flex; align-items: center; gap: 10px;">
                 <div style="background: #2B547E; color: white; width: 30px; height: 30px; border-radius: 50%; display: flex; align-items: center; justify-content: center;">{idx+1}</div>
-                <p style="margin: 0; font-size: 16px;">{insight}</p>
+                <div>
+                    <h4 style="margin: 0; color: #2B547E;">{title}</h4>
+                    <p style="margin: 5px 0 0 0; font-size: 16px;">{insight}</p>
+                </div>
             </div>
             {f'<img src="/file={visuals[idx]}" style="max-width: 100%; height: auto; margin-top: 10px; border-radius: 6px; box-shadow: 0 2px 4px rgba(0,0,0,0.1);">' if idx < len(visuals) else ''}
         </div>
-        """ for idx, (key, insight) in enumerate(insights.items())
+        """ for idx, (title, insight) in enumerate(insight_items)
     ])
 
+
 def analyze_data(csv_file, additional_notes=""):
     start_time = time.time()
     process = psutil.Process(os.getpid())