Update app.py

app.py

@@ -316,20 +316,46 @@ def display_table(analyst_output):
 def parse_analyst_output(raw_output):
     structured_data = []
     current_category = None
+    current_values = []
 
     lines = raw_output.split('\n')
-    
+
     for line in lines:
         line = line.strip()
+
+        # Detect new category
         if line.startswith("Category:"):
+            # Save previous category if exists
+            if current_category and current_values:
+                structured_data.append({
+                    "Category": current_category,
+                    "Values": current_values if len(current_values) > 1 else current_values[0]
+                })
+            # Start new category
             current_category = line.replace("Category:", "").strip()
+            current_values = []
+
+        # Detect start of values
         elif line.startswith("Values:"):
-            continue 
+            continue
+
+        # Handle content under a category
         elif line and current_category:
-            structured_data.append({
-                "Category": current_category,
-                "Values": line
-            })
+            try:
+                # Attempt to parse the line as a dictionary or list
+                parsed_value = ast.literal_eval(line)
+                current_values.append(parsed_value)
+            except (ValueError, SyntaxError):
+                # If parsing fails, treat it as plain text
+                current_values.append(line)
+
+    # Save the last category
+    if current_category and current_values:
+        structured_data.append({
+            "Category": current_category,
+            "Values": current_values if len(current_values) > 1 else current_values[0]
+        })
+
     return structured_data