Update app.py

app.py

@@ -143,6 +143,38 @@ def decode_and_save_image(image_b64, filename):
     image.save(full_path)
     return full_path
 
+# app.py  – drop this right above single_image_generation (or put in utils.py)
+import base64, binascii, json, os
+from PIL import Image, UnidentifiedImageError
+from io import BytesIO
+
+def safe_decode_image(image_b64: str, filename: str) -> str:
+    """
+    Decode a base-64 PNG/JPEG **only if** the payload is valid.
+    Returns the fully-qualified file path or raises ValueError.
+    """
+    # Bail fast if tool returned JSON / HTML / plain-text
+    txt = image_b64.lstrip()        # strip leading whitespace
+    if txt.startswith("{") or txt.startswith("<") or txt.startswith("Error"):
+        raise ValueError(f"Received non-image payload: {txt[:120]}…")
+
+    # Strict base-64 decode
+    try:
+        image_bytes = base64.b64decode(txt, validate=True)
+    except binascii.Error as e:
+        raise ValueError(f"Invalid base-64: {e}") from None
+
+    # Pillow decode
+    try:
+        img = Image.open(BytesIO(image_bytes))
+    except UnidentifiedImageError:
+        raise ValueError("Decoded bytes are not a valid image")
+
+    # Persist
+    os.makedirs("AI-Marketing-Content-Creator/created_image", exist_ok=True)
+    full_path = os.path.join("AI-Marketing-Content-Creator/created_image", filename)
+    img.save(full_path)
+    return full_path
 
 def single_image_generation(prompt, num_steps, style):
     """Generate a single image with optional style"""