Update handler.py

handler.py

@@ -41,11 +41,27 @@ class EndpointHandler:
     
     def __call__(self, data: Dict[str, Any]) -> str:
         try:
-            inputs = data.get("inputs")
-            base64_string = inputs["base64"]
-              
+            base64_string = None
+            if "inputs" in data and isinstance(data["inputs"], str):
+                base64_string = data["inputs"]
+            
+            # Case 2: Base64 in nested inputs dictionary
+            elif "inputs" in data and isinstance(data["inputs"], dict):
+                base64_string = data["inputs"].get("base64")
+            
+            # Case 3: Direct base64 at root level
+            elif "base64" in data:
+                base64_string = data["base64"]
+            
+            # Case 4: Try raw data as base64
+            elif isinstance(data, str):
+                base64_string = data
+            
             if not base64_string:
-                return {"error": "No base64 string provided"}
+                return {"error": "No base64 string found in input data. Available keys: " + str(data.keys())}
+            
+            print("Found base64 string, length:", len(base64_string))
+              
             # Remove data URL prefix if present
             if ',' in base64_string:
                 base64_string = base64_string.split(',')[1]