Update handler.py

handler.py

@@ -37,6 +37,28 @@ class EndpointHandler():
         self.generator = torch.Generator(device="cpu").manual_seed(3)
 
     def __call__(self, data: Any) -> Dict[str, str]:
+        # Example JSON payload for testing
+        example_payload = {
+            "prompt": "a beautiful landscape",
+            "negative_prompt": "blur",
+            "width": 1024,
+            "height": 1024,
+            "steps": 30,
+            "cfg_scale": 7,
+            "alwayson_scripts": {
+                "controlnet": {
+                    "args": [
+                        {
+                            "enabled": True,
+                            "input_image": "image in base64",
+                            "model": "control_sd15_depth [fef5e48e]",
+                            "control_mode": "Balanced"
+                        }
+                    ]
+                }
+            }
+        }
+
         # Extract parameters from the payload
         prompt = data.get("prompt", None)
         negative_prompt = data.get("negative_prompt", None)
@@ -45,8 +67,6 @@ class EndpointHandler():
         num_inference_steps = data.get("steps", 30)
         guidance_scale = data.get("cfg_scale", 7)
         
-        
-       
         # Extract controlnet configuration from payload
         controlnet_config = data.get("alwayson_scripts", {}).get("controlnet", {}).get("args", [{}])[0]
 
@@ -61,7 +81,6 @@ class EndpointHandler():
             width=width,
             controlnet_conditioning_scale=1.0,
             generator=self.generator,
-        
         )
 
         # Get the generated image