Update app.py

app.py

@@ -79,13 +79,28 @@ def format_response(data):
     if not data:
         return "No data available for this query."
 
+    # Handle nested dictionaries
+    def flatten_dict(d, parent_key='', sep=': '):
+        items = []
+        for k, v in d.items():
+            new_key = f"{parent_key}{sep}{k}" if parent_key else k
+            if isinstance(v, dict):
+                items.extend(flatten_dict(v, new_key, sep=sep).items())
+            else:
+                items.append((new_key, v))
+        return dict(items)
+
     if isinstance(data, dict):
+        # Flatten nested dictionaries and format the result
+        flattened_data = flatten_dict(data)
         formatted_response = "### Insights:\n\n"
-        for key, value in data.items():
+        for key, value in flattened_data.items():
             formatted_response += f"**{key.capitalize()}**: {value}\n\n"
     elif isinstance(data, list):
+        # Handle list response
         formatted_response = "### Insights:\n\n" + "\n".join(f"{idx+1}. {item}" for idx, item in enumerate(data))
     else:
+        # Handle simple string or non-dict response
         formatted_response = f"### Key Insights:\n\n{data}"
 
     return formatted_response.strip()