ComfyUI Extension: ComfyUI-YogurtNodes

Authored by yogurt7771

Created

Updated

0 stars

ComfyUI-YogurtNodes is a collection of custom nodes for ComfyUI, providing a series of practical image processing and workflow enhancement functionalities.

Custom Nodes (0)

    README

    ComfyUI-YogurtNodes

    ComfyUI-YogurtNodes is a collection of custom nodes for ComfyUI, providing a series of practical image processing and workflow enhancement functionalities.

    ✨ Features

    • Custom node integration
    • Easy-to-use image processing functions
    • Full compatibility with ComfyUI workflows
    • Text and image manipulation capabilities
    • Advanced string processing utilities
    • Model management and selection tools
    • Comprehensive I/O operations support
    • Integrated Gemini API for language and image understanding
    • Logic control nodes for complex workflows

    📦 Installation

    Requirements

    • ComfyUI (installed and running)
    • Python 3.x
    • Required Python packages:
      • numpy
      • pillow
      • google-generativeai (for Gemini nodes)

    Installation Steps

    1. Navigate to your ComfyUI custom nodes directory:
    cd custom_nodes
    1. Clone this repository:
    git clone https://github.com/[your-username]/ComfyUI-YogurtNodes.git
    1. Install dependencies:
    cd ComfyUI-YogurtNodes
pip install -r requirements.txt

    🚀 Usage

    1. Start ComfyUI
    2. Look for "Yogurt Nodes" category in the node browser
    3. Drag and drop desired nodes into your workflow

    🔧 Available Nodes

    All nodes are marked with "YogurtNodes" prefix for easy identification in the ComfyUI interface.

    Image Processing Nodes

    Batch Images

    • Category: YogurtNodes/Image
    • Description: Combines multiple images into a batch with customizable settings
    • Features:
      • Support for up to 16 input images
      • Multiple interpolation methods (nearest, bilinear, bicubic, etc.)
      • Various resizing methods (stretch, fill/crop, pad)
      • Customizable padding values
      • Flexible batch slicing with start, end, and step parameters

    Add Text To Image

    • Category: YogurtNodes/Image
    • Description: Adds text overlay to images with extensive customization options
    • Features:
      • Text placement at top or bottom
      • Center alignment option
      • Multiple font support
      • Customizable font size
      • Text and background color configuration
      • Multi-line text support with automatic wrapping

    None Image

    • Category: YogurtNodes/Image
    • Description: A utility node that returns None as an image output
    • Use Case: Helpful for conditional workflows where a null image output is needed

    Range

    • Category: YogurtNodes/Number
    • Description: Generate a sequence of numbers within a specified range
    • Features:
      • Customizable start, end, step, or count
      • Outputs torch tensor, count, and step

    RangeItem

    • Category: YogurtNodes/Number
    • Description: Extract a value from a range sequence by index
    • Features:
      • Supports negative index
      • Outputs string, int, and float

    String Processing Nodes

    String Lines Count

    • Category: YogurtNodes/String
    • Description: Counts the number of lines in a multiline string
    • Features:
      • Optional line stripping
      • Empty line filtering control
      • Configurable counting behavior

    String To Value

    • Category: YogurtNodes/String
    • Description: Converts string input to multiple value types
    • Features:
      • Automatic conversion to integer
      • Automatic conversion to float
      • Preserves original string
      • Safe conversion with fallback values

    String Lines Switch

    • Category: YogurtNodes/String
    • Description: Extracts a specific line from a multiline string by index
    • Features:
      • Line indexing
      • Multiple output formats (string, int, float)
      • Line count output
      • Empty line handling options

    Replace Delimiter

    • Category: YogurtNodes/String
    • Description: Replaces delimiters in strings with custom separators
    • Features:
      • Regular expression support
      • Custom delimiter specification
      • Flexible replacement options

    Split Path

    • Category: YogurtNodes/String
    • Description: Splits file paths into their component parts
    • Features:
      • Path component extraction
      • Returns parent directory
      • Filename and extension separation
      • Multiple format options for extensions

    Regex Node

    • Category: YogurtNodes/String
    • Description: Regex-based extraction and replacement for multiline text
    • Features:
      • Supports 'extract' and 'replace' modes
      • Group reference and formatting
      • Extract or replace by count/order
      • Suitable for complex text processing

    Logic Processing Nodes

    None Node

    • Category: YogurtNodes/Logic
    • Description: A utility node that returns None as its output
    • Use Case: Useful for conditional workflows where a null value is needed

    Pack Any

    • Category: YogurtNodes/Logic
    • Description: Packs multiple input items into a single output
    • Features:
      • Support for up to 8 input items
      • Handles any data type
      • Simplifies complex workflows

    Unpack Any

    • Category: YogurtNodes/Logic
    • Description: Unpacks input items into multiple outputs
    • Features:
      • Seamlessly works with Pack Any node
      • Outputs up to 8 separate items
      • Useful for data distribution

    Switch

    • Category: YogurtNodes/Logic
    • Description: Condition-based switch node
    • Features:
      • Regular expression matching support
      • Up to 8 condition branches
      • Default value option
      • Flexible condition control

    Model Selection Nodes

    ControlNet Selector

    • Category: YogurtNodes/Models
    • Description: Advanced ControlNet model selection and configuration
    • Features:
      • Dynamic model list from available ControlNets
      • Adjustable strength parameter
      • Start and end percentage control
      • Model path and name extraction
      • None option for conditional workflows

    Diffusion Model Selector

    • Category: YogurtNodes/Models
    • Description: Stable Diffusion model selection utility
    • Features:
      • Dynamic list of available diffusion models
      • Trigger word support
      • Model path and name extraction
      • None option for conditional workflows

    Checkpoint Selector

    • Category: YogurtNodes/Models
    • Description: Checkpoint model selection and management
    • Features:
      • Dynamic checkpoint list
      • Trigger word support
      • Model information extraction
      • None option for conditional workflows

    Lora Selector

    • Category: YogurtNodes/Models
    • Description: LoRA model selection and configuration
    • Features:
      • Dynamic LoRA model list
      • Separate model and CLIP strength controls
      • Trigger word support
      • Model path and name extraction
      • None option for conditional workflows

    I/O Operation Nodes

    Save Image Bridge Ex

    • Category: YogurtNodes/IO
    • Description: Enhanced image saving functionality with extensive options
    • Features:
      • Multiple format support (PNG, JPEG)
      • Customizable output directory
      • Dynamic filename prefixes with variables
      • Metadata control
      • Overwrite protection
      • Compression level control
      • Quality settings for JPEG
      • Custom suffix support

    Save Image Bridge

    • Category: YogurtNodes/IO
    • Description: Basic image saving functionality
    • Features:
      • PNG and JPEG support
      • Basic output options
      • Metadata handling
      • Compression settings

    Preview Image Bridge

    • Category: YogurtNodes/IO
    • Description: Quick image preview functionality
    • Features:
      • Temporary file handling
      • Fast preview generation
      • Automatic cleanup

    Save Text Bridge

    • Category: YogurtNodes/IO
    • Description: Text file saving with multiple format support
    • Features:
      • Multiple format support (.txt, .json, .md, .csv)
      • Custom file extensions
      • Dynamic filename generation
      • Overwrite protection
      • UTF-8 encoding support

    Any Bridge

    • Category: YogurtNodes/IO
    • Description: Universal data bridge for workflow control
    • Features:
      • Handles any data type
      • Optional blackhole mode
      • Workflow control support

    Create Directory

    • Category: YogurtNodes/IO
    • Description: Directory creation utility
    • Features:
      • Creates single directory
      • Automatic parent directory creation
      • Path validation

    Create Parent Directory

    • Category: YogurtNodes/IO
    • Description: Parent directory creation utility
    • Features:
      • Creates parent directories
      • Recursive creation support
      • Path validation
      • Safe operation with existing directories

    Save Mask Bridge Ex

    • Category: YogurtNodes/IO
    • Description: Enhanced mask saving node, supports multiple formats and custom suffix
    • Features:
      • Supports PNG/JPEG/custom formats
      • Custom output directory, filename prefix, suffix
      • Metadata, compression, overwrite protection

    Save Mask Bridge

    • Category: YogurtNodes/IO
    • Description: Basic mask saving node
    • Features:
      • Supports PNG/JPEG formats
      • Custom output directory, filename prefix
      • Metadata, compression, overwrite protection

    Save Mask Bridge Simple

    • Category: YogurtNodes/IO
    • Description: Simplified mask saving node
    • Features:
      • Minimal parameters, quick mask saving

    Preview Mask Bridge

    • Category: YogurtNodes/IO
    • Description: Mask preview node
    • Features:
      • Saves mask as temporary file for quick preview

    Language Model Nodes

    Gemini Generate Text

    • Category: YogurtNodes/LLM
    • Description: Generate text using Gemini API
    • Features:
      • Support for various Gemini models
      • System and user prompt control
      • Customizable generation parameters (temperature, top_p, top_k, etc.)
      • Safety settings control
      • Maximum output token count
      • Automatic retry mechanism

    Gemini Image Understand

    • Category: YogurtNodes/LLM
    • Description: Image understanding using Gemini API
    • Features:
      • Image content analysis
      • Combined with text prompts
      • Customizable generation parameters
      • Multi-language support
      • Detailed image descriptions

    Gemini Generate Image

    • Category: YogurtNodes/LLM
    • Description: Generate images using Gemini API
    • Features:
      • Support for various Gemini image generation models
      • System and user prompt control
      • Customizable generation parameters (temperature, top_p, top_k, etc.)
      • Safety settings control
      • Maximum output token count
      • Automatic retry mechanism
      • Outputs image (torch.Tensor), text description, and image count

    🔑 Gemini API Key Setup

    Before using Gemini-related nodes, you must obtain and configure your Gemini API Key. There are three supported methods, in the following order of priority:

    1. Code Argument

      • Pass the api_key argument directly when initializing GeminiClient (highest priority).

    2. api_key.json File

      • Create an api_key.json file in custom_nodes/ComfyUI-YogurtNodes/yogurt_nodes/llm/ with the following content: 
        {
  "gemini": "YOUR_API_KEY"
}
      • This will be used only if the code argument is not provided.

    3. Environment Variable

      • Set the environment variable GEMINI_API_KEY (used only if the above two are not set).
      • Example (Windows command line): 
        set GEMINI_API_KEY=YOUR_API_KEY

    If the API Key is not configured correctly, Gemini nodes will not work. You can obtain your API Key from Google AI Studio.

    🤝 Contributing

    Pull Requests are welcome to help improve the project!

    📄 License

    This project is licensed under the MIT License - see the LICENSE file for details.

    📞 Contact

    For questions, bug reports, or feature requests, please open an issue.

    🙏 Acknowledgments

    • ComfyUI Community
    • All Contributors