Extensions/Robin Yogurt
ComfyUI Extension

Robin Yogurt

A ComfyUI extension.

By robinjhuang·Created about a year ago·Updated about a year ago· 0
robinjhuang/ComfyUI-YogurtNodes
Nodes—
On cloudLocal install
Stars0
Updatedabout a year ago
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-genai (for Gemini nodes)
    • openai (for OpenAI and OpenRouter nodes)
    • requests (for API calls)
    • opencv-python (for Poisson Blend)

Installation Steps

  1. Navigate to your ComfyUI custom nodes directory:
cd custom_nodes
  1. Clone this repository:
git clone https://github.com/yogurt7771/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

Poisson Blend

  • Category: YogurtNodes/Image
  • Description: Uses OpenCV Poisson blending (seamlessClone) to blend foreground into background
  • Features:
    • Seamless blending using OpenCV's Poisson editing
    • Customizable blend position and size
    • Automatic mask processing and resizing
    • Supports various image formats

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

Glob Files

  • Category: YogurtNodes/IO
  • Description: Traverse folders using glob patterns and return matching file path lists
  • Features:
    • Support for glob and rglob patterns
    • Flexible sorting options (alphabetical, reverse)
    • File-only filtering option
    • Path format options (POSIX, absolute, resolved)
    • Customizable search modes

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

OpenAI Generate Text

  • Category: YogurtNodes/LLM
  • Description: Generate text using OpenAI API
  • Features:
    • Support for various OpenAI models (GPT-4, GPT-3.5, etc.)
    • Custom base URL support for compatible APIs
    • System and user prompt control
    • Customizable generation parameters (temperature, top_p, max_tokens, etc.)
    • Frequency and presence penalty controls
    • Seed support for reproducible outputs
    • Automatic retry mechanism
    • Dynamic model list fetching

OpenAI Image Understand

  • Category: YogurtNodes/LLM
  • Description: Image understanding using OpenAI vision models
  • Features:
    • Support for OpenAI vision models (GPT-4V, GPT-4o, etc.)
    • Multi-image analysis support
    • Combined image and text prompt processing
    • Custom base URL support for compatible APIs
    • Customizable generation parameters
    • High-quality image detail processing
    • Automatic retry mechanism

OpenRouter Generate Text

  • Category: YogurtNodes/LLM
  • Description: Generate text using OpenRouter API
  • Features:
    • Support for various OpenRouter models
    • Custom infrastructure provider selection
    • System and user prompt control
    • Customizable generation parameters (temperature, top_p, max_tokens)
    • Chat template support for message formatting
    • History support for conversation continuity
    • Automatic retry mechanism
    • Dynamic model list fetching

OpenRouter Image Understand

  • Category: YogurtNodes/LLM
  • Description: Image understanding using OpenRouter vision models
  • Features:
    • Support for OpenRouter vision models
    • Multi-image analysis support
    • Combined image and text prompt processing
    • Custom infrastructure provider selection
    • Customizable generation parameters
    • History support for conversation continuity
    • Automatic retry mechanism

History Builder

  • Category: YogurtNodes/LLM
  • Description: Build and manage conversation history compatible with all LLM nodes
  • Features:
    • Compatible HISTORY type output for seamless integration
    • Incremental history building with user and assistant messages
    • Automatic message filtering (empty messages ignored)
    • Simple and focused functionality
    • Essential for maintaining conversation context across LLM calls

🔑 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.

🔑 OpenAI API Key Setup

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

  1. Code Argument

    • Pass the api_key argument directly when initializing OpenAIClient (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: 
      {
  "openai": "YOUR_API_KEY",
  "openai_base_url": "https://api.openai.com/v1"
}
    • The openai_base_url is optional and defaults to the official OpenAI API.
    • This will be used only if the code argument is not provided.

  3. Environment Variable

    • Set the environment variables OPENAI_API_KEY and optionally OPENAI_BASE_URL (used only if the above two are not set).
    • Example (Windows command line): 
      set OPENAI_API_KEY=YOUR_API_KEY
set OPENAI_BASE_URL=https://api.openai.com/v1

Custom Base URL Support

The OpenAI nodes support custom base URLs, making them compatible with:

  • Official OpenAI API
  • Azure OpenAI Service
  • OpenAI-compatible APIs (like LocalAI, Ollama, etc.)
  • Self-hosted OpenAI-compatible servers

Simply set the base_url parameter to your preferred endpoint.

If the API Key is not configured correctly, OpenAI nodes will not work. You can obtain your API Key from OpenAI Platform.

🔑 OpenRouter API Key Setup

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

  1. Code Argument

    • Pass the api_key argument directly when initializing OpenRouterClient (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: 
      {
  "openrouter": "YOUR_API_KEY"
}
    • This will be used only if the code argument is not provided.

  3. Environment Variable

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

If the API Key is not configured correctly, OpenRouter nodes will not work. You can obtain your API Key from OpenRouter Platform.

🤝 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