![COMFY-UI (3)](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/998725d9-45c4-441d-901e-99833af93734) # ComfyUI-DataSet ![image](https://img.shields.io/badge/version-1.1.1-green) ![image](https://img.shields.io/badge/last_update-Jan_2025-green) Data research, preparation, and manipulation nodes for model trainers and artists.
##### Drag & drop image into your workspace for node layout ![COMFY-UI-DATASET](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/b2265e40-5596-4d8f-951b-b6ba20c428e6)
# Updates [**Nov 12th 2024**] - Now **ClaudeAIChat**, **GroqAIChat** and **OpenAIChat** works on **API KEYS** Defined in System Environment. - Keys names are: **OPENAI_API_KEY**, **GROQ_API_KEY**, **ANTHROPIC_API_KEY** ## Installation > **Note**: Please upgrade as this is a major update that renders all previous updates invalid. #### Using `comfy-cli` (https://github.com/yoland68/comfy-cli) - `comfy node registry-install ComfyUI-DataSet` - https://registry.comfy.org/publishers/daxcay/nodes/comfyui-dataset #### Manual Method - Go to your Comfyui > Custom Nodes folder path > Run CMD - Copy and Paste this command git clone ```https://github.com/daxcay/ComfyUI-DataSet.git``` - Then go inside ComfyUI-DataSet with cmd or open new. - type ```pip install -r requirements.txt``` to install the dependencies #### Automatic Method with [Comfy Manager](https://github.com/ltdrdata/ComfyUI-Manager) - Inside ComfyUI > Click Manager Button on Side. - Click `Custom Nodes Manager` and Search for `DataSet` and Install this node: ![image](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/796bf982-20fc-4859-8096-7138d207214e) - Restart ComfyUI and it should be good to go #### Recommended Plugin - **ComfyUI-JDCN** (https://github.com/daxcay/ComfyUI-JDCN)
### You can find DataSet under this category: ![image](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/4930b434-1993-47c4-bd71-d16f762f94a4)
## DataSet_Visualizer The `DataSet_Visualizer` node is designed to visualize dataset captions. It generates graphs offering various perspectives on token analysis. The word cloud represents token frequency with different sized fonts. The network graph illustrates the relationships between tokens. The frequency graph provides an exact metric of how often each token appears in your captions. ![Screenshot 2024-07-05 195633](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/3b6ef307-6d86-48e4-b9a0-d59610aeeb82) #### Inputs - **TextFileContents**(STRING, required): the contents of the text file to be processed. - **Seperator**(['comma', 'colon', 'space', 'pipe'], required): the delimiter used to separate tokens in the text file. - **WordCloudTop**(INT, min: 1, max: 9999, required): the number of top tokens to be plotted in WordCloud. - **NetworkGraphTop**(INT, min: 1, max: 9999, required): the number of top tokens having the highest interconnections within the captions. - **FrequencyGraphTop**(INT, min: 1, max: 9999, required): the number of top tokens with highest frequency from highest to lowest. #### Outputs - **GraphsPaths**(STRING, list): the file paths of the generated visualizations. It includes paths for: WordCloud image, NetworkGraph image, FrequencyTable image - **GraphsImages**(IMAGE, list): the generated images for the visualizations which can be used with PreviewImage and SaveImage node. #### Example
## DataSet_CopyFiles The `DataSet_CopyFiles` node provides a method to copy files from a source folder to a destination folder using different modes: `BlindCopy` and `CopyByDestinationFiles`. ![Screenshot 2024-07-05 195651](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/7176f6b8-8e88-4c64-bb78-cb53afe8e368) #### Inputs - **source_folder** (STRING, default: "directory path", required): source folder path to the files. - **destination_folder** (STRING, default: "directory path", required): destination folder path for the files copied. - **copy_mode** (['BlindCopy', 'CopyByDestinationFiles'], required): - `BlindCopy`: copies all files from source to the destination folder. - `CopyByDestinationFiles`: copies files from source folder to the destination only if there is a matching file (based on the base name) already present in the destination.
## DataSet_TriggerWords The `DataSet_TriggerWords` node is designed to extract triggerwords from captions. The node identifies triggerwords as tokens containing BOTH letters and numbers. ![Screenshot 2024-07-05 195619](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/1ab3f438-6b47-4054-8b99-107fdb04d1d4) #### Inputs - **TextFileContents** (`STRING`, required): the contents of the text file(s) to be processed. - **search**(`['trigger_word_only', 'trigger_word_phrase']`, required): - `'trigger_word_only'`: extracts individual triggerwords only - `'trigger_word_phrase'`: extracts entire phrase (contained within two comma's) which contains a triggerword #### Outputs - **Words** (`STRING`, list): the extracted triggerwords or triggerword-containing phrase
## DataSet_TextFilesLoadFromList The `DataSet_TextFilesLoad` node is designed to process the basic attributes of txt files. It can for instance extract filenames or filenames WITHOUT extensions, file-paths and file-contents. Useful for certain batched workflows. It takes a file directory path as input ![Screenshot 2024-07-05 204333](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/a674f684-85f3-4f34-8c3e-b1b34bf89bac) #### Inputs - **TextFilePathsList**(`STRING`, required): a list of file paths to the text files to be loaded. Only paths ending with `.txt` will be processed. #### Outputs - **TextFileNames**(`STRING`, list): the names of the text files. - **TextFileNamesWithoutExtension**(`STRING`, list): the names of the text files without their extensions. - **TextFilePaths**(`STRING`, list): the file paths of the text files. - **TextFileContents**(`STRING`, list): the contents of the text files.
## DataSet_TextFilesLoad Same as above, but uses a widget path to file directory for input ![Screenshot 2024-07-05 204321](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/a136dec1-38d1-454e-9f0e-014837428225) #### Inputs - **directory**(`STRING`, required): the directory path where the text files are located. The path should be specified as a string. #### Outputs - **TextFileNames**(`STRING`, list): the names of the text files in the directory. - **TextFileNamesWithoutExtension**(`STRING`, list): the names of the text files without their extensions. - **TextFilePaths**(`STRING`, list): the file paths of the text files in the directory. - **TextFileContents**(`STRING`, list): the contents of the text files in the directory.
## DataSet_TextFilesSave The `DataSet_TextFilesSave` node is designed to save text file contents to a specified directory. Supports the following modes: overwriting, merging, creating new files, and merging before saving new files DONT UNDERSTAND THIS. ![Screenshot 2024-07-05 195656](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/303fee7e-5b1a-4343-95a8-dfb8c9aebf6c) #### Inputs - **TextFileNames**(`STRING`, required): the names of the text files to be saved. - **TextFileContents**(`STRING`, required): the contents of the text files to be saved. - **destination**(`STRING`, required): the directory path where the text files will be saved. - **save_mode**(['Overwrite', 'Merge', 'SaveNew', 'MergeAndSaveNew'], required): the mode of saving the files: - `Overwrite`: overwrites existing files with the same name. - `Merge`: appends content to existing files with the same name. - `SaveNew`: saves new files with a unique name if a file with the same name already exists. - `MergeAndSaveNew`: merges content with existing files and then saves as a new file with a unique name if a file with the same name already exists.
## DataSet_FindAndReplace The `DataSet_FindAndReplace` node finds and replaces a text pattern within caption text files. ![Screenshot 2024-07-05 195639](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/12b9c2a7-54ea-4d01-ae2b-53898865ca7f) #### Inputs - **TextFileContents**(`STRING`, required): the text file contents to be processed - **SearchFor**(`STRING`, default: "search-text", required): the searched text pattern within the `TextFileContents`. Supports multiline input. - **ReplaceWith**(`STRING`, default: "replacement-text", required): the replacement text for the `SearchFor` pattern. Supports multiline input. #### Outputs - **TextFileContents**(`STRING`, list): the modified contents of the text files
## DataSet_PathSelector The `DataSet_PathSelector` is useful for identifying images in a sub-dataset which are missing caption text files from a larger parent repository of image-text pairings. The node will search for orphaned text/image files in one directory which require the missing pair files with matching names from another directory. ![Screenshot 2024-07-05 204328](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/181cca81-8a67-40a4-99ef-56c6dc8c0c76) #### Inputs - **search_in_directory**(`STRING`, required): the sub-dataset directory missing pairings - **search_for_extensions**(`STRING`, required): the extensions of the orphaned files, separated by commas (e.g., `.txt, .csv`). - **select_from_directory**(`STRING`, required): the repository directory containing the complete text-image pairings. - **select_extensions**(`STRING`, required): the extensions of the required files to be added, separated by commas (e.g., `.txt, .csv`). #### Outputs - **SelectedNamesWithExtension**(`STRING`, list): the names of the required files with their extensions. - **SelectedNamesWithoutExtension**(`STRING`, list): the names of the required files without their extensions. - **SelectedPaths**(`STRING`, list): the full paths of the required files.
## DataSet_ConceptManager The `DataSet_ConceptManager` node is designed to add/remove tokens within caption files, and it will allow you to place these tokens at designated positions within the caption ![Screenshot 2024-07-05 195624](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/de1c1dd8-b688-435c-96cd-3368b4f13162) #### Inputs - **TextFileContents**(`STRING`, required): the contents of the text file(s) to be processed. - **Mode**(`STRING`, required): the mode of operation: `'add'` to add tokens or `'remove'` to remove tokens. - **Concepts**(`STRING`, required): the concepts to add or remove, formatted as text + position (e.g., `"tag1 0, tag2 2"` for adding, `"tag1, tag2"` for removing). #### Outputs - **TextFileContents**(`STRING`, list): the modified contents of the caption file(s)
## DataSet_OpenAIChat The `DataSet_OpenAIChat` uses the OpenAI GPT chat to help you generate prompts. ![Screenshot 2024-07-05 195559](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/0876389a-6617-4974-9705-4240ddcf7545) #### Inputs - **model**(STRING, required): select the OpenAI model. Options include `"GPTo"`, `"gpt-3.5-turbo"`, etc. - **api_url**(STRING, default: `"https://api.openai.com/v1"`): the base URL for the API. - **api_key**(STRING, required): the API key for authentication. - **prompt**(STRING, default: ""): the query chat. Prompt GPT to generate prompts - **token_length**(INT, default: 1024): the maximum number of tokens (words). #### Outputs - **STRING**: the GPT generated new prompt
## DataSet_LoadImage The `DataSet_LoadImage` node provides essential image file attributes for captioning with the `DataSet_OpenAIChat` node. It leverages Pillow and Numpy libraries. ![Screenshot 2024-07-05 204341](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/9530778f-1998-4a37-8708-3a2ab562c6ff) #### Inputs - **image** (STRING, required): the name of the image file to load from the input directory. #### Outputs - **IMAGE**: the image file. - **MASK**: the mask associated with the image. - **STRING**: the name of the image file. - **STRING**: the name of the image file without extension. - **STRING**: the full path of the image file. - **STRING**: the directory path of the image file.
## DataSet_SaveImage The `DataSet_SaveImage` node batch saves images to a specified directory with optional PNG metadata. Also uses Pillow and Numpy. ![Screenshot 2024-07-05 195646](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/e7aed0b4-526a-4b74-94c7-bc2c417063d6) #### Inputs - **Images**(IMAGE, required): list of images to save. - **ImageFilePrefix**(STRING, default: "Image"): prefix for the saved image filenames. - **destination**(STRING): directory path where images will be saved.
## DataSet_OpenAIChatImage The `DataSet_OpenAIChat` uses the OpenAI GPTo multi-modal vision API in a chat framework, in order to caption images. ![Screenshot 2024-07-05 195607](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/8a9313e4-9f73-4b9c-a483-63491af41f3f) #### Inputs - **image**(IMAGE, required): image to be processed. - **image_detail**(STRING, default: "high": detail level of the image ("low" or "high"). - **prompt**(STRING, default: ""): text prompt for the AI model. - **model**(STRING, default: "gpt-4o"): select the OpenAI model. Options include `"GPTo"`, `"gpt-3.5-turbo"`, etc. - **api_url**(STRING, default: "https://api.openai.com/v1"): OpenAI API endpoint URL. - **api_key**(STRING): the API key for authentication. - **token_length**(INT, default: 1024): maximum token length for the generated response. #### Outputs - STRING: generated captions
## DataSet_OpenAIChatImageBatch The `DataSet_OpenAIChatImageBatch` class extends the functionality of `DataSet_OpenAIChatImage` to process batches of images with OpenAI's chat API for generating text catpions. ![Screenshot 2024-07-05 195615](https://github.com/daxcay/ComfyUI-DataSet/assets/164315771/5fe30dba-c9e0-4fb1-a97f-306210831502) #### Inputs - **images**(IMAGE, required): list of images to be processed. - **image_detail**(STRING, default: "high"): detail level of the images ("low" or "high"). - **prompt**(STRING, default: ""): text prompt for the AI model. - **model**(STRING, default: "gpt-4o"): select the OpenAI model. Options include `"GPTo"`, `"gpt-3.5-turbo"`, etc. - **api_url**(STRING, default: "https://api.openai.com/v1"): OpenAI API endpoint URL. - **api_key**(STRING): the API key for authentication. - **token_length**(INT, default: 1024): maximum token length for the generated response. #### Outputs - STRING: list of generated captions
## Credits ### Raf Stahelin - Testing and Feedback ### Daxton Caylor - ComfyUI Node Developer - ### Contact - **Twitter**: @daxcay27 - **Email** - daxtoncaylor@gmail.com - **Discord** - daxtoncaylor - **DiscordServer**: https://discord.gg/UyGkJycvyW - ### Support - **Buy me a coffee**: https://buymeacoffee.com/daxtoncaylor - **Support me on paypal**: https://paypal.me/daxtoncaylor

Name	Type	Default	Description
image	IMAGE		—
model	COMBO	Key Error	1 options: Key Error
user_prompt	STRING		—
max_tokens	INT	1024	—

DataSet_GroqChatImage

Inputs (4)

Outputs (1)