From 7238c1f40bdb9e2a22a4fe45e3e186dc57d97f06 Mon Sep 17 00:00:00 2001 From: wl-zhao Date: Fri, 13 Dec 2024 11:45:43 +0800 Subject: [PATCH] fix error message when empty input image --- comfy-nodes/input_image.py | 6 +++++- dependency_checker.py | 5 ++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py index ebfe847..65fb2c2 100755 --- a/comfy-nodes/input_image.py +++ b/comfy-nodes/input_image.py @@ -55,10 +55,14 @@ class ShellAgentPluginInputImage: @classmethod def VALIDATE_INPUTS(s, input_name, default_value, description=""): image = default_value + if image.startswith("http"): return True - if not folder_paths.exists_annotated_filepath(image): + if not os.path.isfile(image): + return "Invalid image file: please check if the image is empty or invalid" + + if folder_paths.exists_annotated_filepath(image): return "Invalid image file: {}".format(image) return True diff --git a/dependency_checker.py b/dependency_checker.py index 1e73e05..7aa176d 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -283,7 +283,10 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an print(f"failed to resolve repo info of {custom_node}") requirement_file = os.path.join(BASE_PATH, custom_node.replace(".", "/"), "requirements.txt") if os.path.isfile(requirement_file): - requirements_lines += open(requirement_file).readlines() + try: + requirements_lines += open(requirement_file).readlines() + except: + pass requirements_lines = list(set(requirements_lines)) requirements_packages = [package_name for package_name, version_specifier in map(split_package_version, requirements_lines) if package_name is not None] package_names = set(requirements_packages + extra_packages)