fix error message when empty input image

This commit is contained in:
wl-zhao
2024-12-13 11:45:43 +08:00
parent 623ea454cc
commit 7238c1f40b
2 changed files with 9 additions and 2 deletions
+5 -1
View File
@@ -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
+4 -1
View File
@@ -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)