diff --git a/custom_nodes/example_node.py.example b/custom_nodes/example_node.py.example index 29ab2aa72..68821a3b3 100644 --- a/custom_nodes/example_node.py.example +++ b/custom_nodes/example_node.py.example @@ -21,6 +21,8 @@ class Example: If this node is an output node that outputs a result/image from the graph. The SaveImage node is an example. The backend iterates on these output nodes and tries to execute all their parents if their parent graph is properly connected. Assumed to be False if not present. + DISPLAY_NAME (`str`): + A friendly/humanly readable title for the node. CATEGORY (`str`): The category the node should appear in the UI. DEPRECATED (`bool`): @@ -88,6 +90,7 @@ class Example: #OUTPUT_NODE = False + DISPLAY_NAME = "Example Node" CATEGORY = "Example" def check_lazy_status(self, image, string_field, int_field, float_field, print_to_screen): @@ -150,6 +153,7 @@ NODE_CLASS_MAPPINGS = { } # A dictionary that contains the friendly/humanly readable titles for the nodes +# NOTE: This is the same as using the DISPLAY_NAME property NODE_DISPLAY_NAME_MAPPINGS = { "Example": "Example Node" } diff --git a/server.py b/server.py index 62667ce18..a31559df4 100644 --- a/server.py +++ b/server.py @@ -561,7 +561,7 @@ class PromptServer(): info['output_is_list'] = obj_class.OUTPUT_IS_LIST if hasattr(obj_class, 'OUTPUT_IS_LIST') else [False] * len(obj_class.RETURN_TYPES) info['output_name'] = obj_class.RETURN_NAMES if hasattr(obj_class, 'RETURN_NAMES') else info['output'] info['name'] = node_class - info['display_name'] = nodes.NODE_DISPLAY_NAME_MAPPINGS[node_class] if node_class in nodes.NODE_DISPLAY_NAME_MAPPINGS.keys() else node_class + info['display_name'] = obj_class.DISPLAY_NAME if hasattr(obj_class,'DISPLAY_NAME') else nodes.NODE_DISPLAY_NAME_MAPPINGS[node_class] if node_class in nodes.NODE_DISPLAY_NAME_MAPPINGS.keys() else node_class info['description'] = obj_class.DESCRIPTION if hasattr(obj_class,'DESCRIPTION') else '' info['python_module'] = getattr(obj_class, "RELATIVE_PYTHON_MODULE", "nodes") info['category'] = 'sd'