Update utils.py

utils.py

@@ -5,6 +5,17 @@ import json
 from PIL import Image
 
 
+def extract_json_content(text):
+    matches = re.search(r"```json\n(.*?)\n```", text, re.DOTALL)
+    if matches:
+        text = matches.group(1).strip()
+    else:
+        partial_md_match = re.search(r"```json\n(.*)", text, re.DOTALL)
+        if partial_md_match:
+            text = partial_md_match.group(1).strip()
+    return text
+
+
 def truncate_last_incomplete_element(text: str):
     """Truncates the last incomplete element"""