Cyber Dick Lang
6a8be76176
- 将ImageBlendAdvanceV3_UTK重命名为ImageBlendAdvance_UTK - 更新节点显示名称为Image Blend Advance (UTK) - 更新所有相关的导入和注册逻辑 - 更新更新日志中的描述
806 lines
37 KiB
Python
806 lines
37 KiB
Python
"""
|
||
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.3.7"
|
||
__author__ = "CyberDickLang"
|
||
__email__ = "286878701@qq.com"
|
||
__url__ = "https://github.com/whmc76"
|
||
|
||
# 更新日志
|
||
CHANGELOG = {
|
||
"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
|
||
|
||
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 = {}
|
||
|
||
# 尝试导入其他可能有依赖的节点
|
||
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 = {}
|
||
|
||
# 合并所有节点映射
|
||
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(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_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(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_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",
|
||
]
|
||
}
|
||
|
||
__all__ = [
|
||
"NODE_CLASS_MAPPINGS",
|
||
"NODE_DISPLAY_NAME_MAPPINGS",
|
||
"NODE_CATEGORIES",
|
||
"__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")
|