Commit Graph
34 Commits
Author SHA1 Message Date
ausboss ec50d3acfe Promote the Krea 2 outpaint pair from the lab
Krea 2 accepts reference latents, but they arrive with no position, so the
model treats them as a loose style hint: it borrows the look and reinvents
the content. That is the whole reason an outpaint drifts instead of
continuing the photo.

Krea 2 Encode attaches the reference latents and emits the negative from
the same node, so a turbo graph at CFG 1.0 stops carrying a second text
encode that does nothing. Krea 2 Outpaint Model Patch registers those
reference tokens into the target grid at the rectangle the stitcher
records, on frame axis 1.

Load Image + Pad gained a `reference` output - the unpadded source, fitted
to a 384px long edge and a multiple of 16, because the VAE downsamples by
8 and the DiT patchifies by 2. It is APPENDED after `stitcher`, never
inserted: a workflow stores links by output slot index, so any other
position silently rewires every saved graph.

The stitcher now records where the source sits, as source_bbox in pixels
and bbox_normalized in 0..1. Padding knows the rectangle exactly, so it
rides with the canvas instead of a parallel wire that can be left
unplugged. apply_stitch reads neither key, so an older stitcher stitches
identically.

Ported from the lab rather than copied: the lab's _krea2_helpers had its
own pad_image, feather_pad_mask, plan_pad_canvas, parse_fill_color and
build_canvas_stitcher. Duplicating the padding engine here would let the
outpaint path and the inpaint path disagree about what a stitcher is, so
the public module keeps only what is specific to Krea 2 and uses the
existing helpers for the rest.

Two departures from the lab version. The model patch declared a kv_cache
input and never read it - the cache was always on - so it is wired up now.
And it imports comfy's flux internals when you run it rather than at
startup: a core release that moves one then surfaces as an error on that
node, instead of deleting it from the menu and leaving saved workflows
reporting it missing.
2026-08-17 22:17:19 -05:00
ausboss 772f78d031 Remove Pad Image
Load Image + Pad does the same job from the same stage and starts from
the file, so the only thing the second node added was taking an IMAGE
wire - one idea holding two slots in the menu.

Deletes the node, its panel, its help doc and its stage-only shared
helpers (padEmptyStateText, findPadPreview), and drops it from
NODE_MODULES, PUBLIC_NODE_IDS, the README and the panel guard list.

The padding helpers are untouched; Load Image + Pad and Align Image both
use them. Two console labels there named the removed node, so
_fill_tensor now takes the source name - Align Image passes its own
pad_color, and a colour it cannot read is no longer reported against
padding. The BHWC and mode errors read "Padding"/"Pad mode" instead.

The on-node pad stage moves to ComfyUI-AusBoss-Lab, which has its own
AUSBOSS_LAB_PadImage.
2026-08-17 21:59:07 -05:00
ausboss 10382f1643 Remove Drop Shadow
The node produced a mask offset, grown and blurred - no contact
darkening, no perspective - so it read as a sticker halo rather than a
shadow in the scene. That is the approach, not a tuning problem, so it
comes out rather than shipping under the pack's name.

Deletes the node, its helper module, its test file and its help doc, and
drops it from NODE_MODULES, PUBLIC_NODE_IDS and the README. The colour
helper tests keep their two-callers case using Pad Image and Transform,
the real remaining call sites.

blur_mask and grow_shrink_mask stay in _mask_helpers - Crop For Inpaint
and Pad Image use both.
2026-08-17 21:59:07 -05:00
ausboss e152b903e0 Gate the public node list in both directions
PUBLIC_NODE_IDS only checked that every listed id was still registered. A
brand-new id could register and validation stayed green, so the release gate
could not tell when a node reached the public pack without anyone deciding it
should ship.

Also pins SimpleWatermarkRemover in its own LEGACY_NODE_IDS set. It is the
pre-convention id kept so already-published workflows keep loading, and it was
never listed - so the new check now asserts the compatibility shim is present
rather than flagging it as a stray.
2026-08-17 16:33:20 -05:00
ausboss c15c6c9317 Implement the pack audit batch: on-node pad editors, feature gaps, removals
The audit compared every node against ComfyUI core and the major packs;
this lands its outcomes in one branch.

On-node visual editing (the old Codex-pack interaction, rebuilt):
- New Load Image + Pad: image loader with whole-edge pad dragging on the
  node canvas, align-to-multiple rounding, and a megapixel target that
  rescales the source before padding so the mask seam stays binary.
- Pad Image gains the same handle canvas, fed by an execution-time
  snapshot of its input; wireframe with diagnostic empty states before
  the first run. Widgets, outputs, and saved workflows unchanged.
- Image Crop + Rotate + Pad's compact panel is now a live stage - crop,
  pad, and rotate handles on the node, one shared implementation with
  the full-screen editor so the two surfaces cannot drift.

Feature gaps closed against the ecosystem:
- Crop For Inpaint: target_megapixels, selectable rescale_algorithm
  (recorded in the stitcher for the paste-back), and extend_* inputs
  that grow the frame itself for outpainting.
- Save Video: format choice - mp4 h264, mp4 h265, webm vp9 with Opus
  (resampled to 48 kHz); CRF drives all three.
- Load Video: every_nth thinning (fps output divides to match) and a
  max_frames cap that ends the decode early.
- LoRA Loader: saved stack templates behind a bar button, and a console
  warning when a LoRA's base-model family disagrees with the checkpoint.
- LM Studio Chat: history in/out for multi-turn chains and a json_schema
  widget for structured replies.
- Drop Shadow: multiply blend and a shadow_mask output. Align Image:
  offset outputs for un-aligning after sampling. Color Match:
  first_frame reference mode as a one-node video flicker fix.

Removed (backed up in git history): Select Frame Range and the Video
Bundle family - core ImageFromBatch and the core VIDEO wire now cover
both jobs. Dead NODE_ID module variables cleaned out of every node file.

455 Python tests, 216 JS tests, validate_nodes.py green on 19 node files.
2026-08-17 10:49:55 -05:00
ausboss 85b70389ff Add Align Image, Image Size, and LM Studio Chat
Align Image snaps width and height to a clean multiple - 16, 32,
whatever the model wants - because Qwen image models, VAEs and plenty of
samplers only behave on cleanly divisible sizes. Each mode implies its
own rounding, so there is no separate direction widget: resize rescales
to the nearest multiple, crop center-crops down, pad replicate-pads up,
and a side smaller than one multiple always grows to exactly one rather
than erroring. The new size comes out as INTs for wiring into latents.

Image Size reads width, height, longest edge and shortest edge off the
wire so resizes and latents key off the actual image instead of retyped
numbers that drift.

LM Studio Chat posts a prompt - and optionally one image - to a local
LM Studio server, or anything else speaking the OpenAI-compatible chat
API. The interface is the middle ground the installed ecosystem packs
miss on either side: endpoint string, model string where EMPTY means
whatever the server has loaded (a dropdown fetched at ComfyUI startup
goes stale whenever LM Studio was not running yet), optional system
prompt, prompt required unless an image is connected, temperature,
max_tokens, seed as the re-roll knob, a timeout, and an image downscale
cap. Reasoning-model think blocks land on their own output so the text
stays clean for conditioning. Stdlib urllib only - no new dependencies -
and the POST runs off the executor thread. Every error names what to
fix, with the server's own words when it had any.

Verified live against a running LM Studio: a JIT-loaded qwen3.5-9b
answered a text round trip in 7.6 s through the real node; the
no-model-loaded 400 and a dead port both surfaced their actionable
messages. The vision payload shape is unit-tested; the two local VL
quants refused images server-side (missing projector / failed load), so
the live vision check stays on the deployment checklist.

48 offline tests cover the alignment math, both size nodes, the URL and
payload builders, think-splitting, image encoding, and every mocked
network error path.
2026-08-15 20:51:51 -05:00
ausboss 5e6cdb2ecc Fix the fifteen defects the review found
Frame Chooser, mostly. A pause did not survive a workflow load: LiteGraph
clears a graph by removing every node, so undo, a tab switch, Clear
Workflow and opening another file all fired the teardown a deleted node
uses to release its pause, and killed a run that was still going. That
now only fires for a real deletion. Answering "keep all" wrote the batch
out as 1,2,...,N into pick_list, so the next run dropped whatever the
enumeration did not reach or failed outright on a shorter clip; it
writes the empty answer it was given. And pick_list is only written back
under "keep last selection" - it pre-answers the node, so filling it in
automatically meant a chooser left at the default paused exactly once.

Refine Mask: smooth applies only the change it made to the jaggy edge
rather than binarizing everything, so an already-soft matte keeps its
alpha (binary masks stay bit-identical), and blur now reaches the
matting solve instead of being thresholded back out of the trimap.

Frame Interpolate's copy path honours batch_size; it gathered every
copied frame at once, which on a long clip at an integer multiple was a
multi-gigabyte allocation on the frames device that no setting bounded.

Subgraph execution ids are resolved by walking the chain instead of
stripping the prefix, which could match an unrelated node with the same
number. Status and runtime badges appear inside subgraphs again, and a
subgraph pause renders on the node that actually paused.

Also: the Custom colour scheme no longer strips every node's colour when
its stored colour is unreadable; Drop Shadow and Pad Image name their
own widget for an unreadable colour instead of blaming Transform; a
non-ASCII token and a non-object JSON body answer 400 rather than
throwing a 500 and stranding the pause; pick_list rejects Unicode
digits, which used to raise during validation or silently read as a
frame nobody typed.

Tooling: validate_nodes parses NODE_MODULES rather than matching it
anywhere in the text, and an unregistered module is an error - it used
to exit 0 printing nothing. It also refuses mapping keys declared
outside nodes/, which is how the registry fixtures came to be advertised
to ComfyUI-Manager as installable nodes; they are .py.txt now. And the
video test no longer leaves ComfyUI on sys.path, which was flipping
every fail-soft seam for whatever ran next.
2026-08-15 00:00:15 -05:00
ausboss 2c9c16ad16 Merge pull request #29 from ausboss/chore/registry-guard
Guard mapping keys against a scanner-invisible regression
2026-08-14 10:18:21 -05:00
ausboss e0813c765d Merge pull request #27 from ausboss/feat/frame-interpolate
Add FPS-aware frame interpolation
2026-08-14 10:18:07 -05:00
ausboss 63125fa021 Close the gaps the registry contract still left open
Four holes. Two empty literals agree with each other, so a module that
registers nothing passed every rule. A del could retire a node the
scanner goes on advertising - the mirror image of the update() case
already covered. An alias made the mutation invisible: _registry =
NODE_CLASS_MAPPINGS then _registry.update(...) never says the mapping
name at all. And the permanent-id check grepped the whole file for
AUSBOSS_NODES_*, so a key renamed in the literal kept passing as long as
the old id survived in a docstring or a search alias.

The contract is now that a mapping is assigned once to a non-empty dict
literal and its name is never used again. That is what makes the rest
enforceable: the alias is refused where it is made, the last point
anything static can still see it, rather than chased through code no
checker can follow. Permanent ids are measured against the keys parsed
out of the literal.

bad_empty, bad_delete and bad_alias join the fixtures; each one was
dropped into nodes/ and confirmed to exit 1 on its own line.
2026-08-14 08:33:56 -05:00
ausboss 2675a32535 Check the whole registry contract, not just literal keys 2026-08-13 22:56:24 -05:00
ausboss bfccf46a34 Describe the mapping key check in the validator docstring 2026-08-13 20:14:26 -05:00
ausboss c1fa6bb494 Guard mapping keys against a scanner-invisible regression 2026-08-13 20:08:22 -05:00
ausboss 2f034c61ee Add FPS-aware frame interpolation 2026-08-13 17:55:49 -05:00
ausboss 8a42a26a13 Add a masked Drop Shadow node 2026-08-13 17:55:49 -05:00
ausboss 2dcb89fa3e Add Pad Image node with four fill modes and an outpaint mask
Pads any side with a solid color, an averaged edge-color band,
replicated edge pixels, or a blurred and dimmed pillarbox backdrop.
The original region stays bit-identical in every mode and the second
output masks exactly the new padding, ready for outpainting.
2026-08-13 17:55:49 -05:00
ausboss e22219c4a8 Add Color Match node with pure-torch LAB statistics transfer
Per-channel LAB mean/std matching against a reference image, with an
optional mask that restricts both measurement and application, a
strength blend, and batch-1 reference broadcasting. Pixels outside the
mask stay bit-identical.
2026-08-13 17:55:49 -05:00
copilot-swe-agent[bot]andausboss a5625ccdb6 Merge remote-tracking branch 'origin/main' into feat/video-bundle-compare
# Conflicts:
#	__init__.py
#	scripts/validate_nodes.py

Co-authored-by: ausboss <102309304+ausboss@users.noreply.github.com>
2026-08-13 16:35:53 +00:00
copilot-swe-agent[bot]andausboss 5063b83ed6 Merge origin/main into feat/inpaint-crop-stitch
Co-authored-by: ausboss <102309304+ausboss@users.noreply.github.com>
2026-08-13 16:33:06 +00:00
ausboss 688a905302 Merge remote-tracking branch 'origin/main' into feat/inpaint-crop-stitch
# Conflicts:
#	scripts/validate_nodes.py
2026-08-13 09:44:40 -05:00
ausboss d41b316efd Merge remote-tracking branch 'origin/main' into feat/frame-chooser 2026-08-13 09:43:47 -05:00
ausboss 1119528d11 Add the Crop For Inpaint and Stitch Inpaint node pair 2026-08-13 09:43:32 -05:00
ausboss ee252ee041 Add the Frame Chooser backend with a pausing selection flow 2026-08-13 09:43:08 -05:00
ausboss a0667f8c93 Merge remote-tracking branch 'origin/main' into feat/video-bundle-compare
# Conflicts:
#	__init__.py
#	scripts/validate_nodes.py
2026-08-13 09:42:35 -05:00
ausboss 728bf63a1f Add the A/B Compare node with a slide and hold panel
An OUTPUT_NODE that saves temp previews of each batch's first frame
and renders them stacked in a DOM panel: slide mode clips B to the
pointer with a seam line, hold mode flashes B while pressed. image_a
passes through unchanged so the comparison can sit mid-graph. Pure
clip and mode logic lives in js/shared/compare.mjs under node:test.
2026-08-13 09:40:56 -05:00
ausboss 50b8cffb16 Add the AUSBOSS_VIDEO bundle wire and its node trio
Video Bundle packs frames, fps, and optional audio into one custom
wire; Video Unbundle unpacks it; Video Bundle Edit copies a bundle
overriding only the connected inputs. Frame count, size, and duration
are derived from the frames tensor and fps at bundle time and
recomputed on every edit, so they can never drift from the frames.
2026-08-13 09:37:46 -05:00
ausboss 148c11eb6c Merge pull request #7 from ausboss/feat/lora-loader
Add the LoRA Loader node
2026-08-13 09:33:53 -05:00
ausboss d6875fcaf8 Add the LoRA Loader node with stacked rows and trigger words 2026-08-12 22:54:06 -05:00
ausboss dce3c149de Add Select Frame Range node 2026-08-12 22:51:00 -05:00
ausboss 873662997e Add Save Video node and cut VHS and Easy-Use from the watermark workflow 2026-08-11 18:49:33 -05:00
ausboss 64fe157a7f Add Load Video and Refine Mask nodes 2026-08-11 18:36:48 -05:00
ausboss c3d0a8e8b6 Add LaMa Inpaint and Select Frame nodes with the watermark remover workflow 2026-08-11 18:03:19 -05:00
ausboss 83fae3b36b Build polished transform editors 2026-07-24 20:43:53 -05:00
ausboss 8254f647ca Scaffold ComfyUI-AusBoss node pack
Project structure modeled on ComfyUI-Pixaroma (MIT) with original code:
fail-soft module loader in __init__.py, one file per node under nodes/,
per-node frontend folders under js/ with shared .mjs utilities, offline
validation script, docs, and a smoke-test workflow.

Phase 1 placeholder nodes: Text Box, Show Text (with JS frontend),
Random Line, Image Dimensions.
2026-07-24 17:21:30 -05:00