Vito Sansevero
5485aa8c19
BREAKING CHANGE: Node categories now use emoji-based organization Add 6 new xyz-helper nodes adapted from comfyui-essentials-nodes: - FluxSamplerParams: FLUX-optimized parameter generator with batch support - LoRAFolderBatch: Batch process multiple LoRAs from folders - PlotParameters: Visualize parameter effects with graphs - SamplerSelectHelper: Intelligent sampler selection with recommendations - SchedulerSelectHelper: Optimal scheduler selection for samplers - TextEncodeSamplerParams: Combined text encoding and parameter management Changes: - Port and enhance nodes from comfyui-essentials (now in maintenance mode) - Add comprehensive documentation with attribution to original author (cubiq) - Create example workflows for xyz-helpers tools - Update all node categories to use emoji-based organization - Fix all unit tests to pass with new category system - Update README with xyz-helpers section and attribution Attribution: xyz-helpers adapted from github.com/cubiq/ComfyUI_essentials All tests passing (318 pass, 2 skip)
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
"""
|
|
KikoTools package initialization and node registry
|
|
Handles automatic discovery and registration of all ComfyAssets tools
|
|
"""
|
|
|
|
from .tools.resolution_calculator import ResolutionCalculatorNode
|
|
from .tools.width_height_selector import WidthHeightSelectorNode
|
|
from .tools.seed_history import SeedHistoryNode
|
|
from .tools.sampler_combo import SamplerComboNode, SamplerComboCompactNode
|
|
from .tools.empty_latent_batch import EmptyLatentBatchNode
|
|
from .tools.kiko_save_image import KikoSaveImageNode
|
|
from .tools.image_to_multiple_of import ImageToMultipleOfNode
|
|
from .tools.image_scale_down_by import ImageScaleDownByNode
|
|
from .tools.gemini_prompt import GeminiPromptNode
|
|
from .tools.display_any import DisplayAnyNode
|
|
from .tools.display_text import DisplayTextNode
|
|
from .tools.xyz_helpers import (
|
|
SamplerSelectHelperNode,
|
|
SchedulerSelectHelperNode,
|
|
TextEncodeSamplerParamsNode,
|
|
FluxSamplerParamsNode,
|
|
PlotParametersNode,
|
|
LoRAFolderBatchNode,
|
|
)
|
|
|
|
# ComfyUI node registration mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ResolutionCalculator": ResolutionCalculatorNode,
|
|
"WidthHeightSelector": WidthHeightSelectorNode,
|
|
"SeedHistory": SeedHistoryNode,
|
|
"SamplerCombo": SamplerComboNode,
|
|
"SamplerComboCompact": SamplerComboCompactNode,
|
|
"EmptyLatentBatch": EmptyLatentBatchNode,
|
|
"KikoSaveImage": KikoSaveImageNode,
|
|
"ImageToMultipleOf": ImageToMultipleOfNode,
|
|
"ImageScaleDownBy": ImageScaleDownByNode,
|
|
"GeminiPrompt": GeminiPromptNode,
|
|
"DisplayAny": DisplayAnyNode,
|
|
"DisplayText": DisplayTextNode,
|
|
"SamplerSelectHelper": SamplerSelectHelperNode,
|
|
"SchedulerSelectHelper": SchedulerSelectHelperNode,
|
|
"TextEncodeSamplerParams": TextEncodeSamplerParamsNode,
|
|
"FluxSamplerParams": FluxSamplerParamsNode,
|
|
"PlotParameters+": PlotParametersNode,
|
|
"LoRAFolderBatch": LoRAFolderBatchNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ResolutionCalculator": "Resolution Calculator",
|
|
"WidthHeightSelector": "Width Height Selector",
|
|
"SeedHistory": "Seed History",
|
|
"SamplerCombo": "Sampler Combo",
|
|
"SamplerComboCompact": "Sampler Combo (Compact)",
|
|
"EmptyLatentBatch": "Empty Latent Batch",
|
|
"KikoSaveImage": "Kiko Save Image",
|
|
"ImageToMultipleOf": "Image to Multiple of",
|
|
"ImageScaleDownBy": "Image Scale Down By",
|
|
"GeminiPrompt": "Gemini Prompt Engineer",
|
|
"DisplayAny": "Display Any",
|
|
"DisplayText": "Display Text",
|
|
"SamplerSelectHelper": "Sampler Select Helper",
|
|
"SchedulerSelectHelper": "Scheduler Select Helper",
|
|
"TextEncodeSamplerParams": "Text Encode for Sampler Params",
|
|
"FluxSamplerParams": "Flux Sampler Parameters",
|
|
"PlotParameters+": "Plot Parameters",
|
|
"LoRAFolderBatch": "LoRA Folder Batch",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|