Update app.py

app.py

@@ -79,29 +79,29 @@ 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)
+    formatted_response = ""
 
     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 flattened_data.items():
-            formatted_response += f"**{key.capitalize()}**: {value}\n\n"
+        formatted_response += "### Key Insights:\n\n"
+        for key, value in data.items():
+            if isinstance(value, dict):
+                # For nested dictionaries, add a heading for the sub-keys
+                formatted_response += f"**{key.capitalize()}**:\n"
+                for sub_key, sub_value in value.items():
+                    formatted_response += f"  - **{sub_key.capitalize()}**: {sub_value}\n"
+            elif isinstance(value, list):
+                # If the value is a list, show it as a bullet-point list
+                formatted_response += f"**{key.capitalize()}**:\n"
+                for idx, item in enumerate(value):
+                    formatted_response += f"  - {item}\n"
+            else:
+                # For other types of data, simply display key-value pairs
+                formatted_response += f"**{key.capitalize()}**: {value}\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))
+        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}"
+        formatted_response = f"### Key Insight:\n\n{data}"
 
     return formatted_response.strip()