From b8b6657801938d4e73b99b190185ef6bdfd77301 Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Thu, 17 Oct 2024 23:38:46 +0800 Subject: [PATCH] fix input image bug --- comfy-nodes/input_image.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py index df72f6b..4fdf744 100755 --- a/comfy-nodes/input_image.py +++ b/comfy-nodes/input_image.py @@ -20,8 +20,8 @@ class ShellAgentPluginInputImage: {"multiline": False, "default": "input_image", "forceInput": False}, ), "default_value": ( - "STRING", {"image_upload": True, "default": files[0] if len(files) else ""}, - # sorted(files), {"image_upload": True, "forceInput": False} + # "STRING", {"image_upload": True, "default": files[0] if len(files) else ""}, + sorted(files), {"image_upload": True, "forceInput": False} ), }, "optional": { @@ -49,6 +49,16 @@ class ShellAgentPluginInputImage: "url_type": "image" } return schema + + @classmethod + def VALIDATE_INPUTS(s, image): + if image.startswith("http"): + return True + + if not folder_paths.exists_annotated_filepath(image): + return "Invalid image file: {}".format(image) + + return True def run(self, input_name, default_value=None, display_name=None, description=None): input_dir = folder_paths.get_input_directory()