diff --git a/__init__.py b/__init__.py index 19a1886..0ece481 100644 --- a/__init__.py +++ b/__init__.py @@ -1,23 +1,33 @@ -from .nodes.save_string import SaveString -from .nodes.print_string import PrintString -from .nodes.replace_string import ReplaceString +from .nodes.string_save import StringSave +from .nodes.string_print import StringPrint +from .nodes.string_replace import StringReplace +from .nodes.string_node import StringNode +from .nodes.string_concat import StringConcat from .nodes.tokenizer_vocab import TokenizerVocab -from .nodes.pick_item_json import PickItemJson +from .nodes.json_pick_item import JsonPickItem NODE_CLASS_MAPPINGS = { - "IW_SaveString": SaveString, - "IW_PrintString": PrintString, - "IW_ReplaceString": ReplaceString, + "IW_StringSave": StringSave, + "IW_StringPrint": StringPrint, + "IW_StringReplace": StringReplace, + "IW_StringNode": StringNode, + "IW_StringConcat": StringConcat, + "IW_TokenizerVocab": TokenizerVocab, - "IW_PickItemJson": PickItemJson, + + "IW_JsonPickItem": JsonPickItem, } # A dictionary that contains the friendly/humanly readable titles for the nodes NODE_DISPLAY_NAME_MAPPINGS = { - "IW_SaveString": "IW SaveString", + "IW_StringSave": "IW SaveString", + "IW_StringPrint": "IW PrintString", + "IW_StringReplace": "IW ReplaceString", + "IW_StringNode": "IW StringNode", + "IW_StringConcat": "IW StringConcat", + "IW_TokenizerVocab": "IW TokenizerVocab", - "IW_PickItemJson": "IW PickItemJson", - "IW_PrintString": "IW PrintString", - "IW_ReplaceString": "IW ReplaceString", + + "IW_JsonPickItem": "IW JsonPickItem", } diff --git a/nodes/pick_item_json.py b/nodes/json_pick_item.py similarity index 79% rename from nodes/pick_item_json.py rename to nodes/json_pick_item.py index e8b0392..e4a739e 100644 --- a/nodes/pick_item_json.py +++ b/nodes/json_pick_item.py @@ -1,7 +1,7 @@ import os import json -class PickItemJson: +class JsonPickItem: def __init__(self): pass @@ -10,7 +10,7 @@ class PickItemJson: return { "required": { "pick_value": ("INT",{"defaultInput": True, "default": 0}), - "input": ("STRING",{"defaultInput": True, "default": "", "dynamicPrompts": False, "multiline": True}), + "input": ("STRING",{"defaultInput": False, "default": "", "dynamicPrompts": False, "multiline": True}), }, "optional": { }, diff --git a/nodes/string_concat.py b/nodes/string_concat.py new file mode 100644 index 0000000..21f7630 --- /dev/null +++ b/nodes/string_concat.py @@ -0,0 +1,25 @@ +import os +import json + +class StringConcat: + def __init__(self): + pass + + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "prefix": ("STRING",{"defaultInput": True, "default": "", "multiline": True, "dynamicPrompts": False}), + "suffix": ("STRING",{"defaultInput": True, "default": "", "multiline": True, "dynamicPrompts": False, "hidden": True}), + }, + "optional": { + }, + } + + RETURN_TYPES = ("STRING", ) + FUNCTION = "execute" + + CATEGORY = "iw" + + def execute(self, prefix, suffix): + return (prefix + suffix,) diff --git a/nodes/string_node.py b/nodes/string_node.py new file mode 100644 index 0000000..2d895ca --- /dev/null +++ b/nodes/string_node.py @@ -0,0 +1,24 @@ +import os +import json + +class StringNode: + def __init__(self): + pass + + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "input": ("STRING",{"defaultInput": False, "default": "", "multiline": True, "dynamicPrompts": False}), + }, + "optional": { + }, + } + + RETURN_TYPES = ("STRING", ) + FUNCTION = "execute" + + CATEGORY = "iw" + + def execute(self, input): + return (input,) diff --git a/nodes/print_string.py b/nodes/string_print.py similarity index 97% rename from nodes/print_string.py rename to nodes/string_print.py index 7f21072..c362f0c 100644 --- a/nodes/print_string.py +++ b/nodes/string_print.py @@ -1,5 +1,5 @@ -class PrintString: +class StringPrint: def __init__(self): pass diff --git a/nodes/replace_string.py b/nodes/string_replace.py similarity index 97% rename from nodes/replace_string.py rename to nodes/string_replace.py index f7c7346..e92d0f6 100644 --- a/nodes/replace_string.py +++ b/nodes/string_replace.py @@ -1,7 +1,7 @@ import os import json -class ReplaceString: +class StringReplace: def __init__(self): pass diff --git a/nodes/save_string.py b/nodes/string_save.py similarity index 98% rename from nodes/save_string.py rename to nodes/string_save.py index ce1f75c..356dcb1 100644 --- a/nodes/save_string.py +++ b/nodes/string_save.py @@ -2,7 +2,7 @@ import folder_paths import os import json -class SaveString: +class StringSave: def __init__(self): self.output_dir = folder_paths.get_output_directory() pass