diff --git a/README.md b/README.md index bf0e5c7..453df45 100644 --- a/README.md +++ b/README.md @@ -9,7 +9,7 @@ You can contact me on my discord server: https://discord.gg/Z44Zjpurjp ![image](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/b7d4a51d-6f23-43cc-bc40-c8ae469199e2) -## 1. DRMN_CaptionVisualizer +## 1. DATASET_CaptionVisualizer This node helps anyone to visualize captions in 3 different forms: word cloud, networkgraph and frequency graph. @@ -42,7 +42,7 @@ Images are also saved in ```visualize``` folder where the caption files are stor ![image](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/3371d531-ad4c-4e72-ae00-43d71e4a87a6) -## 2. DRMN_TXTFileSaver +## 2. DATASET_TXTFileSaver Saves text file in an innovative way. has mode Merge | Overwrite | SaveNew | MergeAndSaveNew. Useful for caption creators and manipulators. @@ -78,7 +78,7 @@ If directory does not exists new directory will be created. Text file in directory. -## 3. DRMN_TagManipulatorByImageNames +## 3. DATASET_TagManipulatorByImageNames There may be a possiblity where you named your caption file with the same name as your image file (without extension 😅). This node is useful then. You can add a concept in all the caption files at once. @@ -145,7 +145,7 @@ Tags directory will have something like this. ![image](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/15d27de2-6b99-4a32-92c4-3796ff4ae4d1) -## 3. DRMN_SearchAndReplace +## 3. DATASET_SearchAndReplace Helps anyone manipulate caption files to search for a given text in ```SearchFor``` and replace it with the given ```ReplaceWith``` text. diff --git a/__init__.py b/__init__.py index 395e68f..6aa074c 100644 --- a/__init__.py +++ b/__init__.py @@ -13,16 +13,16 @@ """ @author: Daxton Caylor -@title: ComfyUI-DRMN -@nickname: ComfyUI-DRMN +@title: ComfyUI-DataSet +@nickname: ComfyUI-DataSet @description: Data Research And Manipulators Nodes for Model Trainers, Artists, Designers and Animators. """ -from .classes.DRMN_TXTFileSaver import N_CLASS_MAPPINGS as TXTFileSaverMappings, N_DISPLAY_NAME_MAPPINGS as TXTFileSaverNameMappings -from .classes.DRMN_TagManipulatorByImageNames import N_CLASS_MAPPINGS as TagManipulatorByImageNamesMappings, N_DISPLAY_NAME_MAPPINGS as TagManipulatorByImageNamesNameMappings -from .classes.DRMN_CaptionVisualizer import N_CLASS_MAPPINGS as CaptionVisualizerMappings, N_DISPLAY_NAME_MAPPINGS as CaptionVisualizerNameMappings -from .classes.DRMN_SearchAndReplace import N_CLASS_MAPPINGS as SearchAndReplaceMappings, N_DISPLAY_NAME_MAPPINGS as SearchAndReplaceNameMappings -from .classes.DRMN_xCopy import N_CLASS_MAPPINGS as xCopyMappings, N_DISPLAY_NAME_MAPPINGS as xCopyNameMappings +from .classes.DATASET_TXTFileSaver import N_CLASS_MAPPINGS as TXTFileSaverMappings, N_DISPLAY_NAME_MAPPINGS as TXTFileSaverNameMappings +from .classes.DATASET_TagManipulatorByImageNames import N_CLASS_MAPPINGS as TagManipulatorByImageNamesMappings, N_DISPLAY_NAME_MAPPINGS as TagManipulatorByImageNamesNameMappings +from .classes.DATASET_CaptionVisualizer import N_CLASS_MAPPINGS as CaptionVisualizerMappings, N_DISPLAY_NAME_MAPPINGS as CaptionVisualizerNameMappings +from .classes.DATASET_SearchAndReplace import N_CLASS_MAPPINGS as SearchAndReplaceMappings, N_DISPLAY_NAME_MAPPINGS as SearchAndReplaceNameMappings +from .classes.DATASET_xCopy import N_CLASS_MAPPINGS as xCopyMappings, N_DISPLAY_NAME_MAPPINGS as xCopyNameMappings NODE_CLASS_MAPPINGS = {} NODE_CLASS_MAPPINGS.update(TXTFileSaverMappings) diff --git a/classes/DRMN_CaptionVisualizer.py b/classes/DATASET_CaptionVisualizer.py similarity index 97% rename from classes/DRMN_CaptionVisualizer.py rename to classes/DATASET_CaptionVisualizer.py index 5b2e518..00dc365 100644 --- a/classes/DRMN_CaptionVisualizer.py +++ b/classes/DATASET_CaptionVisualizer.py @@ -142,7 +142,7 @@ def create_empty_image(width=100, height=100, color=(255, 255, 255)): return empty_image -class DRMN_CaptionVisualizer: +class DATASET_CaptionVisualizer: def __init__(self): pass @@ -164,7 +164,7 @@ class DRMN_CaptionVisualizer: OUTPUT_IS_LIST = (True,) FUNCTION = "Visualize" OUTPUT_NODE = True - CATEGORY = "🔶DRMN🔶" + CATEGORY = "🔶DATASET🔶" def Visualize(self, TextFilePathList, WordCloudTop, NetworkGraphTop, FrequencyGraphTop): @@ -183,9 +183,9 @@ class DRMN_CaptionVisualizer: N_CLASS_MAPPINGS = { - "DRMN_CaptionVisualizer": DRMN_CaptionVisualizer, + "DATASET_CaptionVisualizer": DATASET_CaptionVisualizer, } N_DISPLAY_NAME_MAPPINGS = { - "DRMN_CaptionVisualizer": "DRMN_CaptionVisualizer", + "DATASET_CaptionVisualizer": "DATASET_CaptionVisualizer", } diff --git a/classes/DRMN_SearchAndReplace.py b/classes/DATASET_SearchAndReplace.py similarity index 90% rename from classes/DRMN_SearchAndReplace.py rename to classes/DATASET_SearchAndReplace.py index ce96e46..3835afc 100644 --- a/classes/DRMN_SearchAndReplace.py +++ b/classes/DATASET_SearchAndReplace.py @@ -20,7 +20,7 @@ def save_file(file_path, content): print(f"File saved successfully at {file_path}") -class DRMN_SearchAndReplace: +class DATASET_SearchAndReplace: def __init__(self): pass @@ -40,7 +40,7 @@ class DRMN_SearchAndReplace: RETURN_NAMES = () FUNCTION = "SaR" OUTPUT_NODE = True - CATEGORY = "🔶DRMN🔶" + CATEGORY = "🔶DATASET🔶" def SaR(self, TextFilePathList, SearchFor, ReplaceWith): @@ -70,9 +70,9 @@ class DRMN_SearchAndReplace: N_CLASS_MAPPINGS = { - "DRMN_SearchAndReplace": DRMN_SearchAndReplace, + "DATASET_SearchAndReplace": DATASET_SearchAndReplace, } N_DISPLAY_NAME_MAPPINGS = { - "DRMN_SearchAndReplace": "DRMN_SearchAndReplace", + "DATASET_SearchAndReplace": "DATASET_SearchAndReplace", } diff --git a/classes/DRMN_TXTFileSaver.py b/classes/DATASET_TXTFileSaver.py similarity index 92% rename from classes/DRMN_TXTFileSaver.py rename to classes/DATASET_TXTFileSaver.py index 96d7a98..66a29fe 100644 --- a/classes/DRMN_TXTFileSaver.py +++ b/classes/DATASET_TXTFileSaver.py @@ -29,7 +29,7 @@ def save_file(filename, output_dir, content, mode='SaveNew'): file.write(content) print(f"File saved successfully at {file_path}") -class DRMN_TXTFileSaver: +class DATASET_TXTFileSaver: def __init__(self): pass @@ -50,7 +50,7 @@ class DRMN_TXTFileSaver: FUNCTION = "SaveIT" OUTPUT_NODE = True - CATEGORY = "🔶DRMN🔶" + CATEGORY = "🔶DATASET🔶" def SaveIT(self, content, file_name, directory, mode): try: @@ -67,9 +67,9 @@ class DRMN_TXTFileSaver: N_CLASS_MAPPINGS = { - "DRMN_TXTFileSaver": DRMN_TXTFileSaver, + "DATASET_TXTFileSaver": DATASET_TXTFileSaver, } N_DISPLAY_NAME_MAPPINGS = { - "DRMN_TXTFileSaver": "DRMN_TXTFileSaver", + "DATASET_TXTFileSaver": "DATASET_TXTFileSaver", } diff --git a/classes/DRMN_TagManipulatorByImageNames.py b/classes/DATASET_TagManipulatorByImageNames.py similarity index 94% rename from classes/DRMN_TagManipulatorByImageNames.py rename to classes/DATASET_TagManipulatorByImageNames.py index 675c483..fd832bd 100644 --- a/classes/DRMN_TagManipulatorByImageNames.py +++ b/classes/DATASET_TagManipulatorByImageNames.py @@ -88,7 +88,7 @@ def read_files_from_directory(file_names, directory): return file_contents -class DRMN_TagManipulatorByImageNames: +class DATASET_TagManipulatorByImageNames: def __init__(self): pass @@ -109,7 +109,7 @@ class DRMN_TagManipulatorByImageNames: FUNCTION = "SaveIT" OUTPUT_NODE = True - CATEGORY = "🔶DRMN🔶" + CATEGORY = "🔶DATASET🔶" def SaveIT(self, ImageNames, TagsDirectory, Captions, Backup): try: @@ -136,9 +136,9 @@ class DRMN_TagManipulatorByImageNames: N_CLASS_MAPPINGS = { - "DRMN_TagManipulatorByImageNames": DRMN_TagManipulatorByImageNames, + "DATASET_TagManipulatorByImageNames": DATASET_TagManipulatorByImageNames, } N_DISPLAY_NAME_MAPPINGS = { - "DRMN_TagManipulatorByImageNames": "DRMN_TagManipulatorByImageNames", + "DATASET_TagManipulatorByImageNames": "DATASET_TagManipulatorByImageNames", } diff --git a/classes/DRMN_xCopy.py b/classes/DATASET_xCopy.py similarity index 93% rename from classes/DRMN_xCopy.py rename to classes/DATASET_xCopy.py index b5d78f7..c099cbf 100644 --- a/classes/DRMN_xCopy.py +++ b/classes/DATASET_xCopy.py @@ -19,7 +19,7 @@ def copy_by_destination_files(source, destination): if dest_file_match and filename not in dest_files: shutil.copy(source_file, destination) -class DRMN_xCopy: +class DATASET_xCopy: def __init__(self): pass @@ -39,7 +39,7 @@ class DRMN_xCopy: FUNCTION = "SaveIT" OUTPUT_NODE = True - CATEGORY = "🔶DRMN🔶" + CATEGORY = "🔶DATASET🔶" def SaveIT(self, source_folder, destination_folder, mode): try: @@ -56,9 +56,9 @@ class DRMN_xCopy: N_CLASS_MAPPINGS = { - "DRMN_xCopy": DRMN_xCopy, + "DATASET_xCopy": DATASET_xCopy, } N_DISPLAY_NAME_MAPPINGS = { - "DRMN_xCopy": "DRMN_xCopy", + "DATASET_xCopy": "DATASET_xCopy", } diff --git a/images/logo.jpg b/images/logo.jpg new file mode 100644 index 0000000..7b14e94 Binary files /dev/null and b/images/logo.jpg differ diff --git a/images/logo.png b/images/logo.png deleted file mode 100644 index 3424c31..0000000 Binary files a/images/logo.png and /dev/null differ