Update app.py

app.py

@@ -435,6 +435,21 @@ def extract_json_from_llm_response(raw_response: str) -> dict:
         logger.error("Sanitized JSON still invalid:\n%s", json_string)
         raise
 
+def clean_base64_for_model(raw_b64: str) -> str:
+    if not raw_b64:
+        return ""
+    # Remove prefix if exists
+    clean_b64 = re.sub(r"^data:image\/[a-zA-Z]+;base64,", "", raw_b64)
+    # Remove spaces/newlines
+    clean_b64 = clean_b64.replace("\n", "").replace("\r", "").strip()
+    # Validate
+    try:
+        base64.b64decode(clean_b64)
+    except Exception as e:
+        logger.error(f"Invalid Base64 passed to model: {e}")
+        raise
+    return f"data:image/png;base64,{clean_b64}"
+
 # Node 1: Logic updating if any issue here
 def pseudo_generator_node(state: GameState):
     logger.info("--- Running plan_logic_aligner_node ---")
@@ -598,7 +613,8 @@ If you find any "Code-Blocks" then,
     image_input = {
         "type": "image_url",
         "image_url": {
-            "url": f"data:image/png;base64,{image}"
+            # "url": f"data:image/png;base64,{image}"
+            "url": clean_base64_for_model(image)
         }
     }