Update hedis_engine.py

hedis_engine.py

@@ -130,14 +130,28 @@ class InteractiveHedisComplianceEngine:
             # Remove any leading/trailing whitespace and newlines
             result_str = result_str.strip()
             
+            # Remove surrounding quotes if the entire JSON is quoted
+            if len(result_str) >= 2 and result_str.startswith('"') and result_str.endswith('"'):
+                # Check if it's actually a quoted JSON string by trying to parse the unquoted version
+                try:
+                    unquoted = result_str[1:-1]
+                    # Handle escaped quotes in the JSON
+                    unquoted = unquoted.replace('\\"', '"').replace('\\n', '\n')
+                    test_parse = json.loads(unquoted)
+                    result_str = unquoted
+                except:
+                    # If unquoting fails, keep original
+                    pass
+            
             # Try to parse as JSON
             criteria_json = json.loads(result_str)
             return criteria_json
         except json.JSONDecodeError as e:
             print(f"JSON parsing error: {e}")
             print(f"Raw result: {repr(str(result))}")
+            print(f"Cleaned result: {repr(result_str)}")
             # If not valid JSON, return as string
-            return {"raw_output": str(result)}
+            return {"raw_output": str(result), "parse_error": str(e)}
 
     def display_criteria(self, criteria: dict) -> None:
         """Display the HEDIS criteria in a user-friendly format."""