nodes/whmc76-ComfyUI-UniversalToolkit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
8efa2b7d6d1e245bbaa2e959695f0b016dc520fb
whmc76-ComfyUI-UniversalToo…/__init__.py
T
Cyber Dick LangandClaude Sonnet 4.6 8efa2b7d6d v1.4.11: 新增 Save Image Plus (UTK) 节点，支持高质量自定义保存图像 
- 新增 SaveImagePlus_UTK 节点，参考 LayerStyle SaveImagePlusV2 实现
- 支持 PNG、JPEG、TIFF、WebP 四种输出格式
- 支持自定义 DPI 元数据嵌入（默认 300 DPI，适用于印刷质量输出）
- PNG：pHYs 块嵌入 DPI + PngInfo 工作流 JSON + 文本元数据
- JPEG：JFIF DPI + 可选 EXIF 作者/描述（需 piexif）
- TIFF：LZW 压缩 + XResolution/YResolution 标签 + Artist/ImageDescription + 工作流 JSON
- WebP：质量控制 + 可选 EXIF DPI（需 piexif，不可用则跳过）
- 支持可选 author、description 元数据字段
- 支持子目录命名（filename_prefix 使用 '/' 分隔）
- 更新 pyproject.toml 版本至 1.4.11 以触发 Comfy Registry 版本抓取

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-03-16 13:56:48 +08:00

1024 lines
48 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
"""
ComfyUI Universal Toolkit
~~~~~~~~~~~~~~~~~~~~~~~
A comprehensive toolkit for ComfyUI that provides various utility nodes for image processing, text manipulation, and more.
:copyright: (c) 2024 by May
:license: MIT, see LICENSE for more details.
"""
__version__ = "1.4.11"
__author__ = "CyberDickLang"
__email__ = "286878701@qq.com"
__url__ = "https://github.com/whmc76"
# 更新日志
CHANGELOG = {
"1.4.11": [
"新增 Save Image Plus (UTK) 节点：",
"- 支持 PNG、JPEG、TIFF、WebP 四种格式输出",
"- 支持自定义 DPI 元数据嵌入（默认 300 DPI，用于印刷质量输出）",
"- PNG：嵌入 pHYs 分辨率块 + ComfyUI 工作流 + 文本元数据",
"- JPEG：支持 DPI 与可选 EXIF（需 piexif）",
"- TIFF：LZW 压缩、DPI + Artist/ImageDescription 标签 + 工作流 JSON",
"- WebP：支持质量控制与可选 EXIF DPI（需 piexif）",
"- 支持 author、description 可选元数据字段",
"- 支持子目录命名（filename_prefix 使用 '/' 分隔）",
"- 分类：UniversalToolkit/Image",
],
"1.4.10": [
"新增 Image Batch Extend With Overlap (UTK) 节点：",
"- 自 comfyui-kjnodes 移植，用于视频/图像序列扩展与重叠混合",
"- 输出：source_images、start_images、extended_images",
"- 支持 overlap_side（source/new_images）与多种 overlap_mode：cut、linear_blend、ease_in_out、filmic_crossfade、perceptual_crossfade（需可选 kornia）",
"- 分类：UniversalToolkit/Image",
],
"1.4.9": [
"新增Show Any (UTK)节点：",
"- 参考comfyui-easy-use的showAnything节点实现",
"- 支持显示任意类型的数据（string、int、float、json、list、torch.Tensor等）",
"- 在节点内部文本框中显示数据类型和内容",
"- 支持数据透传，不做任何修改",
"- 支持工作流保存和加载时恢复显示内容",
"- 分类：UniversalToolkit/Tools",
],
"1.4.8": [
"版本更新和代码优化：",
"- 更新插件版本号为 1.4.8",
"- 代码优化和稳定性改进",
],
"1.4.7": [
"修复 resize 与 pad 方法表现相同的问题：",
"- ResizeImageVerKJ (UTK)：resize 模式只等比缩放不填充，pad 模式填充到目标尺寸",
"- ImageMaskScaleAs (UTK)：resize 返回实际缩放尺寸，pad 填充到目标尺寸并正确输出尺寸",
"- ImageScaleByAspectRatio (UTK)：resize 返回实际缩放尺寸，pad 填充到目标尺寸并正确输出尺寸",
"- resize：等比缩放，输出尺寸 = 缩放后尺寸（可能小于目标尺寸）",
"- pad：等比缩放 + 背景填充，输出尺寸 = 目标尺寸（固定尺寸）",
],
"1.4.6": [
"新增 Resize Image ver KJ (UTK)：",
"- 复刻 KJ v2 的调整模式：stretch/resize/pad/pad_edge/pad_edge_pixel/crop/pillarbox_blur/total_pixels",
"- 支持 mask 同步缩放与对齐，pad_edge/pad_edge_pixel 行为与 KJ 对齐",
"升级 Image Mask Scale As (UTK)：",
"- fit 与 KJ v2 对齐，新增 crop_position，支持预设 pad_color",
"升级 Image Scale By Aspect Ratio (UTK)：",
"- fit 与 KJ v2 对齐，新增 crop_position，background_color 改为预设清单",
"修正 pad_edge 与 pad_edge_pixel 的边缘与角点处理逻辑，匹配 KJ 视觉表现",
],
"1.4.5": [
"新增Best Context Window (UTK)节点：",
"- 计算满足4n+1且位于[min,max]区间的最佳窗口，以最小化补帧",
"- 输出best_window、padding、padded_total、segments",
"- 分类：UniversalToolkit/Tools",
"新增Blockify Mask (UTK)节点：",
"- 将掩码按block_size块化，支持cpu/cuda",
"- 可选二值化binarize与threshold",
"- 分类：UniversalToolkit/Mask",
"统一分类命名：将UniversalToolkit/tools合并为UniversalToolkit/Tools",
"修复：Get Image or Mask Range From Batch (UTK) 分类名不一致问题",
],
"1.4.4": [
"新增Get Image or Mask Range From Batch (UTK)节点：",
"- 支持从图像批次或遮罩批次中提取指定范围的元素",
"- 灵活的索引控制：支持正索引和-1（从末尾开始）",
"- 智能范围处理：自动处理超出范围的情况",
"- 双输入支持：可同时处理图像和遮罩批次",
"- 详细日志输出：显示提取的索引范围和数量",
"- 错误处理：提供清晰的错误信息指导用户",
"- 工具分类：归类到UniversalToolkit/tools分类下",
"- 提升批次处理工作流的灵活性和效率",
],
"1.4.3": [
"修复翻译API节点错误消息显示问题：",
"- 修复API密钥获取网址在错误消息中不显示的问题",
"- 改进特定提供者模式下的错误处理逻辑",
"- 现在API密钥缺失时会显示完整的错误消息和获取网址",
"- 错误消息格式：'Error: Service requires API key. Get it at: [URL]'",
"- 提升用户体验，用户可直接点击链接申请API密钥",
"- 统一了auto模式和特定提供者模式的错误消息显示",
],
"1.4.2": [
"优化翻译API节点服务可靠性：",
"- 移除LibreTranslate和MyMemory等不可靠的翻译服务",
"- 修复Bing Translator API密钥要求，正确标记为需要API密钥",
"- 重新排序翻译服务：Google Translate作为唯一免费服务",
"- 简化提供者列表，仅保留经过验证的可靠服务",
"- 改进Bing Translator错误处理，提供更准确的API密钥验证",
"- 清理代码，移除116行不可靠服务的代码",
"- 提升翻译成功率和用户体验",
],
"1.4.1": [
"修复翻译API节点错误处理和重命名：",
"- 重命名Text Translator节点为Text Translator API (UTK)，更准确反映功能定位",
"- 增强LibreTranslate错误处理：添加JSON解析验证和空响应检查",
"- 修复MyMemory 403速率限制错误处理，提供更友好的错误信息",
"- 为Google Translate添加备用URL，提高服务可靠性",
"- 改进所有翻译服务的错误消息，提供更详细的调试信息",
"- 增强翻译服务健壮性，提升用户体验",
"- 优化免费翻译服务的优先级排序和错误恢复机制",
],
"1.4.0": [
"新增Text Translator API (UTK)节点：",
"- 支持多种翻译服务提供商：Google Translate、LibreTranslate、MyMemory等",
"- 支持自动模式：按优先级尝试多个免费API，失败时自动切换",
"- 支持付费API：DeepL、Azure Translator、GLM-4 Flash等",
"- 支持原生语言名称：中文、日本語、English等，提升用户体验",
"- 智能API密钥检测：自动检测服务商是否需要API密钥",
"- 详细日志输出：提供翻译过程的完整调试信息",
"- 错误处理机制：优雅处理API失败和网络错误",
"- 支持批量翻译和列表输入处理",
],
"1.3.7": [
"新增Image Blend Advance节点和Crop By Mask功能增强：",
"- 新增Image Blend Advance (UTK)节点：高级图像混合与变换功能",
"- 支持17种混合模式：normal、multiply、screen、overlay等专业混合效果",
"- 完整变换功能：位置、缩放、宽高比、旋转、镜像翻转控制",
"- 6种插值方法：lanczos、bicubic、bilinear等高质量缩放算法",
"- 智能边界处理：自动处理图层超出画布的情况",
"- 自动背景生成：未提供背景时自动创建透明背景",
"- Alpha通道支持：自动提取和处理RGBA图像的透明度",
"- 增强Crop By Mask (UTK)节点：新增remaining_area输出",
"- remaining_area显示被裁剪区域：白色矩形标记已处理区域",
"- 完善批处理支持：智能处理多图像和蒙版的对应关系",
],
"1.3.6": [
"增强Image Crop By Mask And Resize节点功能：",
"- 新增4种resize方法：fill、crop、letterbox、stretch",
"- fill模式：缩放填满目标尺寸，可能裁剪边缘（默认）",
"- crop模式：保持宽高比，居中放置，黑色填充不足部分",
"- letterbox模式：保持宽高比，添加黑边，完整保留内容",
"- stretch模式：直接拉伸到目标尺寸，可能扭曲比例",
"- 新增4种插值方法：nearest、bilinear、bicubic、lanczos",
"- 支持高质量Lanczos插值（默认）和快速nearest插值",
"- 完善的resize逻辑：智能处理各种宽高比场景",
"- 提供专业级图像处理选项，满足不同应用需求",
],
"1.3.5": [
"新增Image Crop By Mask And Resize节点和代码优化：",
"- 新增Image Crop By Mask And Resize (UTK)节点：完全按照kjnodes标准实现",
"- 支持16像素对齐：确保所有输出尺寸能被16整除，AI模型友好",
"- 三阶段批处理策略：分析→统一→处理，确保输出尺寸一致性",
"- 智能宽高比处理：基于最大宽高比计算最优目标分辨率",
"- 高质量缩放算法：图像使用Lanczos，mask使用双线性插值",
"- 灵活的分辨率约束：支持min/max crop resolution参数",
"- 恢复Crop By Mask (UTK)节点：保持原有稳定功能不变",
"- 提供两种裁剪选择：原版保持兼容，新版提供kjnodes标准",
],
"1.3.4": [
"新增Lazy Switch KJ节点和重要功能修复：",
"- 新增Lazy Switch KJ (UTK)节点：支持懒加载评估的条件流程控制",
"- 支持任意数据类型的条件切换，提供真正的懒加载机制",
"- 修复Crop By Mask (UTK)节点批处理逻辑：现在正确支持图像和mask批次对应",
"- 改进批处理算法：每个图像使用对应位置的mask进行独立裁剪",
"- 智能处理批次数量不匹配：自动重复或截断mask以匹配图像数量",
"- 增强日志输出：每个图像的裁剪信息单独记录，便于调试",
"- 保持向后兼容性：单图像+单mask的使用方式保持不变",
"- 优化性能：避免不必要的计算，特别适用于条件工作流",
],
"1.3.3": [
"新增多个kjnodes节点移植和架构优化：",
"- 新增Color Match (UTK)节点：支持6种颜色匹配算法，用于图像间色彩转移",
"- 新增Color To Mask (UTK)节点：根据RGB颜色值创建掩码，支持阈值调节",
"- 新增Separate Masks (UTK)节点：分离连通组件为独立掩码，支持3种输出模式",
"- 新增Bbox Visualize (UTK)节点：在图像上绘制边界框，支持xywh和xyxy格式",
"- 重构mask分类架构：创建独立py文件封装，与image分类保持一致",
"- 优化Color Match节点输入顺序：image_target在前，避免bypass节点传递错误",
"- 完善节点分类和导航：所有新节点正确显示在右侧导航面板",
"- 增强依赖管理：添加color-matcher、scipy等必要依赖",
"- 提升代码质量：完整的错误处理、进度显示和参数验证",
],
"1.3.1": [
"修复Audio Crop Process节点duration=0时的裁剪逻辑：",
"- 修复当duration_seconds为0时仍会进行音频裁剪的问题",
"- 优化裁剪逻辑：只有当duration大于0时才进行裁剪",
"- 支持duration=0且offset>0时只应用offset裁剪",
"- 支持duration=0且offset=0时不进行任何裁剪，保持原始音频",
"- 提升音频处理节点的用户体验和功能准确性",
],
"1.3.0": [
"改进Image Remove Alpha节点和优化专业产品图预设：",
"- 改进Image Remove Alpha (UTK)节点：将background_color参数从手动输入改为预设颜色下拉菜单",
"- 新增10种预设颜色选项：black、white、gray、red、green、blue、yellow、cyan、magenta、transparent",
"- 优化专业产品图预设：强化场景驱动环境优先级和商业摄影质量标准",
"- 改进场景驱动环境：严格遵循用户场景要求，避免默认影棚背景",
"- 增强真实灯光与阴影：根据场景上下文匹配灯光风格",
"- 提升专业构图技巧：使用三分法则、引导线和受控景深",
"- 完善环境真实感：产品与环境无缝融合，匹配透视和表面反射",
"- 强化营销级质量标准：保持锐利对焦和清洁专业的色彩分级",
"- 优化用户体验：提供更直观的颜色选择和更精确的场景控制",
],
"1.2.9": [
"优化蓝图视角预设，重命名为设计图模式，支持多种技术图纸风格与详细指令描述。",
],
"1.2.8": [
"新增模特手持特写预设和优化产品摄影预设：",
"- 新增Photo-Model Hand Product Close-Up (模特手持特写)预设，专门用于生成模特手持产品的特写摄影场景",
"- 将Photo-Model Product Trial改名为Photo-Product Lifestyle Scene (产品生活场景图)，更准确反映功能",
"- 升级Photo-Professional Product Photography (专业产品图)预设，强调场景驱动环境优先级",
"- 优化专业产品图预设：严格遵循用户场景要求，避免默认影棚背景",
"- 改进场景匹配：提供购物中心橱窗、咖啡厅、户外公园等具体场景示例",
"- 增强环境匹配灯光：根据场景使用商场环境光、户外自然光、室内温暖灯光",
"- 提升构图焦点：强调自然前景/背景整合，保持产品作为明确焦点",
"- 完善输出要求：详细描述场景设置、相机角度、背景元素、灯光风格和构图",
"- 保持所有预设的完整功能，总数达到32个，覆盖更全面的图像处理需求",
],
"1.2.7": [
"升级专业产品摄影和模特试用产品图预设：",
"- 新增Photo-Model Product Trial (模特试用产品图)预设，专门用于生成模特使用商品的场景图",
"- 升级Photo-Professional Product Photography (专业产品图)预设，使用更精确的商业摄影指令",
"- 优化专业产品图预设：增强场景响应式环境、专业灯光阴影控制、构图焦点卓越性",
"- 改进模特试用产品图预设：强调模特产品互动、生活化环境整合、专业模特呈现",
"- 提升商业摄影质量标准：确保锐利对焦、准确曝光、专业调色",
"- 增强真实故事叙述：创造既真实又令人向往的生活方式故事",
"- 完善输出要求：详细描述摄影设置、灯光安排、背景处理、相机角度和构图风格",
"- 保持所有预设的完整功能，总数达到31个，覆盖更全面的图像处理需求",
],
"1.2.6": [
"进一步优化角色姿势视角变换预设：",
"- 使用更精确的提示词结构，提升指令的准确性",
"- 明确指定相机角度、位置和角色动作的描述要求",
"- 强调角色正在执行的具体动作，让变换更加生动",
"- 优化brief描述，支持更具体的用户需求",
"- 保持所有角色特征的一致性要求",
"- 确保光照、阴影和透视的自然性",
"- 提升预设指令的精确性和实用性",
],
"1.2.5": [
"优化Kontext预设系统，提升用户体验：",
"- 为所有预设添加英文分类前缀，提升节点中的可读性",
"- 重新组织预设分类：Core-核心编辑、Composite-图像合成、Scene-场景环境、Photo-摄影技术、Character-人物变换、Art-艺术风格、Effect-特殊效果、Utility-实用功能",
"- 将花纹提取预设移动到实用功能类，更符合其功能定位",
"- 升级身材改造预设，支持多种身材变化（高矮胖瘦、强壮肌肉、性感身材等）",
"- 优化衣橱改造预设，强调保持角色特征不变和服装自然性",
"- 升级专业产品图预设，确保产品特征不变和环境自然集成",
"- 改进角色姿势视角变换预设，支持多种角度和姿势变化",
"- 结合社区最佳实践，优化所有预设的brief描述",
"- 提升预设指令的精确性和实用性",
"- 保持所有预设的完整功能，总数维持30个",
],
"1.2.4": [
"优化预设分类结构，提升用户体验：",
"- 重新整理预设分类，合并相似类别，减少重复",
"- 将原11个分类优化为7个主要分类：核心编辑、图像合成、场景环境、摄影技术、人物变换、艺术风格、特殊效果",
"- 将视角变换类合并到摄影技术类，将花纹提取类合并到核心编辑类",
"- 将环境变换类合并到场景环境类，简化分类结构",
"- 优化预设顺序，保持万能编辑作为默认选项",
"- 提升预设分类的逻辑性和易用性",
"- 保持所有30个预设的完整功能不变",
],
"1.2.3": [
"新增花纹提取预设和优化用户体验：",
"- 添加Pattern Extraction (花纹提取)预设，支持指定提取对象",
"- 将Universal Editor (万能编辑)预设移到最前面作为默认选项",
"- 优化预设顺序，提升用户使用体验",
"- 支持从任意对象中提取花纹、logo、图案等",
"- 完全支持$user_prompt$占位符替换机制",
"- 预设总数达到30个，覆盖各种图像处理需求",
],
"1.2.2": [
"新增万能编辑预设功能：",
"- 添加Universal Editor (万能编辑)预设，支持精确的图像编辑指令转换",
"- 基于Kontext格式和Flux模型的编辑指令生成",
"- 支持人物、物体、背景、风格、文本等多种编辑类型",
"- 严格遵循9条编辑规则，确保视觉一致性和精确性",
"- 完全支持$user_prompt$占位符替换机制",
"- 优化预设指令结构，移除重复的英文输出要求",
],
"1.2.1": [
"修复自动发布问题：",
"- 精简pyproject.toml依赖配置，仅保留核心依赖",
"- 解决Comfy Registry自动发布失败问题",
"- 更新版本号避免重复发布",
"- 确保依赖配置与Comfy Registry兼容",
],
"1.2.0": [
"版本更新发布：",
"- 全面更新README文档，提供详细的功能介绍和使用指南",
"- 优化项目结构和文档组织",
"- 完善节点功能说明和分类",
"- 更新版本推送代码，确保发布流程顺畅",
"- 提升项目整体文档质量和用户体验",
],
"1.1.9": [
"升级 LoadKontextPresets_UTK 节点功能：",
"- 新增用户提示输入参数（user_prompt），支持多行文本输入",
"- 为所有27个预设添加$user_prompt$占位符，支持动态场景描述",
"- 改造所有预设的brief描述，使其能够根据用户输入生成具体指令",
"- 智能占位符替换：有用户输入时替换为具体需求，无输入时使用通用描述",
"- 支持重新布光、场景传送、季节变换等预设的用户自定义场景",
"- 支持人物变换、艺术风格、特殊效果等预设的个性化需求",
"- 提升VLM模型生成指令的针对性和实用性",
"- 保持原有27个预设的完整功能和分类结构",
],
"1.1.8": [
"新增 LoadKontextPresets_UTK 节点（Kontext VLM System Presets）：",
"- 基于 Kontext 项目集成，提供25种图像变换预设",
"- 支持情境深度融合、无痕融合、场景传送等多种预设",
"- 支持移动镜头、重新布光、专业产品图等专业预设",
"- 支持画面缩放、图像上色、电影海报等创意预设",
"- 支持卡通漫画化、移除文字、更换发型等功能预设",
"- 支持肌肉猛男化、清空家具、室内设计等特殊预设",
"- 支持季节变换、时光旅人、材质置换等艺术预设",
"- 支持微缩世界、幻想领域、衣橱改造等创意预设",
"- 支持艺术风格模仿、蓝图视角、添加倒影等效果预设",
"- 支持像素艺术、铅笔手绘、油画风格等艺术风格预设",
"- 所有预设均按照官方指南进行重写和升级，提供专业的图像变换指令",
],
"1.1.7": [
"新增 ThinkRemover_UTK 节点：",
"- 支持分离文本中的<think>内容和剩余内容，便于上下文处理和提示词优化",
],
"1.1.6": [
"新增 TextboxNode_UTK 节点（文本框节点）：",
"- 基于 ComfyUI-LayerStyle 项目集成",
"- 支持在图像上添加自定义样式的文本框",
"- 支持文本颜色、背景颜色、透明度调节",
"- 支持圆角边框、边框宽度和颜色设置",
"- 支持文本对齐方式（左对齐、居中、右对齐）",
"- 支持自定义字体路径和字体大小",
"- 支持文本框位置精确定位",
"- 支持文本自动换行和最大宽度限制",
"- 支持批量图像处理",
],
"1.1.5": [
"版本更新发布：",
"- 优化ImitationHueNode_UTK节点的色彩迁移算法",
"- 改进皮肤保护功能，提高肤色保持效果",
"- 增强自动亮度、对比度、饱和度调节的稳定性",
"- 完善影调模仿功能的实现",
"- 优化区域色彩迁移的掩码处理",
"- 提升整体色彩迁移的自然度和准确性",
],
"1.1.4": [
"版本更新发布：",
"- 更新插件版本号以支持ComfyUI-Manager和Registry获取",
"- 确保与ComfyUI v3完全兼容",
"- 优化插件元数据和发布配置",
],
"1.1.3": [
"修复 PurgeVRAM_UTK 节点类型不匹配问题：",
"- 创建 AnyType 类实现，参考 ComfyUI-LayerStyle 项目",
"- 解决 'received_type(IMAGE) mismatch input_type(*)' 错误",
"- 支持接受任何类型输入并正确返回",
"修复 CheckMask_UTK 节点 NoneType 错误：",
"- 添加空值检查，防止 mask 为 None 时出错",
"- 改进 tensor2pil 转换失败的处理",
"- 增强错误处理和日志输出",
"完成模块化重构：",
"- 将 common_utils.py 功能拆分到不同目录",
"- 创建 image_converters.py、color_utils.py、logging_utils.py、any_type.py",
"- 删除 common_utils.py，避免依赖冲突",
"- 更新所有相关文件的导入语句",
"- 提高代码可维护性和模块化程度",
],
"1.1.2": [
"修复 DepthMapBlur_UTK 节点 kernel size 类型和 OpenCV 奇数断言问题，保证所有模糊核为正奇数，完全兼容 ComfyUI 规范。",
"修正 EmptyUnitGenerator_UTK 输出 shape，所有节点输入输出严格遵循 ComfyUI 官方规范。",
"完善节点导入路径为相对导入，兼容 ComfyUI v3 插件机制。",
],
"1.1.1": [
"项目重构：",
"- 将大型节点文件按功能拆分成多个独立模块",
"- 创建共用工具函数文件 common_utils.py",
"- 按功能分类：image/、tools/、mask/、audio/",
"- 提高代码可维护性和模块化程度",
"- 参考 ComfyUI-LayerStyle 项目的模块化组织方式",
"新增多个图像处理节点：",
"- ImageScaleByAspectRatio_UTK: 按比例缩放图像",
"- ImageMaskScaleAs_UTK: 按参考图像缩放",
"- ImageScaleRestore_UTK: 恢复图像尺寸",
"- ImageRemoveAlpha_UTK: 移除Alpha通道",
"- ImageCombineAlpha_UTK: 合并Alpha通道",
"- CheckMask_UTK: 检查掩码有效性",
"- PurgeVRAM_UTK: 清理显存",
"- CropByMask_UTK: 基于掩码裁剪",
"- RestoreCropBox_UTK: 恢复裁剪框",
],
"1.1.0": [
"新增 CropByMaskV3_UTK 节点（掩码裁剪节点）:",
"- 基于 ComfyUI-LayerStyle 项目集成",
"- 支持三种检测模式：mask_area、min_bounding_rect、max_inscribed_rect",
"- 支持掩码反转功能",
"- 支持四个方向的预留边距设置",
"- 支持整除倍数调整",
"- 提供裁剪框预览功能",
"新增 RestoreCropBox_UTK 节点（恢复裁剪框节点）:",
"- 基于 ComfyUI-LayerStyle 项目集成",
"- 支持将裁剪后的图像恢复到原始背景",
"- 支持掩码反转功能",
"- 支持自定义裁剪框位置",
"- 自动处理批量图像和掩码",
],
"1.0.9": [
"新增 PurgeVRAM_UTK 节点（显存清理节点）:",
"- 基于 ComfyUI-LayerStyle 项目集成",
"- 支持清理GPU缓存和模型内存",
"- 支持选择性清理（缓存/模型）",
"- 添加内存清理辅助函数",
"- 支持任意类型输入，保持数据流连续性",
],
"1.0.8": [
"新增 ImitationHueNode_UTK 节点（追色节点）:",
"- 基于 ComfyUI-MingNodes 项目集成",
"- 支持图像色彩迁移和追色功能",
"- 支持皮肤保护参数，避免肤色失真",
"- 支持自动亮度、对比度、饱和度调节",
"- 支持影调模仿功能",
"- 支持区域色彩迁移（通过掩码）",
"- 添加 opencv-python 依赖支持",
],
"1.0.7": [
"改进 ImagePadForOutpaintMasked (UTK) 节点:",
"- 新增数据模式（data_mode）参数，支持 'pixel' 和 'percent' 两种模式",
"- 在 'percent' 模式下，允许输入大于100的百分比",
"- 新增背景颜色（background_color）预设选项",
"改进 ImageAndMaskPreview (UTK) 节点:",
"- 将颜色输入从手动输入字符串改为预设颜色下拉菜单",
],
"1.0.6": [
"新增 ImageAndMaskPreview_UTK 节点，用于同时预览图像和掩码：",
"- 支持叠加模式（overlay）：在图像上叠加彩色掩码",
"- 支持并排模式（side_by_side）：图像和掩码并排显示",
"- 支持单独显示模式（mask_only/image_only）",
"- 支持多种掩码颜色和透明度调节",
],
"1.0.5": [
"新增 ImagePadForOutpaintMasked_UTK 节点，用于外绘时扩展图像尺寸：",
"- 支持上下左右四个方向的独立扩展",
"- 支持多种背景颜色（黑、白、灰、透明）",
"- 支持掩码边缘羽化效果",
"- 自动生成对应的掩码用于后续处理",
],
"1.0.4": [
"新增 FillMaskedArea_UTK 节点，支持三种填充模式：",
"- neutral: 使用灰色填充，适合添加全新内容",
"- telea: 基于 Telea 算法的边界填充",
"- navier-stokes: 基于流体动力学的边界填充",
"添加 opencv-python 依赖支持",
],
"1.0.3": [
"新增 MaskAnd_UTK、MaskSub_UTK、MaskAdd_UTK 三个mask像素级运算节点 (UTK)",
"修正节点注册与显示名风格统一，完善导入路径",
],
"1.0.2": [
"删除无用节点 Load Audio Plus Upload (UTK)",
"新增 Audio Crop Process (UTK) 支持原生音频上传",
"修复音频处理相关bug，完善依赖",
],
"1.0.1": [
"改进 ImageConcatenate_UTK 节点：",
"- 添加 match_image_size 参数，支持自动匹配图像尺寸",
"- 添加 max_size 参数，限制最大输出尺寸",
"- 添加背景颜色选项",
"- 优化图像拼接逻辑，保持宽高比",
"改进 ImageConcatenateMulti_UTK 节点：",
"- 与 ImageConcatenate_UTK 保持一致的功能",
"- 支持自动方向选择",
"- 支持图像尺寸匹配",
"- 支持最大尺寸限制",
"- 优化多图拼接逻辑",
],
"1.0.0": [
"初始版本发布",
"包含基础图像处理节点",
"包含文本处理节点",
"包含工具类节点",
],
}
# 导入节点模块
try:
# 工具类节点
# 音频节点
from .nodes.audio.audio_crop_process import \
NODE_CLASS_MAPPINGS as AUDIO_CROP_MAPPINGS
from .nodes.audio.audio_crop_process import \
NODE_DISPLAY_NAME_MAPPINGS as AUDIO_CROP_DISPLAY_MAPPINGS
from .nodes.image.image_and_mask_preview import \
NODE_CLASS_MAPPINGS as AND_MASK_PREVIEW_MAPPINGS
from .nodes.image.image_and_mask_preview import \
NODE_DISPLAY_NAME_MAPPINGS as AND_MASK_PREVIEW_DISPLAY_MAPPINGS
# 图像节点
from .nodes.image.image_concatenate_multi import \
NODE_CLASS_MAPPINGS as CONCATENATE_MULTI_MAPPINGS
from .nodes.image.image_concatenate_multi import \
NODE_DISPLAY_NAME_MAPPINGS as CONCATENATE_MULTI_DISPLAY_MAPPINGS
from .nodes.image.image_pad_for_outpaint_masked import \
NODE_CLASS_MAPPINGS as PAD_OUTPAINT_MAPPINGS
from .nodes.image.image_pad_for_outpaint_masked import \
NODE_DISPLAY_NAME_MAPPINGS as PAD_OUTPAINT_DISPLAY_MAPPINGS
from .nodes.image.imitation_hue_node import \
NODE_CLASS_MAPPINGS as IMITATION_HUE_MAPPINGS
from .nodes.image.imitation_hue_node import \
NODE_DISPLAY_NAME_MAPPINGS as IMITATION_HUE_DISPLAY_MAPPINGS
# 掩码节点
from .nodes.mask import \
NODE_CLASS_MAPPINGS as MASK_MAPPINGS
from .nodes.mask import \
NODE_DISPLAY_NAME_MAPPINGS as MASK_DISPLAY_MAPPINGS
from .nodes.tools.math_expression_node import \
NODE_CLASS_MAPPINGS as MATH_EXPRESSION_MAPPINGS
from .nodes.tools.math_expression_node import \
NODE_DISPLAY_NAME_MAPPINGS as MATH_EXPRESSION_DISPLAY_MAPPINGS
from .nodes.tools.text_concatenate_node import \
NODE_CLASS_MAPPINGS as TEXT_CONCATENATE_MAPPINGS
from .nodes.tools.text_concatenate_node import \
NODE_DISPLAY_NAME_MAPPINGS as TEXT_CONCATENATE_DISPLAY_MAPPINGS
# 文本框节点
from .nodes.tools.textbox_node import \
NODE_CLASS_MAPPINGS as TEXTBOX_MAPPINGS
from .nodes.tools.textbox_node import \
NODE_DISPLAY_NAME_MAPPINGS as TEXTBOX_DISPLAY_MAPPINGS
# API图像生成器节点
from .nodes.tools.api_image_generator import \
NODE_CLASS_MAPPINGS as API_IMAGE_GENERATOR_MAPPINGS
from .nodes.tools.api_image_generator import \
NODE_DISPLAY_NAME_MAPPINGS as API_IMAGE_GENERATOR_DISPLAY_MAPPINGS
except ImportError as e:
print(f"导入错误: {e}")
# 如果模块化导入失败，使用空字典
AUDIO_CROP_MAPPINGS = {}
AUDIO_CROP_DISPLAY_MAPPINGS = {}
MASK_MAPPINGS = {}
MASK_DISPLAY_MAPPINGS = {}
CONCATENATE_MULTI_MAPPINGS = {}
CONCATENATE_MULTI_DISPLAY_MAPPINGS = {}
PAD_OUTPAINT_MAPPINGS = {}
PAD_OUTPAINT_DISPLAY_MAPPINGS = {}
AND_MASK_PREVIEW_MAPPINGS = {}
AND_MASK_PREVIEW_DISPLAY_MAPPINGS = {}
IMITATION_HUE_MAPPINGS = {}
IMITATION_HUE_DISPLAY_MAPPINGS = {}
TEXTBOX_MAPPINGS = {}
TEXTBOX_DISPLAY_MAPPINGS = {}
TEXT_CONCATENATE_MAPPINGS = {}
TEXT_CONCATENATE_DISPLAY_MAPPINGS = {}
MATH_EXPRESSION_MAPPINGS = {}
MATH_EXPRESSION_DISPLAY_MAPPINGS = {}
API_IMAGE_GENERATOR_MAPPINGS = {}
API_IMAGE_GENERATOR_DISPLAY_MAPPINGS = {}
# 尝试导入其他可能有依赖的节点
try:
from .nodes.image.fill_masked_area import \
NODE_CLASS_MAPPINGS as FILL_MASKED_MAPPINGS
from .nodes.image.fill_masked_area import \
NODE_DISPLAY_NAME_MAPPINGS as FILL_MASKED_DISPLAY_MAPPINGS
except ImportError:
FILL_MASKED_MAPPINGS = {}
FILL_MASKED_DISPLAY_MAPPINGS = {}
try:
from .nodes.audio.load_audio import \
NODE_CLASS_MAPPINGS as LOAD_AUDIO_MAPPINGS
from .nodes.audio.load_audio import \
NODE_DISPLAY_NAME_MAPPINGS as LOAD_AUDIO_DISPLAY_MAPPINGS
except ImportError:
LOAD_AUDIO_MAPPINGS = {}
LOAD_AUDIO_DISPLAY_MAPPINGS = {}
try:
from .nodes.image.empty_unit_generator import \
NODE_CLASS_MAPPINGS as EMPTY_UNIT_MAPPINGS
from .nodes.image.empty_unit_generator import \
NODE_DISPLAY_NAME_MAPPINGS as EMPTY_UNIT_DISPLAY
except ImportError:
EMPTY_UNIT_MAPPINGS = {}
EMPTY_UNIT_DISPLAY = {}
try:
from .nodes.image.image_ratio_detector import \
NODE_CLASS_MAPPINGS as RATIO_MAPPINGS
from .nodes.image.image_ratio_detector import \
NODE_DISPLAY_NAME_MAPPINGS as RATIO_DISPLAY
except ImportError:
RATIO_MAPPINGS = {}
RATIO_DISPLAY = {}
try:
from .nodes.image.depth_map_blur import \
NODE_CLASS_MAPPINGS as DEPTH_BLUR_MAPPINGS
from .nodes.image.depth_map_blur import \
NODE_DISPLAY_NAME_MAPPINGS as DEPTH_BLUR_DISPLAY
except ImportError:
DEPTH_BLUR_MAPPINGS = {}
DEPTH_BLUR_DISPLAY = {}
try:
from .nodes.image.image_concatenate import \
NODE_CLASS_MAPPINGS as CONCAT_MAPPINGS
from .nodes.image.image_concatenate import \
NODE_DISPLAY_NAME_MAPPINGS as CONCAT_DISPLAY
except ImportError:
CONCAT_MAPPINGS = {}
CONCAT_DISPLAY = {}
try:
from .nodes.image.image_scale_by_aspect_ratio import \
NODE_CLASS_MAPPINGS as SCALE_ASPECT_MAPPINGS
from .nodes.image.image_scale_by_aspect_ratio import \
NODE_DISPLAY_NAME_MAPPINGS as SCALE_ASPECT_DISPLAY
except ImportError:
SCALE_ASPECT_MAPPINGS = {}
SCALE_ASPECT_DISPLAY = {}
try:
from .nodes.image.image_mask_scale_as import \
NODE_CLASS_MAPPINGS as MASK_SCALE_MAPPINGS
from .nodes.image.image_mask_scale_as import \
NODE_DISPLAY_NAME_MAPPINGS as MASK_SCALE_DISPLAY
except ImportError:
MASK_SCALE_MAPPINGS = {}
MASK_SCALE_DISPLAY = {}
try:
from .nodes.image.image_scale_restore import \
NODE_CLASS_MAPPINGS as SCALE_RESTORE_MAPPINGS
from .nodes.image.image_scale_restore import \
NODE_DISPLAY_NAME_MAPPINGS as SCALE_RESTORE_DISPLAY
except ImportError:
SCALE_RESTORE_MAPPINGS = {}
SCALE_RESTORE_DISPLAY = {}
try:
from .nodes.image.image_remove_alpha import \
NODE_CLASS_MAPPINGS as REMOVE_ALPHA_MAPPINGS
from .nodes.image.image_remove_alpha import \
NODE_DISPLAY_NAME_MAPPINGS as REMOVE_ALPHA_DISPLAY
except ImportError:
REMOVE_ALPHA_MAPPINGS = {}
REMOVE_ALPHA_DISPLAY = {}
try:
from .nodes.image.image_combine_alpha import \
NODE_CLASS_MAPPINGS as COMBINE_ALPHA_MAPPINGS
from .nodes.image.image_combine_alpha import \
NODE_DISPLAY_NAME_MAPPINGS as COMBINE_ALPHA_DISPLAY
except ImportError:
COMBINE_ALPHA_MAPPINGS = {}
COMBINE_ALPHA_DISPLAY = {}
try:
from .nodes.image.check_mask import \
NODE_CLASS_MAPPINGS as CHECK_MASK_MAPPINGS
from .nodes.image.check_mask import \
NODE_DISPLAY_NAME_MAPPINGS as CHECK_MASK_DISPLAY
except ImportError:
CHECK_MASK_MAPPINGS = {}
CHECK_MASK_DISPLAY = {}
try:
from .nodes.tools.purge_vram import \
NODE_CLASS_MAPPINGS as PURGE_VRAM_MAPPINGS
from .nodes.tools.purge_vram import \
NODE_DISPLAY_NAME_MAPPINGS as PURGE_VRAM_DISPLAY
except ImportError:
PURGE_VRAM_MAPPINGS = {}
PURGE_VRAM_DISPLAY = {}
try:
from .nodes.image.crop_by_mask import \
NODE_CLASS_MAPPINGS as CROP_MASK_MAPPINGS
from .nodes.image.crop_by_mask import \
NODE_DISPLAY_NAME_MAPPINGS as CROP_MASK_DISPLAY
except ImportError:
CROP_MASK_MAPPINGS = {}
CROP_MASK_DISPLAY = {}
try:
from .nodes.image.restore_crop_box import \
NODE_CLASS_MAPPINGS as RESTORE_CROP_MAPPINGS
from .nodes.image.restore_crop_box import \
NODE_DISPLAY_NAME_MAPPINGS as RESTORE_CROP_DISPLAY
except ImportError:
RESTORE_CROP_MAPPINGS = {}
RESTORE_CROP_DISPLAY = {}
try:
from .nodes.image.color_match_standalone import \
NODE_CLASS_MAPPINGS as COLOR_MATCH_MAPPINGS
from .nodes.image.color_match_standalone import \
NODE_DISPLAY_NAME_MAPPINGS as COLOR_MATCH_DISPLAY
except ImportError:
COLOR_MATCH_MAPPINGS = {}
COLOR_MATCH_DISPLAY = {}
try:
from .nodes.image.bbox_visualize import \
NODE_CLASS_MAPPINGS as BBOX_VISUALIZE_MAPPINGS
from .nodes.image.bbox_visualize import \
NODE_DISPLAY_NAME_MAPPINGS as BBOX_VISUALIZE_DISPLAY
except ImportError:
BBOX_VISUALIZE_MAPPINGS = {}
BBOX_VISUALIZE_DISPLAY = {}
try:
from .nodes.image.image_crop_by_mask_and_resize import \
NODE_CLASS_MAPPINGS as IMAGE_CROP_RESIZE_MAPPINGS
from .nodes.image.image_crop_by_mask_and_resize import \
NODE_DISPLAY_NAME_MAPPINGS as IMAGE_CROP_RESIZE_DISPLAY
except ImportError:
IMAGE_CROP_RESIZE_MAPPINGS = {}
IMAGE_CROP_RESIZE_DISPLAY = {}
try:
from .nodes.image.image_blend_advance_v3 import \
NODE_CLASS_MAPPINGS as IMAGE_BLEND_MAPPINGS
from .nodes.image.image_blend_advance_v3 import \
NODE_DISPLAY_NAME_MAPPINGS as IMAGE_BLEND_DISPLAY
except ImportError:
IMAGE_BLEND_MAPPINGS = {}
IMAGE_BLEND_DISPLAY = {}
try:
from .nodes.tools.think_remover_node import \
NODE_CLASS_MAPPINGS as THINK_REMOVER_MAPPINGS
from .nodes.tools.think_remover_node import \
NODE_DISPLAY_NAME_MAPPINGS as THINK_REMOVER_DISPLAY_MAPPINGS
except ImportError as e:
print(f"导入错误: {e}")
# ... 其它 except ...
THINK_REMOVER_MAPPINGS = {}
THINK_REMOVER_DISPLAY_MAPPINGS = {}
try:
from .nodes.tools.lora_info import \
NODE_CLASS_MAPPINGS as LORA_INFO_MAPPINGS
from .nodes.tools.lora_info import \
NODE_DISPLAY_NAME_MAPPINGS as LORA_INFO_DISPLAY_MAPPINGS
except ImportError as e:
print(f"导入错误: {e}")
LORA_INFO_MAPPINGS = {}
LORA_INFO_DISPLAY_MAPPINGS = {}
try:
from .nodes.tools.kontext_presets import \
NODE_CLASS_MAPPINGS as KONTEXT_PRESETS_MAPPINGS
from .nodes.tools.kontext_presets import \
NODE_DISPLAY_NAME_MAPPINGS as KONTEXT_PRESETS_DISPLAY_MAPPINGS
except ImportError as e:
print(f"导入错误: {e}")
KONTEXT_PRESETS_MAPPINGS = {}
KONTEXT_PRESETS_DISPLAY_MAPPINGS = {}
try:
from .nodes.tools.prompt_helper import \
NODE_CLASS_MAPPINGS as PROMPT_HELPER_MAPPINGS
from .nodes.tools.prompt_helper import \
NODE_DISPLAY_NAME_MAPPINGS as PROMPT_HELPER_DISPLAY_MAPPINGS
except ImportError as e:
print(f"导入错误: {e}")
PROMPT_HELPER_MAPPINGS = {}
PROMPT_HELPER_DISPLAY_MAPPINGS = {}
try:
from .nodes.tools.color_to_mask import \
NODE_CLASS_MAPPINGS as COLOR_TO_MASK_MAPPINGS
from .nodes.tools.color_to_mask import \
NODE_DISPLAY_NAME_MAPPINGS as COLOR_TO_MASK_DISPLAY
except ImportError:
COLOR_TO_MASK_MAPPINGS = {}
COLOR_TO_MASK_DISPLAY = {}
try:
from .nodes.tools.lazy_switch import \
NODE_CLASS_MAPPINGS as LAZY_SWITCH_MAPPINGS
from .nodes.tools.lazy_switch import \
NODE_DISPLAY_NAME_MAPPINGS as LAZY_SWITCH_DISPLAY
except ImportError:
LAZY_SWITCH_MAPPINGS = {}
LAZY_SWITCH_DISPLAY = {}
try:
from .nodes.tools.text_translator import \
NODE_CLASS_MAPPINGS as TEXT_TRANSLATOR_API_MAPPINGS
from .nodes.tools.text_translator import \
NODE_DISPLAY_NAME_MAPPINGS as TEXT_TRANSLATOR_API_DISPLAY
except ImportError:
TEXT_TRANSLATOR_API_MAPPINGS = {}
TEXT_TRANSLATOR_API_DISPLAY = {}
try:
from .nodes.tools.get_image_range_from_batch import \
NODE_CLASS_MAPPINGS as GET_IMAGE_RANGE_MAPPINGS
from .nodes.tools.get_image_range_from_batch import \
NODE_DISPLAY_NAME_MAPPINGS as GET_IMAGE_RANGE_DISPLAY
from .nodes.tools.load_video_frames import \
NODE_CLASS_MAPPINGS as EXTRACT_VIDEO_FRAMES_MAPPINGS
from .nodes.tools.load_video_frames import \
NODE_DISPLAY_NAME_MAPPINGS as EXTRACT_VIDEO_FRAMES_DISPLAY
from .nodes.tools.show_any import \
NODE_CLASS_MAPPINGS as SHOW_ANY_MAPPINGS
from .nodes.tools.show_any import \
NODE_DISPLAY_NAME_MAPPINGS as SHOW_ANY_DISPLAY
from .nodes.tools.optimal_context_window_node import \
NODE_CLASS_MAPPINGS as BEST_CONTEXT_WINDOW_MAPPINGS
from .nodes.tools.optimal_context_window_node import \
NODE_DISPLAY_NAME_MAPPINGS as BEST_CONTEXT_WINDOW_DISPLAY
from .nodes.mask.blockify_mask import \
NODE_CLASS_MAPPINGS as BLOCKIFY_MASK_MAPPINGS
from .nodes.mask.blockify_mask import \
NODE_DISPLAY_NAME_MAPPINGS as BLOCKIFY_MASK_DISPLAY
from .nodes.image.resize_image_ver_kj import \
NODE_CLASS_MAPPINGS as RESIZE_VER_KJ_MAPPINGS
from .nodes.image.resize_image_ver_kj import \
NODE_DISPLAY_NAME_MAPPINGS as RESIZE_VER_KJ_DISPLAY
from .nodes.image.image_batch_extend_with_overlap import \
NODE_CLASS_MAPPINGS as IMAGE_BATCH_EXTEND_MAPPINGS
from .nodes.image.image_batch_extend_with_overlap import \
NODE_DISPLAY_NAME_MAPPINGS as IMAGE_BATCH_EXTEND_DISPLAY
from .nodes.image.save_image_plus import \
NODE_CLASS_MAPPINGS as SAVE_IMAGE_PLUS_MAPPINGS
from .nodes.image.save_image_plus import \
NODE_DISPLAY_NAME_MAPPINGS as SAVE_IMAGE_PLUS_DISPLAY
except ImportError as e:
print(f"[UniversalToolkit] 导入错误: {e}")
GET_IMAGE_RANGE_MAPPINGS = {}
GET_IMAGE_RANGE_DISPLAY = {}
EXTRACT_VIDEO_FRAMES_MAPPINGS = {}
EXTRACT_VIDEO_FRAMES_DISPLAY = {}
SHOW_ANY_MAPPINGS = {}
SHOW_ANY_DISPLAY = {}
BEST_CONTEXT_WINDOW_MAPPINGS = {}
BEST_CONTEXT_WINDOW_DISPLAY = {}
BLOCKIFY_MASK_MAPPINGS = {}
BLOCKIFY_MASK_DISPLAY = {}
RESIZE_VER_KJ_MAPPINGS = {}
RESIZE_VER_KJ_DISPLAY = {}
IMAGE_BATCH_EXTEND_MAPPINGS = {}
IMAGE_BATCH_EXTEND_DISPLAY = {}
SAVE_IMAGE_PLUS_MAPPINGS = {}
SAVE_IMAGE_PLUS_DISPLAY = {}
# 合并所有节点映射
NODE_CLASS_MAPPINGS = {}
NODE_CLASS_MAPPINGS.update(EMPTY_UNIT_MAPPINGS)
NODE_CLASS_MAPPINGS.update(RATIO_MAPPINGS)
NODE_CLASS_MAPPINGS.update(DEPTH_BLUR_MAPPINGS)
NODE_CLASS_MAPPINGS.update(CONCAT_MAPPINGS)
NODE_CLASS_MAPPINGS.update(CONCATENATE_MULTI_MAPPINGS)
NODE_CLASS_MAPPINGS.update(PAD_OUTPAINT_MAPPINGS)
NODE_CLASS_MAPPINGS.update(AND_MASK_PREVIEW_MAPPINGS)
NODE_CLASS_MAPPINGS.update(IMITATION_HUE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(SCALE_ASPECT_MAPPINGS)
NODE_CLASS_MAPPINGS.update(MASK_SCALE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(SCALE_RESTORE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(REMOVE_ALPHA_MAPPINGS)
NODE_CLASS_MAPPINGS.update(COMBINE_ALPHA_MAPPINGS)
NODE_CLASS_MAPPINGS.update(CHECK_MASK_MAPPINGS)
NODE_CLASS_MAPPINGS.update(PURGE_VRAM_MAPPINGS)
NODE_CLASS_MAPPINGS.update(CROP_MASK_MAPPINGS)
NODE_CLASS_MAPPINGS.update(RESTORE_CROP_MAPPINGS)
NODE_CLASS_MAPPINGS.update(COLOR_MATCH_MAPPINGS)
NODE_CLASS_MAPPINGS.update(BBOX_VISUALIZE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(IMAGE_CROP_RESIZE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(IMAGE_BLEND_MAPPINGS)
NODE_CLASS_MAPPINGS.update(FILL_MASKED_MAPPINGS)
NODE_CLASS_MAPPINGS.update(MASK_MAPPINGS)
NODE_CLASS_MAPPINGS.update(LOAD_AUDIO_MAPPINGS)
NODE_CLASS_MAPPINGS.update(AUDIO_CROP_MAPPINGS)
NODE_CLASS_MAPPINGS.update(TEXTBOX_MAPPINGS)
NODE_CLASS_MAPPINGS.update(TEXT_CONCATENATE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(MATH_EXPRESSION_MAPPINGS)
NODE_CLASS_MAPPINGS.update(API_IMAGE_GENERATOR_MAPPINGS)
NODE_CLASS_MAPPINGS.update(THINK_REMOVER_MAPPINGS)
NODE_CLASS_MAPPINGS.update(LORA_INFO_MAPPINGS)
NODE_CLASS_MAPPINGS.update(KONTEXT_PRESETS_MAPPINGS)
NODE_CLASS_MAPPINGS.update(PROMPT_HELPER_MAPPINGS)
NODE_CLASS_MAPPINGS.update(COLOR_TO_MASK_MAPPINGS)
NODE_CLASS_MAPPINGS.update(LAZY_SWITCH_MAPPINGS)
NODE_CLASS_MAPPINGS.update(TEXT_TRANSLATOR_API_MAPPINGS)
NODE_CLASS_MAPPINGS.update(GET_IMAGE_RANGE_MAPPINGS)
NODE_CLASS_MAPPINGS.update(EXTRACT_VIDEO_FRAMES_MAPPINGS)
NODE_CLASS_MAPPINGS.update(SHOW_ANY_MAPPINGS)
NODE_CLASS_MAPPINGS.update(BEST_CONTEXT_WINDOW_MAPPINGS)
NODE_CLASS_MAPPINGS.update(BLOCKIFY_MASK_MAPPINGS)
NODE_CLASS_MAPPINGS.update(RESIZE_VER_KJ_MAPPINGS)
NODE_CLASS_MAPPINGS.update(IMAGE_BATCH_EXTEND_MAPPINGS)
NODE_CLASS_MAPPINGS.update(SAVE_IMAGE_PLUS_MAPPINGS)
# 合并显示名称映射
NODE_DISPLAY_NAME_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS.update(EMPTY_UNIT_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(RATIO_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(DEPTH_BLUR_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(CONCAT_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(CONCATENATE_MULTI_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(PAD_OUTPAINT_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(AND_MASK_PREVIEW_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(IMITATION_HUE_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(SCALE_ASPECT_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(MASK_SCALE_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(SCALE_RESTORE_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(REMOVE_ALPHA_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(COMBINE_ALPHA_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(CHECK_MASK_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(PURGE_VRAM_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(CROP_MASK_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(RESTORE_CROP_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(COLOR_MATCH_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(BBOX_VISUALIZE_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(IMAGE_CROP_RESIZE_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(IMAGE_BLEND_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(FILL_MASKED_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(LOAD_AUDIO_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(AUDIO_CROP_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(MASK_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(TEXTBOX_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(TEXT_CONCATENATE_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(MATH_EXPRESSION_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(API_IMAGE_GENERATOR_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(THINK_REMOVER_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(LORA_INFO_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(KONTEXT_PRESETS_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(PROMPT_HELPER_DISPLAY_MAPPINGS)
NODE_DISPLAY_NAME_MAPPINGS.update(COLOR_TO_MASK_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(LAZY_SWITCH_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(TEXT_TRANSLATOR_API_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(GET_IMAGE_RANGE_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(EXTRACT_VIDEO_FRAMES_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(SHOW_ANY_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(BEST_CONTEXT_WINDOW_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(BLOCKIFY_MASK_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(RESIZE_VER_KJ_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(IMAGE_BATCH_EXTEND_DISPLAY)
NODE_DISPLAY_NAME_MAPPINGS.update(SAVE_IMAGE_PLUS_DISPLAY)
NODE_CATEGORIES = {
"UniversalToolkit": [
"EmptyUnitGenerator_UTK",
"ImageRatioDetector_UTK",
"DepthMapBlur_UTK",
"ImageConcatenate_UTK",
"ImageConcatenateMulti_UTK",
"ImagePadForOutpaintMasked_UTK",
"FillMaskedArea_UTK",
"ImageAndMaskPreview_UTK",
"LoadAudioPlusFromPath_UTK",
"AudioCropProcessUTK",
"MaskAnd_UTK",
"MaskSub_UTK",
"MaskAdd_UTK",
"ImitationHueNode_UTK",
"ImageScaleByAspectRatio_UTK",
"ImageMaskScaleAs_UTK",
"ImageScaleRestore_UTK",
"ImageRemoveAlpha_UTK",
"ImageCombineAlpha_UTK",
"CheckMask_UTK",
"PurgeVRAM_UTK",
"CropByMask_UTK",
"RestoreCropBox_UTK",
"ColorMatch_UTK",
"BboxVisualize_UTK",
"ImageCropByMaskAndResize_UTK",
"ImageBlendAdvance_UTK",
"TextboxNode_UTK",
"TextConcatenate_UTK",
"MathExpression_UTK",
"ThinkRemover_UTK",
"LoraInfo_UTK",
"LoadKontextPresets_UTK",
"ColorToMask_UTK",
"LazySwitchKJ_UTK",
"APIImageGenerator_UTK",
"TextTranslatorAPI_UTK",
"GetImageRangeFromBatch_UTK",
"Extract_Video_Frames_UTK",
"ShowAny_UTK",
"BestContextWindow_UTK",
"BlockifyMask_UTK",
"ResizeImageVerKJ_UTK",
"ImageBatchExtendWithOverlap_UTK",
"SaveImagePlus_UTK",
]
}
# 导出WEB_DIRECTORY以便ComfyUI加载JavaScript文件
import os
WEB_DIRECTORY = os.path.join(os.path.dirname(__file__), "web")
__all__ = [
"NODE_CLASS_MAPPINGS",
"NODE_DISPLAY_NAME_MAPPINGS",
"NODE_CATEGORIES",
"WEB_DIRECTORY",
"__version__",
"__author__",
"__email__",
"__url__",
"CHANGELOG",
]
# 调试：输出所有注册节点的分组属性
if __name__ == "__main__":
with open("node_category_report.txt", "w", encoding="utf-8") as f:
f.write("=== UniversalToolkit 节点分组属性清单 ===\n")
for k, v in NODE_CLASS_MAPPINGS.items():
cat = getattr(v, "CATEGORY", "无CATEGORY")
f.write(f"{k}: CATEGORY = {cat}\n")
print("节点分组清单已导出到 node_category_report.txt")
Reference in New Issue View Git Blame Copy Permalink