Input image data format

handler.py

@@ -33,16 +33,29 @@ class EndpointHandler():
         # Handle different input formats
         if "inputs" in data:
             if isinstance(data["inputs"], str):
+                # Handle data URL format (data:image/jpeg;base64,...)
+                image_data = data["inputs"]
+                if image_data.startswith("data:"):
+                    # Strip data URL prefix
+                    image_data = image_data.split(",", 1)[1]
                 # Base64 encoded image
-                image_bytes = base64.b64decode(data["inputs"])
+                image_bytes = base64.b64decode(image_data)
             elif isinstance(data["inputs"], dict) and "image" in data["inputs"]:
                 # Nested structure with image field
-                image_bytes = base64.b64decode(data["inputs"]["image"])
+                image_data = data["inputs"]["image"]
+                if image_data.startswith("data:"):
+                    # Strip data URL prefix
+                    image_data = image_data.split(",", 1)[1]
+                image_bytes = base64.b64decode(image_data)
             else:
                 raise ValueError("Invalid input format. Expected base64 encoded image string.")
         elif "image" in data:
             # Direct image field
-            image_bytes = base64.b64decode(data["image"])
+            image_data = data["image"]
+            if image_data.startswith("data:"):
+                # Strip data URL prefix
+                image_data = image_data.split(",", 1)[1]
+            image_bytes = base64.b64decode(image_data)
         else:
             raise ValueError("No image found in request. Expected 'inputs' or 'image' field with base64 encoded image.")