Update
Changed plugin name for more clarity
This commit is contained in:
@@ -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",
|
||||
}
|
||||
@@ -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",
|
||||
}
|
||||
@@ -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",
|
||||
}
|
||||
+4
-4
@@ -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",
|
||||
}
|
||||
@@ -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",
|
||||
}
|
||||
Reference in New Issue
Block a user