Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
83 changes: 83 additions & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
name: CI

on:
push:
branches: [main, develop]
pull_request:
branches: [main, develop]

jobs:
test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: [3.8, 3.9, "3.10", "3.11"]

steps:
- uses: actions/checkout@v4

- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}

- name: Cache pip dependencies
uses: actions/cache@v3
with:
path: ~/.cache/pip
key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements*.txt') }}
restore-keys: |
${{ runner.os }}-pip-

- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements-dev.txt
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi

- name: Format check with black
run: |
black --check --diff .

- name: Import sorting check with isort
run: |
isort --check-only --diff .

- name: Lint with flake8
run: |
flake8 .

# Skip mypy due to path conflicts in ComfyUI node structure
# - name: Type check with mypy
# run: |
# mypy .

- name: Run tests
run: |
python run_tests.py

- name: Test individual nodes
run: |
python -c "
import sys
sys.path.append('nodes')

# Mock ComfyUI
from tests.mocks.mock_comfy import MockSamplers, MAX_RESOLUTION
comfy_module = type('MockComfy', (), {})()
comfy_module.samplers = MockSamplers
sys.modules['comfy'] = comfy_module
sys.modules['comfy.samplers'] = MockSamplers
nodes_module = type('MockNodes', (), {})()
nodes_module.MAX_RESOLUTION = MAX_RESOLUTION
sys.modules['nodes'] = nodes_module

# Test imports
from sampler_selector import SamplerSelector
from scheduler_selector import SchedulerSelector
from seed_generator import SeedGenerator
from width_node import WidthNode
from height_node import HeightNode
from width_height_node import WidthHeightNode
print('All nodes import successfully')
"
197 changes: 195 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,2 +1,195 @@
# ComfyUI_Selectors
ComfyUI_Selectors provides custom nodes that centralize commonly shared parameters—such as scheduler, steps, config, and sampler—and allow you to link these settings to multiple nodes in your workflow. This eliminates redundancy and streamlines graph editing without breaking compatibility with ComfyUI’s JSON output format.
# ComfyUI Selectors

**Version 1.0**

A modern ComfyUI custom node package that provides essential UI controls for image generation workflows. These nodes allow you to centralize commonly shared parameters (scheduler, sampler, dimensions, seeds) and link them to multiple nodes in your workflow, eliminating redundancy while maintaining JSON metadata compatibility.

## Features

- **Sampler Selector**: Choose from all current ComfyUI-supported sampling methods
- **Scheduler Selector**: Select from all current ComfyUI-supported schedulers
- **Seed Generator**: Generate, increment, or randomize seeds with advanced controls
- **Width & Height Nodes**: Individual dimension controls with preset support
- **Combined Width/Height Node**: Dual dimension control with swap functionality and presets
- **Modern ComfyUI Standards**: Follows 2024 ComfyUI development best practices
- **JSON Compatible**: Maintains workflow JSON compatibility for sharing and version control

## Installation

1. Navigate to your ComfyUI `custom_nodes` directory
2. Clone this repository:
```bash
git clone https://github.com/your-username/ComfyUI_Selectors.git
```
3. Restart ComfyUI
4. The nodes will appear in the `Selectors/` category

## Available Nodes

### Sampling Nodes (`Selectors/Sampling`)

#### Sampler Selector
- **Function**: Centralized sampler selection
- **Output**: Sampler name for use in KSampler nodes
- **Supported Samplers**: All current ComfyUI samplers including euler, dpmpp_2m, ddim, etc.

#### Scheduler Selector
- **Function**: Centralized scheduler selection
- **Output**: Scheduler name for use in KSampler nodes
- **Supported Schedulers**: normal, karras, exponential, sgm_uniform, simple, ddim_uniform, beta

### Generation Nodes (`Selectors/Generation`)

#### Seed Generator
- **Function**: Advanced seed generation with multiple control modes
- **Controls**:
- `seed`: Base seed value (0-18446744073709551615)
- `control_after_generate`: How to handle seed after generation
- `fixed`: Keep seed unchanged
- `increment`: Increase seed by 1
- `decrement`: Decrease seed by 1
- `randomize`: Generate random seed
- **Output**: Seed value for use in KSampler nodes

### Dimension Nodes (`Selectors/Dimensions`)

#### Width Node
- **Function**: Image width selection with presets
- **Controls**:
- `width`: Custom width value (64-8192, step 8)
- `preset`: Quick selection (custom, 512, 768, 1024, 1152, 1216, 1344, 1408, 1472, 1536)
- **Output**: Width value for use in latent image nodes

#### Height Node
- **Function**: Image height selection with presets
- **Controls**:
- `height`: Custom height value (64-8192, step 8)
- `preset`: Quick selection (custom, 512, 768, 1024, 1152, 1216, 1344, 1408, 1472, 1536)
- **Output**: Height value for use in latent image nodes

#### Width & Height Node
- **Function**: Combined dimension control with advanced features
- **Controls**:
- `width`: Custom width value (64-8192, step 8)
- `height`: Custom height value (64-8192, step 8)
- `preset`: Common dimension presets (512x512, 768x512, 1024x768, etc.)
- `swap_dimensions`: Toggle to swap width and height values
- **Outputs**: Both width and height values
- **Presets Include**: Square formats, landscape, portrait, and popular ratios

## Usage Examples

### Basic Workflow Setup
1. Add a **Sampler Selector** node and choose your preferred sampler
2. Add a **Scheduler Selector** node and choose your scheduler
3. Add a **Seed Generator** node and set control mode
4. Add a **Width & Height Node** and choose dimensions
5. Connect outputs to your KSampler and latent image nodes

### Centralized Parameter Management
- Use selector nodes to set parameters once
- Connect outputs to multiple destination nodes
- Change settings in one place to affect entire workflow
- Maintain clean, organized node graphs

### Preset Workflows
- Use dimension presets for quick setup of common aspect ratios
- Use seed randomize mode for variation generation
- Combine nodes for complex parameter linking scenarios

## Technical Details

### ComfyUI Compatibility
- Built for ComfyUI 2024+ standards
- Compatible with all current sampler and scheduler types
- Maintains JSON workflow compatibility
- Supports widget conversion and parameter linking

### Node Categories
- `Selectors/Sampling`: Sampler and scheduler selection
- `Selectors/Generation`: Seed and generation control
- `Selectors/Dimensions`: Width, height, and dimension control

### Output Types
- **Sampler/Scheduler**: ComfyUI COMBO types with current options
- **Dimensions/Seeds**: Standard INT types with proper validation
- **JSON Metadata**: Maintains workflow sharing compatibility

## Development

### Setup
1. **Create virtual environment:**
```bash
python -m venv venv
source venv/bin/activate # Linux/Mac
# or
venv\Scripts\activate # Windows
```

2. **Install development dependencies:**
```bash
pip install -r requirements-dev.txt
```

### Testing
The project includes comprehensive testing with mock ComfyUI environment:

```bash
# Run all tests (recommended)
python run_tests.py

# Run individual pytest tests
pytest tests/unit/ # Unit tests
pytest tests/integration/ # Integration tests

# Check code formatting
black --check .
isort --check-only .
flake8 .
```

### Project Structure
```
nodes/ # ComfyUI node implementations
tests/ # Test suites with mock ComfyUI
├── mocks/ # Mock ComfyUI modules
├── unit/ # Unit tests
└── integration/ # Integration tests
run_tests.py # Standalone test runner
requirements-dev.txt # Development dependencies
```

### Contributing
1. Follow ComfyUI development standards
2. Use the provided test framework with mock ComfyUI environment
3. Maintain backward compatibility
4. Add tests for new functionality
5. Update documentation
6. Ensure all tests pass before submitting PRs

### Code Quality
- **Formatting:** Black (88 character line limit)
- **Import sorting:** isort
- **Linting:** flake8
- **Testing:** Custom test runner with comprehensive mock setup

## Version History

### Version 1.0 (Initial Release)
- ✅ Sampler Selector with all current ComfyUI samplers
- ✅ Scheduler Selector with all current ComfyUI schedulers
- ✅ Seed Generator with advanced control modes (fixed/increment/decrement/randomize)
- ✅ Width Node with preset support
- ✅ Height Node with preset support
- ✅ Combined Width & Height Node with swap functionality and dimension presets
- ✅ Modern ComfyUI 2024 standards compliance
- ✅ Full JSON workflow compatibility
- ✅ Comprehensive documentation and testing

## License

[Add your license here]

## Support

[Add support information here]
Loading