# ComfyUI-CraftKit

Custom nodes for ComfyUI - image resizing, dataset prep & prompt automation for LoRA training workflows.

All nodes appear under the **Craftopia** category in the node menu.

---

## Nodes

### 🎛️ Smart Prompt Controller
Cycle through up to 4 prompt lists using a single incrementing counter. Counts lines automatically, selects the right prompt, and outputs which list is active - ideal for driving Switch nodes that control aspect ratio, latent size, or other per-category settings.

**Typical usecases:**
- LoRA dataset generation with multiple pose categories (headshots, halfbody, fullbody, tall portrait), each with their own prompt list and aspect ratio
- Any batch workflow where you need to rotate through different prompt sets and switch settings per set

| Input | Type | Description |
|---|---|---|
| counter | INT | Auto-incrementing counter (1-indexed) |
| prompt_list_1 | STRING | First prompt list - one prompt per line (forceInput) |
| prompt_list_2 | STRING | Second prompt list (optional, forceInput) |
| prompt_list_3 | STRING | Third prompt list (optional, forceInput) |
| prompt_list_4 | STRING | Fourth prompt list (optional, forceInput) |

**Outputs:** `prompt` (selected line as STRING), `list_index` (active list number as INT → feed to Switch nodes)

The node displays a status label showing the current position: `List 1 — 4/5 | total: 8`

<img src="assets/screenshot-smart-prompt-controller.png" width="700">

---

### 📐 Smart Resize
Resize any image (or batch) so the **longest side** equals `longest_side`, with aspect ratio preserved.

Use this as a **pipeline node** — IMAGE in, IMAGE out. No files are saved to disk.

**Typical usecases:**
- Downscale a SeedVR (or other upscaler) result back to your target training size — the upscaler adds fine detail, Smart Resize brings it to the right dimensions for LoRA training
- Any workflow step that needs a clean resize before the next node
- Normalize mixed-resolution batches to a consistent size

| Input | Type | Default | Description |
|---|---|---|---|
| image | IMAGE | — | Single image or batch |
| longest_side | INT | 1536 | Target size for longest side |
| multiple_of | INT | 8 | Snap dimensions to this multiple (8 = SD/Flux compatible) |
| interpolation | ENUM | lanczos | lanczos / bicubic / bilinear / nearest |
| upscale_if_smaller | BOOLEAN | false | Also upscale images already smaller than longest_side |

**Outputs:** `image`, `width`, `height`

<img src="assets/screenshot-smart-resize.png" width="700">

---

### 📁 Smart Batch Resize
Load **all images from a folder**, resize each one by longest side, and save with the original filename + a suffix into a subfolder. Original filenames are always preserved.

Use this for **bulk preprocessing** — e.g. preparing a LoRA dataset from a folder of high-res images.

| Input | Type | Default | Description |
|---|---|---|---|
| input_folder | STRING | D:/images/input | Source folder path |
| longest_side | INT | 1536 | Target size for longest side |
| filename_suffix | STRING | _Small | Appended to each output filename before the extension |
| output_subfolder | STRING | resized | Subfolder created inside input_folder |
| multiple_of | INT | 8 | Snap dimensions to this multiple |
| interpolation | ENUM | lanczos | lanczos / bicubic / bilinear / nearest |
| skip_if_exists | BOOLEAN | false | Skip files that already exist in output |

**Outputs:** `images` (list), `filenames` (list), `output_paths` (list), `count`

---

### 📏 Smart Resolution Multiplier
Multiply image dimensions by a factor and output the results as integers.

SeedVR takes a single `resolution` INT (the longest side) — not a width and height separately. Standard math nodes give you a FLOAT or require multiple steps to get there. This node does it cleanly in one step: give it your image and a multiplier, and it outputs `width`, `height`, and `resolution` (longest side) ready to connect directly to SeedVR's `max_resolution` input.

Also useful for LoRA dataset prep — using mixed resolutions in your training set generally produces better results than training on a single fixed size. Smart Resolution Multiplier makes it easy to dynamically calculate the right target size per image rather than hardcoding a value.

| Input | Type | Default | Description |
|---|---|---|---|
| image | IMAGE | — | Source image |
| multiplier | FLOAT | 2.0 | Multiply width and height by this factor |
| multiple_of | INT | 8 | Snap dimensions to this multiple |

**Outputs:** `width`, `height`, `resolution` (longest side as INT → directly into SeedVR)

<img src="assets/screenshot-smart-resolution-multiplier.png" width="700">

---

## Why these nodes?

- ComfyUI's built-in `Resize Images by Longer Edge` [BETA] has no Lanczos and no `multiple_of` snapping
- `JWImageResizeByLongerSide` (comfyui-various) has no Lanczos in the official release
- No existing node combines batch folder loading + longest-side resize + original filename preservation
- No existing node outputs a ready-to-use `resolution` INT for SeedVR
- No existing node cycles through multiple prompt lists with automatic list switching and counter

---

## Installation

**ComfyUI Manager:** search for `ComfyUI-Craftopia` and install.

**Manual:**
```bash
cd ComfyUI/custom_nodes
git clone https://github.com/CraftopiaStudio/ComfyUI-Craftopia
```
Restart ComfyUI. All nodes appear under **Craftopia** in the node menu:

- `Craftopia` → **Smart Prompt Controller 🎛️**
- `Craftopia` → **Smart Resize 📐**
- `Craftopia` → **Smart Batch Resize 📁**
- `Craftopia` → **Smart Resolution Multiplier 📏**

---

## Requirements
Pillow, NumPy, PyTorch — all included with ComfyUI. No extra dependencies.