136 lines
4.5 KiB
Python
136 lines
4.5 KiB
Python
"""
|
|
CaptionForge — ComfyUI Package Entry Point
|
|
|
|
- CaptionForge
|
|
- This package is part of **CaptionForge**, a model-agnostic captioning
|
|
framework for ComfyUI developed by **J. L. Córdova**.
|
|
|
|
- Repository
|
|
https://github.com/Damkohler/CaptionForge
|
|
|
|
- CaptionForge 1.0 provides a production A/B/C/D caption pipeline for LoRA
|
|
dataset preparation: independent witness captions, text-LLM synthesis,
|
|
image-aware VLM validation, SHORT/TAGGY formatting, and JSONL audit trails.
|
|
|
|
- Package Purpose
|
|
- This file is the ComfyUI registration entry point for the CaptionForge
|
|
custom node package.
|
|
|
|
- It exposes:
|
|
• NODE_CLASS_MAPPINGS
|
|
• NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
so ComfyUI can:
|
|
• discover the Python node classes
|
|
• display the nodes in the Add Node menu
|
|
• register the package as a unified CaptionForge node collection
|
|
|
|
- The package registers the Pipeline Planner, Template Options helper,
|
|
Joy/Qwen/Ollama Pass-A caption nodes, and the CaptionForge Orchestrator.
|
|
|
|
- Package Structure
|
|
- CaptionForge keeps ComfyUI-facing node wrappers separate from reusable
|
|
engine modules.
|
|
|
|
- Node wrappers define ComfyUI widgets, IMAGE input handling, output strings,
|
|
categories, and node mappings.
|
|
|
|
- Engine modules handle model loading, prompt resolution, generation,
|
|
cleanup, folder traversal, TXT sidecars, JSONL audit records, run-config
|
|
export, model cache behavior, and Pass B claim extraction.
|
|
|
|
- Web / Icon Assets
|
|
- ``WEB_DIRECTORY`` exposes the package's current ComfyUI frontend assets.
|
|
Node registration remains entirely explicit through the mappings below.
|
|
|
|
- Attribution & License
|
|
- Concept and implementation by **J. L. Córdova**
|
|
with development assistance from **ChatGPT (OpenAI)**.
|
|
|
|
- Designed for use with:
|
|
https://github.com/comfyanonymous/ComfyUI
|
|
|
|
- Copyright (c) 2026 J. L. Córdova
|
|
|
|
- Released under the **MIT License**.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
# ######################
|
|
# Planner node
|
|
# ######################
|
|
from .nodes.jlc_captionforge_pipeline_planner_node import (
|
|
NODE_CLASS_MAPPINGS as PIPELINE_PLANNER_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as PIPELINE_PLANNER_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
# ######################
|
|
# Caption nodes
|
|
# ######################
|
|
# # Helper
|
|
from .nodes.jlc_captionforge_template_options import (
|
|
NODE_CLASS_MAPPINGS as CAPTIONFORGE_EXTRA_OPTIONS_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as CAPTIONFORGE_EXTRA_OPTIONS_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
# Joy
|
|
from .nodes.caption_nodes.jlc_captionforge_joy_caption_node import (
|
|
NODE_CLASS_MAPPINGS as JOY_NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as JOY_NODE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
# Ollama Caption Models
|
|
from .nodes.caption_nodes.jlc_captionforge_ollama_caption_node import (
|
|
NODE_CLASS_MAPPINGS as OLLAMA_NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as OLLAMA_NODE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
#Qwen
|
|
from .nodes.caption_nodes.jlc_captionforge_qwen_caption_node import (
|
|
NODE_CLASS_MAPPINGS as QWEN_NODE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as QWEN_NODE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
|
|
# ######################
|
|
# CaptionForge node
|
|
# ######################
|
|
from .nodes.jlc_captionforge_node import (
|
|
NODE_CLASS_MAPPINGS as CAPTIONFORGE_CLASS_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as CAPTIONFORGE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
# #####################
|
|
# Class mappings
|
|
# #####################
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(PIPELINE_PLANNER_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(CAPTIONFORGE_EXTRA_OPTIONS_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(JOY_NODE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(OLLAMA_NODE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(QWEN_NODE_CLASS_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(CAPTIONFORGE_CLASS_MAPPINGS)
|
|
|
|
# ######################
|
|
# Display name mappings
|
|
# ######################
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CAPTIONFORGE_EXTRA_OPTIONS_DISPLAY_NAME_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PIPELINE_PLANNER_DISPLAY_NAME_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(JOY_NODE_DISPLAY_NAME_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OLLAMA_NODE_DISPLAY_NAME_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(QWEN_NODE_DISPLAY_NAME_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CAPTIONFORGE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
print(f"CaptionForge loaded ({len(NODE_CLASS_MAPPINGS)} nodes)")
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"WEB_DIRECTORY",
|
|
]
|