From 1f16525118f92c8375b4074adc2538c9a8645132 Mon Sep 17 00:00:00 2001 From: Chengwei Ouyang Date: Wed, 12 Mar 2025 20:35:19 +0800 Subject: [PATCH] update --- comfy-nodes/input_image.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py index 986054f..3399a4f 100755 --- a/comfy-nodes/input_image.py +++ b/comfy-nodes/input_image.py @@ -78,21 +78,20 @@ class ShellAgentPluginInputImage: return schema @classmethod - def VALIDATE_INPUTS(s, input_name, default_value, image=None, description=""): - # check default_value first - image_to_check = default_value if default_value else image + def VALIDATE_INPUTS(s, input_name, default_value, description=""): + image = default_value - if image_to_check.startswith("http"): + if image.startswith("http"): return True - if image_to_check == "": + if image == "": return "Invalid image file: please check if the image is empty or invalid" - if os.path.isfile(image_to_check): + if os.path.isfile(image): return True - if not folder_paths.exists_annotated_filepath(image_to_check): - return "Invalid image file: {}".format(image_to_check) + if not folder_paths.exists_annotated_filepath(image): + return "Invalid image file: {}".format(image) return True @@ -137,9 +136,8 @@ class ShellAgentPluginInputImage: return (output_image, output_mask) - def run(self, input_name, default_value=None, image=None, display_name=None, description=None): - # use default_value if it exists, otherwise use image - image_path = default_value if default_value else image + def run(self, input_name, default_value=None, display_name=None, description=None): + image_path = default_value input_dir = folder_paths.get_input_directory() try: if image_path.startswith('http'):