replace json string cleaning function

app.py

@@ -67,15 +67,33 @@ HEADER = """
 """
 
 def clean_json_string(json_str):
-    # Replace single quotes with double quotes, but not apostrophes within words
-    json_str = re.sub(r"(?<!\\)'([^']*)'", r'"\1"', json_str)
-    # Add quotes around PASS or FAIL if they're not already quoted
-    json_str = re.sub(r'"SCORE":\s*(PASS|FAIL)', r'"SCORE": "\1"', json_str)
+    # # Replace single quotes with double quotes, but not apostrophes within words
+    # json_str = re.sub(r"(?<!\\)'([^']*)'", r'"\1"', json_str)
+    # # Add quotes around PASS or FAIL if they're not already quoted
+    # json_str = re.sub(r'"SCORE":\s*(PASS|FAIL)', r'"SCORE": "\1"', json_str)
 
-    # Escape double quotes
-    json_str = re.sub(r'(?<=: ")([^"]*)"([^"]*)"([^"]*)"', lambda m: m.group(1) + m.group(2).replace('"', '\\"') + m.group(3), json_str)
+    # # Escape double quotes
+    # json_str = re.sub(r'(?<=: ")([^"]*)"([^"]*)"([^"]*)"', lambda m: m.group(1) + m.group(2).replace('"', '\\"') + m.group(3), json_str)
     
-    return json_str
+    # return json_str
+    
+    json_str = json_str.replace("'", '"')
+    json_str = re.sub(r'"SCORE":\s*(FAIL|PASS)', r'"SCORE": "\1"', json_str)
+    
+    def escape_quotes(match):
+        return match.group(0).replace('"', '\\"')
+    
+    pattern = r'"([^"\\]*(?:\\.[^"\\]*)*)"'
+    json_str = re.sub(pattern, lambda m: '"' + escape_quotes(m) + '"', json_str)
+    
+    try:
+        parsed_json = json.loads(json_str)
+        return parsed_json
+    except json.JSONDecodeError as e:
+        print(f"Error parsing JSON: {e}")
+        print("Cleaned JSON string:")
+        print(json_str)
+        return None
 
 def model_call(question, document, answer):
     if question == "" or document == "" or answer == "":