fix

handler.py

@@ -23,14 +23,29 @@ class EndpointHandler:
         )
         self.pipe.to("cuda")
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, str]]:
-        # 获取控制图像（支持 URL 或二进制上传）
-        control = data.get("control_image")
-        if isinstance(control, str):
-            resp = requests.get(control)
-            control_image = Image.open(BytesIO(resp.content)).convert("RGB")
+    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+        # 兼容包装与非包装请求体
+        payload = data.get("inputs", data)  # 支持 UI 及直接 POST 两种格式 
+
+        # 读取图像字节
+        img_bytes = None
+        url = payload.get("control_image_url")
+        b64 = payload.get("control_image_base64")
+        if b64:
+            img_bytes = base64.b64decode(b64)
+        elif url:
+            resp = requests.get(url)
+            resp.raise_for_status()  # 捕获 4xx/5xx 错误 :contentReference[oaicite:3]{index=3}
+            img_bytes = resp.content
         else:
-            control_image = Image.open(BytesIO(control)).convert("RGB")
+            raise ValueError("请在 inputs 中提供 control_image_url 或 control_image_base64")
+
+        # 用 PIL 解析并转换
+        try:
+            control_image = Image.open(BytesIO(img_bytes)).convert("RGB")
+        except Exception as e:
+            # 可能是数据损坏或格式不符
+            raise ValueError(f"无法识别图像文件: {str(e)}")
 
         # 可选：调整尺寸（默认 4× 放大）
         w, h = control_image.size