Update

Changed plugin name for more clarity

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.
 

__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)

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",
 }

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",
 }

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",
 }

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",
 }

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",
 }