Update app.py

app.py

@@ -111,17 +111,10 @@ def extract_json_from_codeagent_output(raw_output):
             except json.JSONDecodeError:
                 pass  # fallback to code block extraction
 
-            # Case 3: Extract code blocks with any tag: json, py, python, or untagged
-            code_blocks = re.findall(r"```(?:json|py|python)?\n(.*?)```", raw_output, re.DOTALL)
+            # Case 3: Extract code blocks and search for JSON-like structures
+            code_blocks = re.findall(r"```(?:json|py|python)?\n([\s\S]*?)```", raw_output, re.DOTALL)
 
             for block in code_blocks:
-                # Try loading the block directly as JSON first
-                try:
-                    return json.loads(block)
-                except json.JSONDecodeError:
-                    pass  # continue with patterns
-
-                # Then look for common assignment patterns
                 for pattern in [
                     r"print\(\s*json\.dumps\(\s*(\{[\s\S]*?\})\s*\)\s*\)",
                     r"json\.dumps\(\s*(\{[\s\S]*?\})\s*\)",
@@ -151,6 +144,7 @@ def extract_json_from_codeagent_output(raw_output):
 
 
 
+
 import pandas as pd
 import tempfile
 
@@ -217,7 +211,7 @@ def analyze_data(csv_file, additional_notes=""):
     """, additional_args={"additional_notes": additional_notes, "source_file": cleaned_csv_path})
 
 
-
+    print(f"Raw output: {raw_output[:1000]}")  # To print first 1000 chars
 
     # Parse output
     parsed_result = extract_json_from_codeagent_output(raw_output) or {