From 9dfc92ffbae09d6ee370da7aa09d77991dd5047e Mon Sep 17 00:00:00 2001
From: Yiheng <yihengmen@gmail.com>
Date: Thu, 30 May 2024 15:03:39 +0800
Subject: [PATCH] feat: return success

---
 image_c_postprocessor.py | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/image_c_postprocessor.py b/image_c_postprocessor.py
index 00280ff..9bbf47f 100644
--- a/image_c_postprocessor.py
+++ b/image_c_postprocessor.py
@@ -28,8 +28,8 @@ class ImageCPostprocessor:
             },
         }
 
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("image",)
+    RETURN_TYPES = ("IMAGE", "BOOLEAN")
+    RETURN_NAMES = ("image", "success")
     FUNCTION = "apply_postprocess"
     CATEGORY = "Tools"
 
@@ -48,12 +48,10 @@ class ImageCPostprocessor:
         # 使用 RETR_EXTERNAL 模式找到所有轮廓
         contours, hierarchy = cv2.findContours(
             mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-        if hierarchy is None:
-            print("No contours found.")
-            return None
+
         if len(contours) < 2:
-            print("Alpha channel not good")
-            return input_image
+            print("No contours found.")
+            return (input_image, False)
         # 遍历每个轮廓，处理
         for contour in contours:
             area = cv2.contourArea(contour)
@@ -102,7 +100,7 @@ class ImageCPostprocessor:
         # 将 numpy 数组转回 PIL 图片
         processed_image = Image.fromarray(original_image)
 
-        return processed_image
+        return (processed_image, True)
 
     def apply_postprocess(self, image, kernel_size):
         tensors = []