Update utils.py

utils.py

@@ -4,10 +4,8 @@
 def load_prompt(file_path: str) -> str:
     """
     Loads a prompt from a specified text file.
-    We use this to keep prompts separate from code.
     """
     try:
-        # We specify 'encoding="utf-8"' to ensure all characters load correctly
         with open(file_path, "r", encoding="utf-8") as f:
             return f.read()
     except FileNotFoundError:
@@ -15,4 +13,18 @@ def load_prompt(file_path: str) -> str:
         return f"ERROR: Prompt file not found: {file_path}"
     except Exception as e:
         print(f"FATAL ERROR: Failed to read prompt file {file_path}: {e}")
-        return f"ERROR: Failed to read prompt file: {e}"
+        return f"ERROR: Failed to read prompt file: {e}"
+
+def extract_json_str(text: str) -> str:
+    """
+    Extracts the content inside a ```json ... ``` block.
+    If no block is found, returns the original text.
+    Used to clean 'Chatty' LLM responses before processing.
+    """
+    clean_text = text.strip()
+    if "```json" in clean_text:
+        # Split by ```json, take the second part, then split by ``` and take the first part
+        return clean_text.split("```json")[1].split("```")[0].strip()
+    elif "```" in clean_text:
+        return clean_text.split("```")[1].split("```")[0].strip()
+    return clean_text