fix:easy kSamplerInpainting and js file set full relative path

.gitignore

@@ -10,4 +10,5 @@ autocomplete/**
 docs/**
 .vscode/
 .idea/
-mmb-preset.custom.txt
+mmb-preset.custom.txt
+config.yaml

README.en.md

@@ -28,10 +28,50 @@
 - Fooocus Inpaint integration
 - Integration of common logical calculations, conversion of types, display of all types, etc.
 - Background removal nodes for the RMBG-1.4 model supporting BriaAI, [BriaAI Guide](https://huggingface.co/briaai/RMBG-1.4)
-
+- Forcibly cleared the memory usage of the comfy UI model are supported
+- Stable Diffusion 3 multi-account API nodes are supported
+- 
 ## Changelog
 
-**v1.1.2 (2024/3/25)**
+**v1.1.6**
+
+- Added **alignYourSteps** to **schedulder** widget in all `easy preSampling` and `easy fullkSampler`
+- Added **Preview&Choose** to **image_output** widget in `easy kSampler` & `easy fullkSampler`
+- Added `easy styleAlignedBatchAlign` - Credit of [style_aligned_comfy](https://github.com/brianfitzgerald/style_aligned_comfy)
+- Added `easy ckptNames`
+- Added `easy controlnetNames`
+- Added `easy imagesSplitimage` - Batch images split into single images
+- Added `easy imageCount` - Get Image Count
+- Added `easy textSwitch` - Text Switch
+
+**v1.1.5**
+
+- Rewrite `easy cleanGPUUsed` - the memory usage of the comfyUI can to be cleared
+- Added `easy humanSegmentation` - Human Part Segmentation
+- Added `easy imageColorMatch`
+- Added `easy ipadapterApplyRegional`
+- Added `easy ipadapterApplyFromParams`
+- Added `easy imageInterrogator` - Image To Prompt
+- Added `easy stableDiffusion3API` - Easy Stable Diffusion 3 Multiple accounts API Node
+
+**v1.1.4**
+
+- Added `easy preSamplingCustom` - Custom-PreSampling, can be supported cosXL-edit
+- Added `easy ipadapterStyleComposition`
+- Added the right-click menu to view checkpoints and lora information in all Loaders
+- Fixed `easy preSamplingNoiseIn`、`easy latentNoisy`、`east Unsampler` compatible with ComfyUI Revision>=2098 [0542088e] or later
+ 
+
+**v1.1.3**
+
+- `easy ipadapterApply` Added **COMPOSITION** preset
+- Supported [ResAdapter](https://huggingface.co/jiaxiangc/res-adapter) when load ResAdapter lora
+- Added `easy promptLine`
+- Added `easy promptReplace`
+- Added `easy promptConcat`
+- `easy wildcards` Added **multiline_mode**  
+
+**v1.1.2**
 
 - Optimized some of the recommended nodes for slots related to EasyUse
 - Added **Enable ContextMenu Auto Nest Subdirectories** The setting item is enabled by default, and it can be classified into subdirectories, checkpoints and loras previews
@@ -43,14 +83,11 @@
 - Added `easy ipadapterApplyEmbeds`
 - Added `easy preMaskDetailerFix`
 - Fixed `easy stylesSelector` is change the prompt when not select the style
-
-(4c25580)
-
-- `easy kSamplerInpainting` add *additional* widget，you can choose 'Differential Diffusion' or 'Only InpaintModelConditioning'
 - Fixed `easy pipeEdit` error when add lora to prompt
 - Fixed layerDiffuse xyplot bug
+- `easy kSamplerInpainting` add *additional* widget，you can choose 'Differential Diffusion' or 'Only InpaintModelConditioning'
 
-**v1.1.1 (2024/3/16)**
+**v1.1.1**
 
 - The issue that the seed is 0 when a node with a seed control is added and **control before generate** is fixed for the first time run queue prompt.
 - `easy preSamplingAdvanced` Added **return_with_leftover_noise**
@@ -60,7 +97,7 @@
 - Remove forced **control_before_generate** settings。 If you want to use control_before_generate, change widget_value_control_mode to before in system settings
 - Added `easy imageRemBg` - The default is BriaAI's RMBG-1.4 model, which removes the background effect more and faster
 
-**v1.1.0 (d5ff84e)**
+**v1.1.0**
 
 - Added `easy imageSplitList` - to split every N images
 - Added `easy preSamplingDiffusionADDTL` - It can modify foreground、background or blended additional prompt
@@ -76,15 +113,18 @@
 - Fixed `easy instantIDApply` mask not input right
 
 
-**v1.0.9 (ff1add1)**
+<details>
+<summary><b>v1.0.9</b></summary>
 
 - Fixed the error when ComfyUI-Impack-Pack and ComfyUI_InstantID were not installed
 - Fixed `easy pipeIn`
 - Added `easy instantIDApply` - you need installed [ComfyUI_InstantID](https://github.com/cubiq/ComfyUI_InstantID) fisrt, Workflow[Example](https://github.com/yolain/ComfyUI-Easy-Use/blob/main/README.en.md#InstantID)
 - Fixed `easy detailerFix` not added to the list of nodes available for saving images formatting extensions
 - Fixed `easy XYInputs: PromptSR` errors are reported when replacing negative prompts
+</details>
 
-**v1.0.8 (f28cbf7)**
+<details>
+<summary><b>v1.0.8</b></summary>
 
 - `easy cascadeLoader` stage_c and stage_b support the checkpoint model (Download [checkpoints](https://huggingface.co/stabilityai/stable-cascade/tree/main/comfyui_checkpoints) models) 
 - `easy styleSelector` The search box is modified to be case-insensitive
@@ -94,8 +134,10 @@
 - Fixed the error of SDXLClipModel in ComfyUI revision 2016[c2cb8e88] and above (the revision number was judged to be compatible with the old revision)
 - Fixed `easy detailerFix` generation error when batch size is greater than 1
 - Optimize the code, reduce a lot of redundant code and improve the running speed
+</details>
 
-**v1.0.7 (2024-02-19)**
+<details>
+<summary><b>v1.0.7</b></summary>
 
 - Added `easy cascadeLoader` - stable cascade Loader
 - Added `easy preSamplingCascade` - stable cascade preSampling Settings
@@ -103,22 +145,28 @@
 - Added `easy cascadeKSampler` - stable cascade stage-c ksampler simple
 - 
 - Optimize the image to image[Example](https://github.com/yolain/ComfyUI-Easy-Use/blob/main/README.en.md#image-to-image)
+</details>
 
-**v1.0.6**
+<details>
+<summary><b>v1.0.6</b></summary>
 
 - Added `easy XYInputs: Checkpoint`
 - Added `easy XYInputs: Lora`
 - `easy seed` can manually switch the random seed when increasing the fixed seed value
 - Fixed `easy fullLoader` and all loaders to automatically adjust the node size when switching LoRa
 - Removed the original ttn image saving logic and adapted to the default image saving format extension of ComfyUI
+</details>
 
-- **v1.0.5**
+<details>
+<summary><b>v1.0.5</b></summary>
 
 - Added `easy isSDXL` 
 - Added prompt word control on `easy svdLoader`, which can be used with open_clip model
 - Added **populated_text** on `easy wildcards`, wildcard populated text can be output
+</details>
 
-**v1.0.4**
+<details>
+<summary><b>v1.0.4</b></summary>
 
 - `easy showAnything` added support for converting other types (e.g., tensor conditions, images, etc.)
 - Added `easy showLoaderSettingsNames` can display the model and VAE name in the output loader assembly
@@ -138,9 +186,10 @@
 - Changing the first-time install node package no longer automatically replaces the theme, you need to manually adjust and refresh the page
 - `easy imageSave` added **only_preivew**
 - Adjust the `easy latentCompositeMaskedWithCond` node
+</details>
 
-
-**v1.0.3**
+<details>
+<summary><b>v1.0.3</b></summary>
 
 - Added `easy stylesSelector`
 - Added **scale_soft_weights** in `easy controlnetLoader` and `easy controlnetLoaderADV` 
@@ -158,9 +207,10 @@
 
 - Adjust the UI theme, divided into two sets of styles: the official default background and the dark black background, which can be switched in the color palette in the settings
 - Modify the styles path to be compatible with other environments
+</details>
 
-
-**v1.0.2**
+<details>
+<summary><b>v1.0.2</b></summary>
 
 - Added `easy XYPlotAdvanced` and some nodes about `easy XYInputs`
 - Added **Alt+1-Alt+9** Shortcut keys to quickly paste node presets for Node templates (corresponding to 1~9 sequences)
@@ -177,7 +227,7 @@
 - Removed `easy imageRemBg`
 - Remove the introductory diagram and workflow files from the package to reduce the package size
 - Replaced the font file used in the generation of XY diagrams
-
+</details>
 
 <details>
 <summary><b>v1.0.1</b></summary>
@@ -257,20 +307,22 @@
 
 Disclaimer: Opened source was not easy. I have a lot of respect for the contributions of these original authors. I just did some integration and optimization.
 
-| Nodes Name(Search Name)    | Related libraries                                                                        | Library-related node     |
-|:---------------------------|:----------------------------------------------------------------------------|:-------------------------|
-| easy setNode               | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.SetNode          |
-| easy getNode               | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.GetNode          |
-| easy bookmark              | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | Bookmark 🔖              |
-| easy portraitMarker        | [comfyui-portrait-master](https://github.com/florestefano1975/comfyui-portrait-master) | Portrait Master          |
-| easy LLLiteLoader          | [ControlNet-LLLite-ComfyUI](https://github.com/kohya-ss/ControlNet-LLLite-ComfyUI) | LLLiteLoader             |
-| easy globalSeed            | [ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) | Global Seed (Inspire)    | 
-| easy preSamplingDynamicCFG | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull  | 
-| dynamicThresholdingFull    | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull  | 
-| easy imageInsetCrop        | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | ImageInsetCrop           | 
-| easy poseEditor            | [ComfyUI_Custom_Nodes_AlekPet](https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet) | poseNode                 | 
+| Nodes Name(Search Name)        | Related libraries                                                                        | Library-related node     |
+|:-------------------------------|:----------------------------------------------------------------------------|:-------------------------|
+| easy setNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.SetNode          |
+| easy getNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.GetNode          |
+| easy bookmark                  | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | Bookmark 🔖              |
+| easy portraitMarker            | [comfyui-portrait-master](https://github.com/florestefano1975/comfyui-portrait-master) | Portrait Master          |
+| easy LLLiteLoader              | [ControlNet-LLLite-ComfyUI](https://github.com/kohya-ss/ControlNet-LLLite-ComfyUI) | LLLiteLoader             |
+| easy globalSeed                | [ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) | Global Seed (Inspire)    | 
+| easy preSamplingDynamicCFG     | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull  | 
+| dynamicThresholdingFull        | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull  | 
+| easy imageInsetCrop            | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | ImageInsetCrop           | 
+| easy poseEditor                | [ComfyUI_Custom_Nodes_AlekPet](https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet) | poseNode                 | 
 | easy preSamplingLayerDiffusion | [ComfyUI-layerdiffusion](https://github.com/huchenlei/ComfyUI-layerdiffusion) | LayeredDiffusionApply... | 
-| easy dynamiCrafterLoader | [ComfyUI-layerdiffusion](https://github.com/ExponentialML/ComfyUI_Native_DynamiCrafter) | Apply Dynamicrafter    | 
+| easy dynamiCrafterLoader       | [ComfyUI-layerdiffusion](https://github.com/ExponentialML/ComfyUI_Native_DynamiCrafter) | Apply Dynamicrafter    | 
+| easy imageChooser              | [cg-image-picker](https://github.com/chrisgoringe/cg-image-picker) | Preview Chooser         | 
+| easy styleAlignedBatchAlign              | [style_aligned_comfy](https://github.com/chrisgoringe/cg-image-picker) | styleAlignedBatchAlign       | 
 
 ## Workflow Examples
 
@@ -312,4 +364,14 @@ Disclaimer: Opened source was not easy. I have a lot of respect for the contribu
 
 [ComfyUI-Impact-Pack](https://github.com/ltdrdata/ComfyUI-Impact-Pack) - General modpack 1
 
-[ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) - General Modpack 2
+[ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) - General Modpack 2
+
+[ComfyUI-ResAdapter](https://github.com/jiaxiangc/ComfyUI-ResAdapter) - Make model generation independent of training resolution
+
+[ComfyUI_IPAdapter_plus](https://github.com/cubiq/ComfyUI_IPAdapter_plus) - Style migration
+
+[ComfyUI_InstantID](https://github.com/cubiq/ComfyUI_InstantID) - Face migration
+
+[ComfyUI-Custom-Scripts](https://github.com/pythongosssss/ComfyUI-Custom-Scripts) - pyssss🐍
+
+[cg-image-picker](https://github.com/chrisgoringe/cg-image-picker) - Image Preview Chooser

README.md

@@ -31,11 +31,54 @@
 - 扩展 XYplot 的可用性
 - 整合了Fooocus Inpaint功能
 - 整合了常用的逻辑计算、转换类型、展示所有类型等
+- 支持节点上checkpoint、lora模型子目录分类及预览图 (请在设置中开启上下文菜单嵌套子目录)
 - 支持BriaAI的RMBG-1.4模型的背景去除节点，[技术参考](https://huggingface.co/briaai/RMBG-1.4)
+- 支持 强制清理comfyUI模型显存占用
+- 支持Stable Diffusion 3 多账号API节点
 
 ## 更新日志
 
-**v1.1.2 (2024/3/25)**
+**v1.1.6**
+
+- 增加步调齐整适配 - 在所有的预采样和全采样器节点中的 调度器(schedulder) 增加了 **alignYourSteps** 选项
+- `easy kSampler` 和 `easy fullkSampler` 的 **image_output** 增加 **Preview&Choose**选项
+- 增加 `easy styleAlignedBatchAlign` - 风格对齐 [style_aligned_comfy](https://github.com/brianfitzgerald/style_aligned_comfy)
+- 增加 `easy ckptNames`
+- 增加 `easy controlnetNames`
+- 增加 `easy imagesSplitimage` - 批次图像拆分单张
+- 增加 `easy imageCount` - 图像数量
+- 增加 `easy textSwitch` - 文字切换
+
+**v1.1.5**
+
+- 重写 `easy cleanGPUUsed` - 可强制清理comfyUI的模型显存占用
+- 增加 `easy humanSegmentation` - 多类分割、人像分割
+- 增加 `easy imageColorMatch`
+- 增加 `easy ipadapterApplyRegional`
+- 增加 `easy ipadapterApplyFromParams`
+- 增加 `easy imageInterrogator` - 图像反推
+- 增加 `easy stableDiffusion3API` - 简易的Stable Diffusion 3 多账号API节点
+
+**v1.1.4**
+
+- 增加 `easy imageChooser` - 从[cg-image-picker](https://github.com/chrisgoringe/cg-image-picker)简化的图片选择器
+- 增加 `easy preSamplingCustom` - 自定义预采样，可支持cosXL-edit
+- 增加 `easy ipadapterStyleComposition`
+- 增加 在Loaders上右键菜单可查看 checkpoints、lora 信息
+- 修复 `easy preSamplingNoiseIn`、`easy latentNoisy`、`east Unsampler` 以兼容ComfyUI Revision>=2098 [0542088e] 以上版本
+- 修复 FooocusInpaint修改ModelPatcher计算权重引发的问题，理应在生成model后重置ModelPatcher为默认值
+
+**v1.1.3**
+
+- `easy ipadapterApply` 增加 **COMPOSITION** 预置项
+- 增加 对[ResAdapter](https://huggingface.co/jiaxiangc/res-adapter) lora模型 的加载支持
+- 增加 `easy promptLine`
+- 增加 `easy promptReplace`
+- 增加 `easy promptConcat`
+- `easy wildcards` 增加 **multiline_mode**属性 
+- 增加 当节点需要下载模型时，若huggingface连接超时，会切换至镜像地址下载模型
+
+**v1.1.2**
 
 - 改写 EasyUse 相关节点的部分插槽推荐节点
 - 增加 **启用上下文菜单自动嵌套子目录** 设置项，默认为启用状态，可分类子目录及checkpoints、loras预览图
@@ -46,15 +89,12 @@
 - 增加 `easy ipadapterApplyEncoder`
 - 增加 `easy ipadapterApplyEmbeds`
 - 增加 `easy preMaskDetailerFix`
+- `easy kSamplerInpainting` 增加 **additional** 属性，可设置成 Differential Diffusion 或 Only InpaintModelConditioning
 - 修复 `easy stylesSelector` 当未选择样式时，原有提示词发生了变化
-
-(4c25580)
-
-- `easy kSamplerInpainting` 增加 *additional* 属性，可设置成 Differential Diffusion 或 Only InpaintModelConditioning
 - 修复 `easy pipeEdit` 提示词输入lora时报错
 - 修复 layerDiffuse xyplot相关bug
 
-**v1.1.1 (5c8af8f)**
+**v1.1.1**
 
 - 修复首次添加含seed的节点且当前模式为control_before_generate时，seed为0的问题
 - `easy preSamplingAdvanced` 增加 **return_with_leftover_noise**
@@ -65,7 +105,7 @@
 - 去除强制**control_before_generate**设定
 - 增加 `easy imageRemBg` - 默认为BriaAI的RMBG-1.4模型, 移除背景效果更加，速度更快
 
-**v1.1.0 (d5ff84e)**
+**v1.1.0**
 
 - 增加 `easy imageSplitList` - 拆分每 N 张图像
 - 增加 `easy preSamplingDiffusionADDTL` - 可配置前景、背景、blended的additional_prompt等   
@@ -79,15 +119,18 @@
 - 修复 `easy instantIDApply` mask 未传入正确值
 - 修复 在 非a1111提示词风格下 BREAK 不生效的问题
 
-**v1.0.9 (ff1add1)**
+<details>
+<summary><b>v1.0.9</b></summary>
 
 - 修复未安装 ComfyUI-Impack-Pack 和 ComfyUI_InstantID 时报错
 - 修复 `easy pipeIn` - pipe设为可不必选
 - 增加 `easy instantIDApply` - 需要先安装 [ComfyUI_InstantID](https://github.com/cubiq/ComfyUI_InstantID), 工作流参考[示例](https://github.com/yolain/ComfyUI-Easy-Use?tab=readme-ov-file#InstantID)
 - 修复 `easy detailerFix` 未添加到保存图片格式化扩展名可用节点列表
 - 修复 `easy XYInputs: PromptSR` 在替换负面提示词时报错
+</details>
 
-**v1.0.8 (f28cbf7)**
+<details>
+<summary><b>v1.0.8</b></summary>
 
 - `easy cascadeLoader` stage_c 与 stage_b 支持checkpoint模型 (需要下载[checkpoints](https://huggingface.co/stabilityai/stable-cascade/tree/main/comfyui_checkpoints)) 
 - `easy styleSelector` 搜索框修改为不区分大小写匹配
@@ -101,13 +144,16 @@
 
 （翻译对照已由 [AIGODLIKE-COMFYUI-TRANSLATION](https://github.com/AIGODLIKE/AIGODLIKE-ComfyUI-Translation) 统一维护啦！
 首次下载或者版本较早的朋友请更新 AIGODLIKE-COMFYUI-TRANSLATION 和本节点包至最新版本。）
+</details>
 
-**v1.0.7**
+<details>
+<summary><b>v1.0.7</b></summary>
 
 - 增加 `easy cascadeLoader` - stable cascade 加载器
 - 增加 `easy preSamplingCascade` - stabled cascade stage_c 预采样参数
 - 增加 `easy fullCascadeKSampler` - stable cascade stage_c 完整版采样器
 - 增加 `easy cascadeKSampler` - stable cascade stage-c ksampler simple
+</details>
 
 <details>
 <summary><b>v1.0.6</b></summary>
@@ -266,21 +312,23 @@
 
 声明: 非常尊重这些原作者们的付出，开源不易，我仅仅只是做了一些整合与优化。
 
-| 节点名 (搜索名)                      | 相关的库                                                                        | 库相关的节点                 |
-|:-------------------------------|:----------------------------------------------------------------------------|:-----------------------|
-| easy setNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.SetNode        |
-| easy getNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.GetNode        |
-| easy bookmark                  | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | Bookmark 🔖            |
-| easy portraitMarker            | [comfyui-portrait-master](https://github.com/florestefano1975/comfyui-portrait-master) | Portrait Master        |
-| easy LLLiteLoader              | [ControlNet-LLLite-ComfyUI](https://github.com/kohya-ss/ControlNet-LLLite-ComfyUI) | LLLiteLoader           |
-| easy globalSeed                | [ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) | Global Seed (Inspire)  | 
+| 节点名 (搜索名)                      | 相关的库                                                                        | 库相关的节点                  |
+|:-------------------------------|:----------------------------------------------------------------------------|:------------------------|
+| easy setNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.SetNode         |
+| easy getNode                   | [ComfyUI-extensions](https://github.com/diffus3/ComfyUI-extensions) | diffus3.GetNode         |
+| easy bookmark                  | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | Bookmark 🔖             |
+| easy portraitMarker            | [comfyui-portrait-master](https://github.com/florestefano1975/comfyui-portrait-master) | Portrait Master         |
+| easy LLLiteLoader              | [ControlNet-LLLite-ComfyUI](https://github.com/kohya-ss/ControlNet-LLLite-ComfyUI) | LLLiteLoader            |
+| easy globalSeed                | [ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) | Global Seed (Inspire)   | 
 | easy preSamplingDynamicCFG     | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull | 
 | dynamicThresholdingFull        | [sd-dynamic-thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding) | DynamicThresholdingFull | 
-| easy imageInsetCrop            | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | ImageInsetCrop         | 
-| easy poseEditor                | [ComfyUI_Custom_Nodes_AlekPet](https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet) | poseNode               | 
-| easy if                        | [ComfyUI-Logic](https://github.com/theUpsider/ComfyUI-Logic) | IfExecute              | 
-| easy preSamplingLayerDiffusion | [ComfyUI-layerdiffusion](https://github.com/huchenlei/ComfyUI-layerdiffusion) | LayeredDiffusionApply等 | 
-| easy dynamiCrafterLoader | [ComfyUI-layerdiffusion](https://github.com/ExponentialML/ComfyUI_Native_DynamiCrafter) | Apply Dynamicrafter    | 
+| easy imageInsetCrop            | [rgthree-comfy](https://github.com/rgthree/rgthree-comfy) | ImageInsetCrop          | 
+| easy poseEditor                | [ComfyUI_Custom_Nodes_AlekPet](https://github.com/AlekPet/ComfyUI_Custom_Nodes_AlekPet) | poseNode                | 
+| easy if                        | [ComfyUI-Logic](https://github.com/theUpsider/ComfyUI-Logic) | IfExecute               | 
+| easy preSamplingLayerDiffusion | [ComfyUI-layerdiffusion](https://github.com/huchenlei/ComfyUI-layerdiffusion) | LayeredDiffusionApply等  | 
+| easy dynamiCrafterLoader       | [ComfyUI-layerdiffusion](https://github.com/ExponentialML/ComfyUI_Native_DynamiCrafter) | Apply Dynamicrafter     | 
+| easy imageChooser              | [cg-image-picker](https://github.com/chrisgoringe/cg-image-picker) | Preview Chooser         | 
+| easy styleAlignedBatchAlign              | [style_aligned_comfy](https://github.com/chrisgoringe/cg-image-picker) | styleAlignedBatchAlign       | 
 
 ## 示例
 
@@ -329,3 +377,13 @@
 [ComfyUI-Inspire-Pack](https://github.com/ltdrdata/ComfyUI-Inspire-Pack) - 常规整合包2
 
 [ComfyUI-Logic](https://github.com/theUpsider/ComfyUI-Logic) -  ComfyUI逻辑运算
+
+[ComfyUI-ResAdapter](https://github.com/jiaxiangc/ComfyUI-ResAdapter) - 让模型生成不受训练分辨率限制
+
+[ComfyUI_IPAdapter_plus](https://github.com/cubiq/ComfyUI_IPAdapter_plus) - 风格迁移
+
+[ComfyUI_InstantID](https://github.com/cubiq/ComfyUI_InstantID) - 人脸迁移
+
+[ComfyUI-Custom-Scripts](https://github.com/pythongosssss/ComfyUI-Custom-Scripts) - pyssss 小蛇🐍脚本
+
+[cg-image-picker](https://github.com/chrisgoringe/cg-image-picker) - 图片选择器

__init__.py

@@ -1,3 +1,5 @@
+__version__ = "1.1.6"
+
 import os
 import glob
 import folder_paths
@@ -85,4 +87,4 @@ WEB_DIRECTORY = "./web"
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', "WEB_DIRECTORY"]
 
 
-print('\033[34mComfy-Easy-Use (v1.1.2): \033[92mLoaded\033[0m')
+print(f'\033[34mComfy-Easy-Use v{__version__}: \033[92mLoaded\033[0m')

py/adv_encode.py

@@ -308,9 +308,10 @@ def advanced_encode(clip, text, token_normalization, weight_interpretation, w_ma
 
             embeddings_final, pooled = prepareXL(embs_l, embs_g, pooled, clip_balance)
 
-            cond = [[embeddings_final,
-                             {"pooled_output": pooled, "width": width, "height": height, "crop_w": crop_w,
-                              "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]]
+            cond = [[embeddings_final, {"pooled_output": pooled}]]
+            # cond = [[embeddings_final,
+            #                  {"pooled_output": pooled, "width": width, "height": height, "crop_w": crop_w,
+            #                   "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]]
         else:
             embeddings_final, pooled = advanced_encode_from_tokens(tokenized['l'],
                                                                    token_normalization,

py/api.py

@@ -1,14 +1,15 @@
-import re
 import os
-import torch
+import hashlib
 import sys
 import json
+import shutil
 import folder_paths
+from folder_paths import get_directory_by_type
 from server import PromptServer
 from .config import RESOURCES_DIR, FOOOCUS_STYLES_DIR, FOOOCUS_STYLES_SAMPLES
-from .easyNodes import easyCache
 from .logic import ConvertAnything
 from .libs.model import easyModelManager
+from .libs.utils import getMetadata, cleanGPUUsedForce, get_local_filepath
 
 try:
     import aiohttp
@@ -18,6 +19,24 @@ except ImportError:
     print("pip install aiohttp")
     sys.exit()
 
+@PromptServer.instance.routes.post("/easyuse/cleangpu")
+def cleanGPU(request):
+    try:
+        cleanGPUUsedForce()
+        return web.Response(status=200)
+    except Exception as e:
+        return web.Response(status=500)
+        pass
+
+@PromptServer.instance.routes.get("/easyuse/reboot")
+def reboot(request):
+    try:
+        sys.stdout.close_log()
+    except Exception as e:
+        pass
+
+    return os.execv(sys.executable, [sys.executable] + sys.argv)
+
 # parse csv
 @PromptServer.instance.routes.post("/easyuse/upload/csv")
 async def parse_csv(request):
@@ -119,6 +138,8 @@ async def getModelsThumbnail(request):
     loras = folder_paths.get_filename_list("loras_thumb")
     checkpoints_full = []
     loras_full = []
+    if len(checkpoints) + len(loras) >= 500:
+        return web.Response(status=400)
     for index, i in enumerate(checkpoints):
         full_path = folder_paths.get_full_path('checkpoints_thumb', str(i))
         if full_path:
@@ -129,5 +150,137 @@ async def getModelsThumbnail(request):
             loras_full.append(full_path)
     return web.json_response(checkpoints_full + loras_full)
 
+@PromptServer.instance.routes.post("/easyuse/metadata/notes/{name}")
+async def save_notes(request):
+    name = request.match_info["name"]
+    pos = name.index("/")
+    type = name[0:pos]
+    name = name[pos+1:]
+
+    file_path = None
+    if type == "embeddings" or type == "loras":
+        name = name.lower()
+        files = folder_paths.get_filename_list(type)
+        for f in files:
+            lower_f = f.lower()
+            if lower_f == name:
+                file_path = folder_paths.get_full_path(type, f)
+            else:
+                n = os.path.splitext(f)[0].lower()
+                if n == name:
+                    file_path = folder_paths.get_full_path(type, f)
+
+            if file_path is not None:
+                break
+    else:
+        file_path = folder_paths.get_full_path(
+            type, name)
+    if not file_path:
+        return web.Response(status=404)
+
+    file_no_ext = os.path.splitext(file_path)[0]
+    info_file = file_no_ext + ".txt"
+    with open(info_file, "w") as f:
+        f.write(await request.text())
+
+    return web.Response(status=200)
+
+@PromptServer.instance.routes.get("/easyuse/metadata/{name}")
+async def load_metadata(request):
+    name = request.match_info["name"]
+    pos = name.index("/")
+    type = name[0:pos]
+    name = name[pos+1:]
+
+    file_path = None
+    if type == "embeddings":
+        name = name.lower()
+        files = folder_paths.get_filename_list(type)
+        for f in files:
+            lower_f = f.lower()
+            if lower_f == name:
+                file_path = folder_paths.get_full_path(type, f)
+            else:
+                n = os.path.splitext(f)[0].lower()
+                if n == name:
+                    file_path = folder_paths.get_full_path(type, f)
+
+            if file_path is not None:
+                break
+    else:
+        file_path = folder_paths.get_full_path(type, name)
+    if not file_path:
+        return web.Response(status=404)
+
+    try:
+        header = getMetadata(file_path)
+        header_json = json.loads(header)
+        meta = header_json["__metadata__"] if "__metadata__" in header_json else None
+    except:
+        meta = None
+
+    if meta is None:
+        meta = {}
+
+    file_no_ext = os.path.splitext(file_path)[0]
+
+    info_file = file_no_ext + ".txt"
+    if os.path.isfile(info_file):
+        with open(info_file, "r") as f:
+            meta["easyuse.notes"] = f.read()
+
+    hash_file = file_no_ext + ".sha256"
+    if os.path.isfile(hash_file):
+        with open(hash_file, "rt") as f:
+            meta["easyuse.sha256"] = f.read()
+    else:
+        with open(file_path, "rb") as f:
+            meta["easyuse.sha256"] = hashlib.sha256(f.read()).hexdigest()
+        with open(hash_file, "wt") as f:
+            f.write(meta["easyuse.sha256"])
+
+    return web.json_response(meta)
+
+@PromptServer.instance.routes.post("/easyuse/save/{name}")
+async def save_preview(request):
+    name = request.match_info["name"]
+    pos = name.index("/")
+    type = name[0:pos]
+    name = name[pos+1:]
+
+    body = await request.json()
+
+    dir = get_directory_by_type(body.get("type", "output"))
+    subfolder = body.get("subfolder", "")
+    full_output_folder = os.path.join(dir, os.path.normpath(subfolder))
+
+    if os.path.commonpath((dir, os.path.abspath(full_output_folder))) != dir:
+        return web.Response(status=400)
+
+    filepath = os.path.join(full_output_folder, body.get("filename", ""))
+    image_path = folder_paths.get_full_path(type, name)
+    image_path = os.path.splitext(
+        image_path)[0] + os.path.splitext(filepath)[1]
+
+    shutil.copyfile(filepath, image_path)
+
+    return web.json_response({
+        "image":  type + "/" + os.path.basename(image_path)
+    })
+
+@PromptServer.instance.routes.post("/easyuse/model/download")
+async def download_model(request):
+    post = await request.post()
+    url = post.get("url")
+    local_dir = post.get("local_dir")
+    if local_dir not in ['checkpoints', 'loras', 'controlnet', 'onnx', 'instantid', 'ipadapter', 'dynamicrafter_models', 'mediapipe', 'rembg', 'layer_model']:
+        return web.Response(status=400)
+    local_path = os.path.join(folder_paths.models_dir, local_dir)
+    try:
+        get_local_filepath(url, local_path)
+        return web.Response(status=200)
+    except:
+        return web.Response(status=500)
+
 NODE_CLASS_MAPPINGS = {}
 NODE_DISPLAY_NAME_MAPPINGS = {}

py/config.py

@@ -68,7 +68,7 @@ LAYER_DIFFUSION_VAE = {
         }
     },
     "decode": {
-        "sd15": {
+        "sd1": {
             "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_vae_transparent_decoder.safetensors"
         },
         "sdxl": {
@@ -78,7 +78,7 @@ LAYER_DIFFUSION_VAE = {
 }
 LAYER_DIFFUSION = {
     "Attention Injection": {
-        "sd15": {
+        "sd1": {
           "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_transparent_attn.safetensors"
         },
         "sdxl": {
@@ -89,7 +89,7 @@ LAYER_DIFFUSION = {
         "sdxl": {
             "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_xl_transparent_conv.safetensors"
         },
-        "sd15": {
+        "sd1": {
             "model_url": None
         }
     },
@@ -102,7 +102,7 @@ LAYER_DIFFUSION = {
         }
     },
     "Foreground": {
-        "sd15": {
+        "sd1": {
           "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_fg2bg.safetensors"
         },
         "sdxl": {
@@ -110,7 +110,7 @@ LAYER_DIFFUSION = {
         }
     },
     "Foreground to Background": {
-        "sd15": {
+        "sd1": {
           "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_fg2bg.safetensors"
         },
         "sdxl": {
@@ -118,7 +118,7 @@ LAYER_DIFFUSION = {
         }
     },
     "Background": {
-        "sd15": {
+        "sd1": {
           "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_bg2fg.safetensors"
         },
         "sdxl": {
@@ -126,7 +126,7 @@ LAYER_DIFFUSION = {
         }
     },
     "Background to Foreground": {
-        "sd15": {
+        "sd1": {
             "model_url": "https://huggingface.co/LayerDiffusion/layerdiffusion-v1/resolve/main/layer_sd15_bg2fg.safetensors"
         },
         "sdxl": {
@@ -159,7 +159,7 @@ IPADAPTER_MODELS = {
             "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15.safetensors"
         },
         "sdxl": {
-            "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter_sdxl.safetensors"
+            "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter_sdxl_vit-h.safetensors"
         }
     },
     "VIT-G (medium strength)": {
@@ -167,7 +167,7 @@ IPADAPTER_MODELS = {
             "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15_vit-G.safetensors"
         },
         "sdxl": {
-            "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter_sdxl_vit-h.safetensors"
+            "model_url": "https://huggingface.co/h94/IP-Adapter/resolve/main/sdxl_models/ip-adapter_sdxl.safetensors"
         }
     },
     "PLUS (high strength)": {
@@ -231,6 +231,14 @@ IPADAPTER_MODELS = {
         "sdxl": {
             "model_url": "https://huggingface.co/h94/IP-Adapter-FaceID/resolve/main/ip-adapter-faceid-portrait_sdxl.bin",
         }
+    },
+    "COMPOSITION": {
+        "sd15": {
+            "model_url": "https://huggingface.co/ostris/ip-composition-adapter/resolve/main/ip_plus_composition_sd15.safetensors"
+        },
+        "sdxl": {
+            "model_url": "https://huggingface.co/ostris/ip-composition-adapter/resolve/main/ip_plus_composition_sdxl.safetensors"
+        }
     }
 }
 
@@ -252,4 +260,19 @@ DYNAMICRAFTER_MODELS = {
     "dynamicrafter_unet_256 (2.98GB)": {
         "model_url": "https://huggingface.co/ExponentialML/DynamiCrafterUNet/resolve/main/dynamicrafter_unet_256.safetensors"
     },
+}
+
+#humanParsing
+HUMANPARSING_MODELS = {
+    "parsing_lip": {
+        "model_url": "https://huggingface.co/levihsu/OOTDiffusion/resolve/main/checkpoints/humanparsing/parsing_lip.onnx",
+    },
+}
+
+#mediapipe
+MEDIAPIPE_DIR = os.path.join(folder_paths.models_dir, "mediapipe")
+MEDIAPIPE_MODELS = {
+    "selfie_multiclass_256x256": {
+        "model_url": "https://huggingface.co/yolain/selfie_multiclass_256x256/resolve/main/selfie_multiclass_256x256.tflite"
+    }
 }

py/human_parsing/parsing_api.py

@@ -0,0 +1,156 @@
+import torch
+import numpy as np
+import cv2
+import torchvision.transforms as transforms
+from torch.utils.data import DataLoader
+from .simple_extractor_dataset import SimpleFolderDataset
+from .transforms import transform_logits
+from tqdm import tqdm
+from PIL import Image
+
+def get_palette(num_cls):
+    """ Returns the color map for visualizing the segmentation mask.
+    Args:
+        num_cls: Number of classes
+    Returns:
+        The color map
+    """
+    n = num_cls
+    palette = [0] * (n * 3)
+    for j in range(0, n):
+        lab = j
+        palette[j * 3 + 0] = 0
+        palette[j * 3 + 1] = 0
+        palette[j * 3 + 2] = 0
+        i = 0
+        while lab:
+            palette[j * 3 + 0] |= (((lab >> 0) & 1) << (7 - i))
+            palette[j * 3 + 1] |= (((lab >> 1) & 1) << (7 - i))
+            palette[j * 3 + 2] |= (((lab >> 2) & 1) << (7 - i))
+            i += 1
+            lab >>= 3
+    return palette
+
+
+def delete_irregular(logits_result):
+    parsing_result = np.argmax(logits_result, axis=2)
+    upper_cloth = np.where(parsing_result == 4, 255, 0)
+    contours, hierarchy = cv2.findContours(upper_cloth.astype(np.uint8),
+                                           cv2.RETR_CCOMP, cv2.CHAIN_APPROX_TC89_L1)
+    area = []
+    for i in range(len(contours)):
+        a = cv2.contourArea(contours[i], True)
+        area.append(abs(a))
+    if len(area) != 0:
+        top = area.index(max(area))
+        M = cv2.moments(contours[top])
+        cY = int(M["m01"] / M["m00"])
+
+    dresses = np.where(parsing_result == 7, 255, 0)
+    contours_dress, hierarchy_dress = cv2.findContours(dresses.astype(np.uint8),
+                                                       cv2.RETR_CCOMP, cv2.CHAIN_APPROX_TC89_L1)
+    area_dress = []
+    for j in range(len(contours_dress)):
+        a_d = cv2.contourArea(contours_dress[j], True)
+        area_dress.append(abs(a_d))
+    if len(area_dress) != 0:
+        top_dress = area_dress.index(max(area_dress))
+        M_dress = cv2.moments(contours_dress[top_dress])
+        cY_dress = int(M_dress["m01"] / M_dress["m00"])
+    wear_type = "dresses"
+    if len(area) != 0:
+        if len(area_dress) != 0 and cY_dress > cY:
+            irregular_list = np.array([4, 5, 6])
+            logits_result[:, :, irregular_list] = -1
+        else:
+            irregular_list = np.array([5, 6, 7, 8, 9, 10, 12, 13])
+            logits_result[:cY, :, irregular_list] = -1
+            wear_type = "cloth_pant"
+        parsing_result = np.argmax(logits_result, axis=2)
+    # pad border
+    parsing_result = np.pad(parsing_result, pad_width=1, mode='constant', constant_values=0)
+    return parsing_result, wear_type
+
+
+
+def hole_fill(img):
+    img_copy = img.copy()
+    mask = np.zeros((img.shape[0] + 2, img.shape[1] + 2), dtype=np.uint8)
+    cv2.floodFill(img, mask, (0, 0), 255)
+    img_inverse = cv2.bitwise_not(img)
+    dst = cv2.bitwise_or(img_copy, img_inverse)
+    return dst
+
+def refine_mask(mask):
+    contours, hierarchy = cv2.findContours(mask.astype(np.uint8),
+                                           cv2.RETR_CCOMP, cv2.CHAIN_APPROX_TC89_L1)
+    area = []
+    for j in range(len(contours)):
+        a_d = cv2.contourArea(contours[j], True)
+        area.append(abs(a_d))
+    refine_mask = np.zeros_like(mask).astype(np.uint8)
+    if len(area) != 0:
+        i = area.index(max(area))
+        cv2.drawContours(refine_mask, contours, i, color=255, thickness=-1)
+        # keep large area in skin case
+        for j in range(len(area)):
+          if j != i and area[i] > 2000:
+             cv2.drawContours(refine_mask, contours, j, color=255, thickness=-1)
+    return refine_mask
+
+def refine_hole(parsing_result_filled, parsing_result, arm_mask):
+    filled_hole = cv2.bitwise_and(np.where(parsing_result_filled == 4, 255, 0),
+                                  np.where(parsing_result != 4, 255, 0)) - arm_mask * 255
+    contours, hierarchy = cv2.findContours(filled_hole, cv2.RETR_CCOMP, cv2.CHAIN_APPROX_TC89_L1)
+    refine_hole_mask = np.zeros_like(parsing_result).astype(np.uint8)
+    for i in range(len(contours)):
+        a = cv2.contourArea(contours[i], True)
+        # keep hole > 2000 pixels
+        if abs(a) > 2000:
+            cv2.drawContours(refine_hole_mask, contours, i, color=255, thickness=-1)
+    return refine_hole_mask + arm_mask
+
+def onnx_inference(lip_session, input_dir, mask_components=[0]):
+
+    transform = transforms.Compose([
+        transforms.ToTensor(),
+        transforms.Normalize(mean=[0.406, 0.456, 0.485], std=[0.225, 0.224, 0.229])
+    ])
+    input_size = [473, 473]
+
+    dataset_lip = SimpleFolderDataset(root=input_dir, input_size=input_size, transform=transform)
+    dataloader_lip = DataLoader(dataset_lip)
+    palette = get_palette(20)
+    with torch.no_grad():
+        for _, batch in enumerate(tqdm(dataloader_lip)):
+            image, meta = batch
+            c = meta['center'].numpy()[0]
+            s = meta['scale'].numpy()[0]
+            w = meta['width'].numpy()[0]
+            h = meta['height'].numpy()[0]
+
+            output = lip_session.run(None, {"input.1": image.numpy().astype(np.float32)})
+            upsample = torch.nn.Upsample(size=input_size, mode='bilinear', align_corners=True)
+            upsample_output = upsample(torch.from_numpy(output[1][0]).unsqueeze(0))
+            upsample_output = upsample_output.squeeze()
+            upsample_output = upsample_output.permute(1, 2, 0)  # CHW -> HWC
+            logits_result_lip = transform_logits(upsample_output.data.cpu().numpy(), c, s, w, h,
+                                                 input_size=input_size)
+            parsing_result = np.argmax(logits_result_lip, axis=2)
+
+        output_img = Image.fromarray(np.asarray(parsing_result, dtype=np.uint8))
+        output_img.putpalette(palette)
+
+    mask = np.isin(output_img, mask_components).astype(np.uint8)
+    mask_image = Image.fromarray(mask * 255)
+    mask_image = mask_image.convert("RGB")
+    mask_image = torch.from_numpy(np.array(mask_image).astype(np.float32) / 255.0).unsqueeze(0)
+
+    output_img = output_img.convert('RGB')
+    output_img = torch.from_numpy(np.array(output_img).astype(np.float32) / 255.0).unsqueeze(0)
+
+    return output_img, mask_image
+
+
+
+

py/human_parsing/run_parsing.py

@@ -0,0 +1,23 @@
+from .parsing_api import onnx_inference
+from ..libs.utils import install_package
+
+class HumanParsing:
+    def __init__(self, model_path):
+        self.model_path = model_path
+        self.session = None
+
+    def __call__(self, input_image, mask_components):
+        if self.session is None:
+            install_package('onnxruntime')
+            import onnxruntime as ort
+
+            session_options = ort.SessionOptions()
+            session_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
+            session_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL
+            # session_options.add_session_config_entry('gpu_id', str(gpu_id))
+            self.session = ort.InferenceSession(self.model_path, sess_options=session_options,
+                                                providers=['CPUExecutionProvider'])
+
+        parsed_image, mask = onnx_inference(self.session, input_image, mask_components)
+        return parsed_image, mask
+

py/human_parsing/simple_extractor_dataset.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+"""
+@Author  :   Peike Li
+@Contact :   peike.li@yahoo.com
+@File    :   dataset.py
+@Time    :   8/30/19 9:12 PM
+@Desc    :   Dataset Definition
+@License :   This source code is licensed under the license found in the
+             LICENSE file in the root directory of this source tree.
+"""
+
+import os
+
+import cv2
+import numpy as np
+from PIL import Image
+from torch.utils import data
+from .transforms import get_affine_transform
+
+
+class SimpleFolderDataset(data.Dataset):
+    def __init__(self, root, input_size=[512, 512], transform=None):
+        self.root = root
+        self.input_size = input_size
+        self.transform = transform
+        self.aspect_ratio = input_size[1] * 1.0 / input_size[0]
+        self.input_size = np.asarray(input_size)
+        self.is_pil_image = False
+        if isinstance(root, Image.Image):
+            self.file_list = [root]
+            self.is_pil_image = True
+        elif os.path.isfile(root):
+            self.file_list = [os.path.basename(root)]
+            self.root = os.path.dirname(root)
+        else:
+            self.file_list = os.listdir(self.root)
+
+    def __len__(self):
+        return len(self.file_list)
+
+    def _box2cs(self, box):
+        x, y, w, h = box[:4]
+        return self._xywh2cs(x, y, w, h)
+
+    def _xywh2cs(self, x, y, w, h):
+        center = np.zeros((2), dtype=np.float32)
+        center[0] = x + w * 0.5
+        center[1] = y + h * 0.5
+        if w > self.aspect_ratio * h:
+            h = w * 1.0 / self.aspect_ratio
+        elif w < self.aspect_ratio * h:
+            w = h * self.aspect_ratio
+        scale = np.array([w, h], dtype=np.float32)
+        return center, scale
+
+    def __getitem__(self, index):
+        if self.is_pil_image:
+            img = np.asarray(self.file_list[index])[:, :, [2, 1, 0]]
+        else:
+            img_name = self.file_list[index]
+            img_path = os.path.join(self.root, img_name)
+            img = cv2.imread(img_path, cv2.IMREAD_COLOR)
+        h, w, _ = img.shape
+
+        # Get person center and scale
+        person_center, s = self._box2cs([0, 0, w - 1, h - 1])
+        r = 0
+        trans = get_affine_transform(person_center, s, r, self.input_size)
+        input = cv2.warpAffine(
+            img,
+            trans,
+            (int(self.input_size[1]), int(self.input_size[0])),
+            flags=cv2.INTER_LINEAR,
+            borderMode=cv2.BORDER_CONSTANT,
+            borderValue=(0, 0, 0))
+
+        input = self.transform(input)
+        meta = {
+            'center': person_center,
+            'height': h,
+            'width': w,
+            'scale': s,
+            'rotation': r
+        }
+
+        return input, meta

py/human_parsing/transforms.py

@@ -0,0 +1,167 @@
+# ------------------------------------------------------------------------------
+# Copyright (c) Microsoft
+# Licensed under the MIT License.
+# Written by Bin Xiao (Bin.Xiao@microsoft.com)
+# ------------------------------------------------------------------------------
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+import numpy as np
+import cv2
+import torch
+
+class BRG2Tensor_transform(object):
+    def __call__(self, pic):
+        img = torch.from_numpy(pic.transpose((2, 0, 1)))
+        if isinstance(img, torch.ByteTensor):
+            return img.float()
+        else:
+            return img
+
+class BGR2RGB_transform(object):
+    def __call__(self, tensor):
+        return tensor[[2,1,0],:,:]
+
+def flip_back(output_flipped, matched_parts):
+    '''
+    ouput_flipped: numpy.ndarray(batch_size, num_joints, height, width)
+    '''
+    assert output_flipped.ndim == 4,\
+        'output_flipped should be [batch_size, num_joints, height, width]'
+
+    output_flipped = output_flipped[:, :, :, ::-1]
+
+    for pair in matched_parts:
+        tmp = output_flipped[:, pair[0], :, :].copy()
+        output_flipped[:, pair[0], :, :] = output_flipped[:, pair[1], :, :]
+        output_flipped[:, pair[1], :, :] = tmp
+
+    return output_flipped
+
+
+def fliplr_joints(joints, joints_vis, width, matched_parts):
+    """
+    flip coords
+    """
+    # Flip horizontal
+    joints[:, 0] = width - joints[:, 0] - 1
+
+    # Change left-right parts
+    for pair in matched_parts:
+        joints[pair[0], :], joints[pair[1], :] = \
+            joints[pair[1], :], joints[pair[0], :].copy()
+        joints_vis[pair[0], :], joints_vis[pair[1], :] = \
+            joints_vis[pair[1], :], joints_vis[pair[0], :].copy()
+
+    return joints*joints_vis, joints_vis
+
+
+def transform_preds(coords, center, scale, input_size):
+    target_coords = np.zeros(coords.shape)
+    trans = get_affine_transform(center, scale, 0, input_size, inv=1)
+    for p in range(coords.shape[0]):
+        target_coords[p, 0:2] = affine_transform(coords[p, 0:2], trans)
+    return target_coords
+
+def transform_parsing(pred, center, scale, width, height, input_size):
+
+    trans = get_affine_transform(center, scale, 0, input_size, inv=1)
+    target_pred = cv2.warpAffine(
+            pred,
+            trans,
+            (int(width), int(height)), #(int(width), int(height)),
+            flags=cv2.INTER_NEAREST,
+            borderMode=cv2.BORDER_CONSTANT,
+            borderValue=(0))
+
+    return target_pred
+
+def transform_logits(logits, center, scale, width, height, input_size):
+
+    trans = get_affine_transform(center, scale, 0, input_size, inv=1)
+    channel = logits.shape[2]
+    target_logits = []
+    for i in range(channel):
+        target_logit = cv2.warpAffine(
+            logits[:,:,i],
+            trans,
+            (int(width), int(height)), #(int(width), int(height)),
+            flags=cv2.INTER_LINEAR,
+            borderMode=cv2.BORDER_CONSTANT,
+            borderValue=(0))
+        target_logits.append(target_logit)
+    target_logits = np.stack(target_logits,axis=2)
+
+    return target_logits
+
+
+def get_affine_transform(center,
+                         scale,
+                         rot,
+                         output_size,
+                         shift=np.array([0, 0], dtype=np.float32),
+                         inv=0):
+    if not isinstance(scale, np.ndarray) and not isinstance(scale, list):
+        print(scale)
+        scale = np.array([scale, scale])
+
+    scale_tmp = scale
+
+    src_w = scale_tmp[0]
+    dst_w = output_size[1]
+    dst_h = output_size[0]
+
+    rot_rad = np.pi * rot / 180
+    src_dir = get_dir([0, src_w * -0.5], rot_rad)
+    dst_dir = np.array([0, (dst_w-1) * -0.5], np.float32)
+
+    src = np.zeros((3, 2), dtype=np.float32)
+    dst = np.zeros((3, 2), dtype=np.float32)
+    src[0, :] = center + scale_tmp * shift
+    src[1, :] = center + src_dir + scale_tmp * shift
+    dst[0, :] = [(dst_w-1) * 0.5, (dst_h-1) * 0.5]
+    dst[1, :] = np.array([(dst_w-1) * 0.5, (dst_h-1) * 0.5]) + dst_dir
+
+    src[2:, :] = get_3rd_point(src[0, :], src[1, :])
+    dst[2:, :] = get_3rd_point(dst[0, :], dst[1, :])
+
+    if inv:
+        trans = cv2.getAffineTransform(np.float32(dst), np.float32(src))
+    else:
+        trans = cv2.getAffineTransform(np.float32(src), np.float32(dst))
+
+    return trans
+
+
+def affine_transform(pt, t):
+    new_pt = np.array([pt[0], pt[1], 1.]).T
+    new_pt = np.dot(t, new_pt)
+    return new_pt[:2]
+
+
+def get_3rd_point(a, b):
+    direct = a - b
+    return b + np.array([-direct[1], direct[0]], dtype=np.float32)
+
+
+def get_dir(src_point, rot_rad):
+    sn, cs = np.sin(rot_rad), np.cos(rot_rad)
+
+    src_result = [0, 0]
+    src_result[0] = src_point[0] * cs - src_point[1] * sn
+    src_result[1] = src_point[0] * sn + src_point[1] * cs
+
+    return src_result
+
+
+def crop(img, center, scale, output_size, rot=0):
+    trans = get_affine_transform(center, scale, rot, output_size)
+
+    dst_img = cv2.warpAffine(img,
+                             trans,
+                             (int(output_size[1]), int(output_size[0])),
+                             flags=cv2.INTER_LINEAR)
+
+    return dst_img

py/image.py

@@ -4,9 +4,36 @@ import hashlib
 import folder_paths
 import torch
 import numpy as np
+import comfy.model_management
+from comfy_extras.nodes_compositing import JoinImageWithAlpha
+from server import PromptServer
 from nodes import MAX_RESOLUTION
+from torchvision.transforms.functional import to_pil_image
 from .log import log_node_info
-from .libs.image import pil2tensor, tensor2pil, ResizeMode, get_new_bounds
+from .libs.image import pil2tensor, tensor2pil, ResizeMode, get_new_bounds, RGB2RGBA, image2mask, blendImage
+from .libs.colorfix import adain_color_fix, wavelet_color_fix
+from .libs.chooser import ChooserMessage, ChooserCancelled
+from .config import REMBG_DIR, REMBG_MODELS, HUMANPARSING_MODELS, MEDIAPIPE_MODELS, MEDIAPIPE_DIR
+
+
+# 图像数量
+class imageCount:
+  @classmethod
+  def INPUT_TYPES(s):
+    return {
+      "required": {
+        "images": ("IMAGE",),
+      }
+    }
+
+  CATEGORY = "EasyUse/Image"
+
+  RETURN_TYPES = ("INT",)
+  RETURN_NAMES = ("count",)
+  FUNCTION = "get_count"
+
+  def get_count(self, images):
+    return (images.size(0),)
 
 # 图像裁切
 class imageInsetCrop:
@@ -512,10 +539,27 @@ class imageSplitList:
           new_images[1].append(img)
     return new_images
 
+class imagesSplitImage:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+          "required": {
+              "images": ("IMAGE",),
+          }
+        }
+
+    RETURN_TYPES = ("IMAGE", "IMAGE", "IMAGE", "IMAGE", "IMAGE")
+    RETURN_NAMES = ("image1", "image2", "image3", "image4", "image5")
+    FUNCTION = "split"
+    CATEGORY = "EasyUse/Image"
+
+    def split(self, images,):
+      new_images = torch.chunk(images, len(images), dim=0)
+      return new_images
+
 # 图片背景移除
 from .briaai.rembg import BriaRMBG, preprocess_image, postprocess_image
-from .libs.utils import get_local_filepath, easySave
-from .config import REMBG_DIR, REMBG_MODELS
+from .libs.utils import get_local_filepath, easySave, install_package
 class imageRemBg:
   @classmethod
   def INPUT_TYPES(self):
@@ -582,7 +626,382 @@ class imageRemBg:
     else:
       return (None, None)
 
-    # 姿势编辑器
+# 图像选择器
+class imageChooser(PreviewImage):
+  @classmethod
+  def INPUT_TYPES(self):
+    return {
+      "required":{
+
+      },
+      "optional": {
+        "images": ("IMAGE",),
+      },
+      "hidden": {"prompt": "PROMPT", "my_unique_id": "UNIQUE_ID"},
+    }
+
+  RETURN_TYPES = ("IMAGE",)
+  RETURN_NAMES = ("image",)
+  FUNCTION = "chooser"
+  OUTPUT_NODE = True
+  INPUT_IS_LIST = True
+  CATEGORY = "EasyUse/Image"
+
+  last_ic = {}
+  @classmethod
+  def IS_CHANGED(cls, my_unique_id, **kwargs):
+    return cls.last_ic[my_unique_id[0]]
+
+  def tensor_bundle(self, tensor_in: torch.Tensor, picks):
+    if tensor_in is not None and len(picks):
+      batch = tensor_in.shape[0]
+      return torch.cat(tuple([tensor_in[(x) % batch].unsqueeze_(0) for x in picks])).reshape(
+        [-1] + list(tensor_in.shape[1:]))
+    else:
+      return None
+
+  def chooser(self, prompt=None, my_unique_id=None, **kwargs):
+
+    id = my_unique_id[0]
+    if id not in ChooserMessage.stash:
+      ChooserMessage.stash[id] = {}
+    my_stash = ChooserMessage.stash[id]
+
+    # enable stashing. If images is None, we are operating in read-from-stash mode
+    if 'images' in kwargs:
+      my_stash['images'] = kwargs['images']
+    else:
+      kwargs['images'] = my_stash.get('images', None)
+
+    if (kwargs['images'] is None):
+      return (None, None, None, "")
+
+    images_in = torch.cat(kwargs.pop('images'))
+    self.batch = images_in.shape[0]
+    for x in kwargs: kwargs[x] = kwargs[x][0]
+    result = self.save_images(images=images_in, prompt=prompt)
+
+    images = result['ui']['images']
+    PromptServer.instance.send_sync("easyuse-image-choose", {"id": id, "urls": images})
+
+    # wait for selection
+    try:
+      selections = ChooserMessage.waitForMessage(id, asList=True)
+      choosen = [x for x in selections if x >= 0] if len(selections)>1 else [0]
+    except ChooserCancelled:
+      raise comfy.model_management.InterruptProcessingException()
+
+    return {"ui": {"images": images},
+              "result": (self.tensor_bundle(images_in, choosen),)}
+
+class imageColorMatch(PreviewImage):
+  @classmethod
+  def INPUT_TYPES(cls):
+    return {
+      "required": {
+        "image_ref": ("IMAGE",),
+        "image_target": ("IMAGE",),
+        "method": (['wavelet', 'adain', 'mkl', 'hm', 'reinhard', 'mvgd', 'hm-mvgd-hm', 'hm-mkl-hm'],),
+        "image_output": (["Hide", "Preview", "Save", "Hide/Save"], {"default": "Preview"}),
+        "save_prefix": ("STRING", {"default": "ComfyUI"}),
+      },
+      "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
+    }
+
+  CATEGORY = "EasyUse/Image"
+
+  RETURN_TYPES = ("IMAGE",)
+  RETURN_NAMES = ("image",)
+  OUTPUT_NODE = True
+  FUNCTION = "color_match"
+
+  def color_match(self, image_ref, image_target, method, image_output, save_prefix, prompt=None, extra_pnginfo=None):
+    if method in ["wavelet", "adain"]:
+      result_images = wavelet_color_fix(tensor2pil(image_target), tensor2pil(image_ref)) if method == 'wavelet' else adain_color_fix(tensor2pil(image_target), tensor2pil(image_ref))
+      new_images = pil2tensor(result_images)
+    else:
+      try:
+        from color_matcher import ColorMatcher
+      except:
+        install_package("color-matcher")
+        from color_matcher import ColorMatcher
+      image_ref = image_ref.cpu()
+      image_target = image_target.cpu()
+      batch_size = image_target.size(0)
+      out = []
+      images_target = image_target.squeeze()
+      images_ref = image_ref.squeeze()
+
+      image_ref_np = images_ref.numpy()
+      images_target_np = images_target.numpy()
+      if image_ref.size(0) > 1 and image_ref.size(0) != batch_size:
+        raise ValueError("ColorMatch: Use either single reference image or a matching batch of reference images.")
+      cm = ColorMatcher()
+      for i in range(batch_size):
+        image_target_np = images_target_np if batch_size == 1 else images_target[i].numpy()
+        image_ref_np_i = image_ref_np if image_ref.size(0) == 1 else images_ref[i].numpy()
+        try:
+          image_result = cm.transfer(src=image_target_np, ref=image_ref_np_i, method=method)
+        except BaseException as e:
+          print(f"Error occurred during transfer: {e}")
+          break
+        out.append(torch.from_numpy(image_result))
+
+      new_images = torch.stack(out, dim=0).to(torch.float32)
+
+    results = easySave(new_images, save_prefix, image_output, prompt, extra_pnginfo)
+
+    if image_output in ("Hide", "Hide/Save"):
+      return {"ui": {},
+              "result": (new_images,)}
+
+    return {"ui": {"images": results},
+            "result": (new_images,)}
+
+
+# 图像反推
+from .libs.image import ci
+class imageInterrogator:
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+          "required": {
+              "image": ("IMAGE",),
+              "mode": (['fast','classic','best','negative'],),
+              "use_lowvram": ("BOOLEAN", {"default": True}),
+          }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("prompt",)
+    FUNCTION = "interrogate"
+    CATEGORY = "EasyUse/Image"
+    OUTPUT_NODE = True
+    OUTPUT_IS_LIST = (True,)
+
+    def interrogate(self, image, mode, use_lowvram=False):
+      prompt = ci.image_to_prompt(image, mode, low_vram=use_lowvram)
+      return {"ui":{"text":prompt},"result":(prompt,)}
+
+# 人类分割器
+class humanSegmentation:
+
+    @classmethod
+    def INPUT_TYPES(cls):
+
+        return {
+          "required":{
+            "image": ("IMAGE",),
+            "method": (["selfie_multiclass_256x256", "human_parsing_lip"],),
+            "confidence": ("FLOAT", {"default": 0.4, "min": 0.05, "max": 0.95, "step": 0.01},),
+          },
+          "hidden": {
+              "prompt": "PROMPT",
+              "my_unique_id": "UNIQUE_ID",
+          }
+        }
+
+    RETURN_TYPES = ("IMAGE", "MASK",)
+    RETURN_NAMES = ("image", "mask",)
+    FUNCTION = "parsing"
+    CATEGORY = "EasyUse/Segmentation"
+
+    def get_mediapipe_image(self, image: Image):
+      import mediapipe as mp
+      # Convert image to NumPy array
+      numpy_image = np.asarray(image)
+      image_format = mp.ImageFormat.SRGB
+      # Convert BGR to RGB (if necessary)
+      if numpy_image.shape[-1] == 4:
+        image_format = mp.ImageFormat.SRGBA
+      elif numpy_image.shape[-1] == 3:
+        image_format = mp.ImageFormat.SRGB
+        numpy_image = cv2.cvtColor(numpy_image, cv2.COLOR_BGR2RGB)
+      return mp.Image(image_format=image_format, data=numpy_image)
+
+    def parsing(self, image, confidence, method, prompt=None, my_unique_id=None):
+      mask_components = []
+      if my_unique_id in prompt:
+        if prompt[my_unique_id]["inputs"]['mask_components']:
+          mask_components = prompt[my_unique_id]["inputs"]['mask_components'].split(',')
+      mask_components = list(map(int, mask_components))
+      if method == 'selfie_multiclass_256x256':
+        try:
+          import mediapipe as mp
+        except:
+          install_package("mediapipe")
+          import mediapipe as mp
+
+        from functools import reduce
+
+        model_path = get_local_filepath(MEDIAPIPE_MODELS['selfie_multiclass_256x256']['model_url'], MEDIAPIPE_DIR)
+        model_asset_buffer = None
+        with open(model_path, "rb") as f:
+            model_asset_buffer = f.read()
+        image_segmenter_base_options = mp.tasks.BaseOptions(model_asset_buffer=model_asset_buffer)
+        options = mp.tasks.vision.ImageSegmenterOptions(
+          base_options=image_segmenter_base_options,
+          running_mode=mp.tasks.vision.RunningMode.IMAGE,
+          output_category_mask=True)
+        # Create the image segmenter
+        ret_images = []
+        ret_masks = []
+
+        with mp.tasks.vision.ImageSegmenter.create_from_options(options) as segmenter:
+            for img in image:
+                _image = torch.unsqueeze(img, 0)
+                orig_image = tensor2pil(_image).convert('RGB')
+                # Convert the Tensor to a PIL image
+                i = 255. * img.cpu().numpy()
+                image_pil = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
+                # create our foreground and background arrays for storing the mask results
+                mask_background_array = np.zeros((image_pil.size[0], image_pil.size[1], 4), dtype=np.uint8)
+                mask_background_array[:] = (0, 0, 0, 255)
+                mask_foreground_array = np.zeros((image_pil.size[0], image_pil.size[1], 4), dtype=np.uint8)
+                mask_foreground_array[:] = (255, 255, 255, 255)
+                # Retrieve the masks for the segmented image
+                media_pipe_image = self.get_mediapipe_image(image=image_pil)
+                segmented_masks = segmenter.segment(media_pipe_image)
+                masks = []
+                for i, com in enumerate(mask_components):
+                    masks.append(segmented_masks.confidence_masks[com])
+
+                image_data = media_pipe_image.numpy_view()
+                image_shape = image_data.shape
+                # convert the image shape from "rgb" to "rgba" aka add the alpha channel
+                if image_shape[-1] == 3:
+                    image_shape = (image_shape[0], image_shape[1], 4)
+                mask_background_array = np.zeros(image_shape, dtype=np.uint8)
+                mask_background_array[:] = (0, 0, 0, 255)
+                mask_foreground_array = np.zeros(image_shape, dtype=np.uint8)
+                mask_foreground_array[:] = (255, 255, 255, 255)
+                mask_arrays = []
+                if len(masks) == 0:
+                    mask_arrays.append(mask_background_array)
+                else:
+                    for i, mask in enumerate(masks):
+                        condition = np.stack((mask.numpy_view(),) * image_shape[-1], axis=-1) > confidence
+                        mask_array = np.where(condition, mask_foreground_array, mask_background_array)
+                        mask_arrays.append(mask_array)
+                # Merge our masks taking the maximum from each
+                merged_mask_arrays = reduce(np.maximum, mask_arrays)
+                # Create the image
+                mask_image = Image.fromarray(merged_mask_arrays)
+                # convert PIL image to tensor image
+                tensor_mask = mask_image.convert("RGB")
+                tensor_mask = np.array(tensor_mask).astype(np.float32) / 255.0
+                tensor_mask = torch.from_numpy(tensor_mask)[None,]
+                _mask = tensor_mask.squeeze(3)[..., 0]
+
+                _mask = tensor2pil(tensor_mask).convert('L')
+
+                ret_image = RGB2RGBA(orig_image, _mask)
+                ret_images.append(pil2tensor(ret_image))
+                ret_masks.append(image2mask(_mask))
+
+            output_image = torch.cat(ret_images, dim=0)
+            mask = torch.cat(ret_masks, dim=0)
+
+      elif method == "human_parsing_lip":
+        from .human_parsing.run_parsing import HumanParsing
+        onnx_path = os.path.join(folder_paths.models_dir, 'onnx')
+        model_path = get_local_filepath(HUMANPARSING_MODELS['parsing_lip']['model_url'], onnx_path)
+        parsing = HumanParsing(model_path=model_path)
+        model_image = image.squeeze(0)
+        model_image = model_image.permute((2, 0, 1))
+        model_image = to_pil_image(model_image)
+
+        map_image, mask = parsing(model_image, mask_components)
+
+        mask = mask[:, :, :, 0]
+
+        alpha = 1.0 - mask
+
+        output_image, = JoinImageWithAlpha().join_image_with_alpha(image, alpha)
+
+      return (output_image, mask)
+
+import cv2
+import base64
+class loadImageBase64:
+  @classmethod
+  def INPUT_TYPES(s):
+    return {
+      "required": {
+        "base64_data": ("STRING", {"default": ""}),
+        "image_output": (["Hide", "Preview", "Save", "Hide/Save"], {"default": "Preview"}),
+        "save_prefix": ("STRING", {"default": "ComfyUI"}),
+      },
+      "optional": {
+
+      },
+      "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
+    }
+
+  RETURN_TYPES = ("IMAGE", "MASK")
+  OUTPUT_NODE = True
+  FUNCTION = "load_image"
+  CATEGORY = "EasyUse/Image/LoadImage"
+
+  def convert_color(self, image,):
+    if len(image.shape) > 2 and image.shape[2] >= 4:
+      return cv2.cvtColor(image, cv2.COLOR_BGRA2RGB)
+    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+  def load_image(self, base64_data, image_output, save_prefix, prompt=None, extra_pnginfo=None):
+    nparr = np.frombuffer(base64.b64decode(base64_data), np.uint8)
+
+    result = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
+    channels = cv2.split(result)
+    if len(channels) > 3:
+      mask = channels[3].astype(np.float32) / 255.0
+      mask = torch.from_numpy(mask)
+    else:
+      mask = torch.ones(channels[0].shape, dtype=torch.float32, device="cpu")
+
+    result = self.convert_color(result)
+    result = result.astype(np.float32) / 255.0
+    new_images = torch.from_numpy(result)[None,]
+
+    results = easySave(new_images, save_prefix, image_output, None, None)
+    mask = mask.unsqueeze(0)
+
+    if image_output in ("Hide", "Hide/Save"):
+      return {"ui": {},
+              "result": (new_images, mask)}
+
+    return {"ui": {"images": results},
+            "result": (new_images, mask)}
+
+class imageToBase64:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+        "required": {
+            "image": ("IMAGE",),
+        },
+        }
+
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "to_base64"
+    CATEGORY = "EasyUse/Image"
+    OUTPUT_NODE = True
+
+    def to_base64(self, image, ):
+      import base64
+      from io import BytesIO
+
+      # 将张量图像转换为PIL图像
+      pil_image = tensor2pil(image)
+
+      buffered = BytesIO()
+      pil_image.save(buffered, format="JPEG")
+      image_bytes = buffered.getvalue()
+
+      base64_str = base64.b64encode(image_bytes).decode("utf-8")
+      return {"result": (base64_str,)}
+
+# 姿势编辑器
 class poseEditor:
   @classmethod
   def INPUT_TYPES(self):
@@ -626,6 +1045,7 @@ class poseEditor:
 
 NODE_CLASS_MAPPINGS = {
   "easy imageInsetCrop": imageInsetCrop,
+  "easy imageCount": imageCount,
   "easy imageSize": imageSize,
   "easy imageSizeBySide": imageSizeBySide,
   "easy imageSizeByLongerSide": imageSizeByLongerSide,
@@ -636,14 +1056,22 @@ NODE_CLASS_MAPPINGS = {
   "easy imageRatio": imageRatio,
   "easy imageToMask": imageToMask,
   "easy imageSplitList": imageSplitList,
+  "easy imagesSplitImage": imagesSplitImage,
   "easy imageSave": imageSaveSimple,
   "easy imageRemBg": imageRemBg,
+  "easy imageChooser": imageChooser,
+  "easy imageColorMatch": imageColorMatch,
+  "easy imageInterrogator": imageInterrogator,
+  "easy loadImageBase64": loadImageBase64,
+  "easy imageToBase64": imageToBase64,
   "easy joinImageBatch": JoinImageBatch,
+  "easy humanSegmentation": humanSegmentation,
   "easy poseEditor": poseEditor
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
   "easy imageInsetCrop": "ImageInsetCrop",
+  "easy imageCount": "ImageCount",
   "easy imageSize": "ImageSize",
   "easy imageSizeBySide": "ImageSize (Side)",
   "easy imageSizeByLongerSide": "ImageSize (LongerSide)",
@@ -655,8 +1083,15 @@ NODE_DISPLAY_NAME_MAPPINGS = {
   "easy imageToMask": "ImageToMask",
   "easy imageHSVMask": "ImageHSVMask",
   "easy imageSplitList": "imageSplitList",
+  "easy imagesSplitImage": "imagesSplitImage",
   "easy imageSave": "SaveImage (Simple)",
   "easy imageRemBg": "Image Remove Bg",
+  "easy imageChooser": "Image Chooser",
+  "easy imageColorMatch": "Image Color Match",
+  "easy imageInterrogator": "Image To Prompt",
   "easy joinImageBatch": "JoinImageBatch",
-  "easy poseEditor": "PoseEditor"
+  "easy loadImageBase64": "Load Image (Base64)",
+  "easy imageToBase64": "Image To Base64",
+  "easy humanSegmentation": "Human Segmentation",
+  "easy poseEditor": "PoseEditor",
 }

py/layer_diffuse/func.py

@@ -58,15 +58,15 @@ class LayerDiffuse:
         except:
             pass
 
-        if method in [LayerMethod.FG_ONLY_CONV, LayerMethod.FG_ONLY_ATTN] and sd_version == 'sd15':
+        if method in [LayerMethod.FG_ONLY_CONV, LayerMethod.FG_ONLY_ATTN] and sd_version == 'sd1':
             self.frames = 1
-        elif method in [LayerMethod.BG_TO_BLEND, LayerMethod.FG_TO_BLEND, LayerMethod.BG_BLEND_TO_FG, LayerMethod.FG_BLEND_TO_BG] and sd_version == 'sd15':
+        elif method in [LayerMethod.BG_TO_BLEND, LayerMethod.FG_TO_BLEND, LayerMethod.BG_BLEND_TO_FG, LayerMethod.FG_BLEND_TO_BG] and sd_version == 'sd1':
             self.frames = 2
             batch_size, _, height, width = samples['samples'].shape
             if batch_size % 2 != 0:
                 raise Exception(f"The batch size should be a multiple of 2. 批次大小需为2的倍数")
             control_img = image
-        elif method == LayerMethod.EVERYTHING and sd_version == 'sd15':
+        elif method == LayerMethod.EVERYTHING and sd_version == 'sd1':
             batch_size, _, height, width = samples['samples'].shape
             self.frames = 3
             if batch_size % 3 != 0:
@@ -77,7 +77,7 @@ class LayerDiffuse:
         model_path = get_local_filepath(model_url, LAYER_DIFFUSION_DIR)
         layer_lora_state_dict = load_layer_model_state_dict(model_path)
         work_model = model.clone()
-        if sd_version == 'sd15':
+        if sd_version == 'sd1':
             patcher = AttentionSharingPatcher(
                 work_model, self.frames, use_control=control_img is not None
             )
@@ -97,7 +97,7 @@ class LayerDiffuse:
             else:
                 c_concat = model.model.latent_format.process_in(torch.cat([samples["samples"], blend_samples["samples"]], dim=1))
             samp_model, positive, negative = (work_model,) + self.apply_layer_c_concat(positive, negative, c_concat)
-        elif sd_version == 'sd15':
+        elif sd_version == 'sd1':
             if method in [LayerMethod.BG_TO_BLEND, LayerMethod.BG_BLEND_TO_FG]:
                 additional_cond = (additional_cond[0], None)
             elif method in [LayerMethod.FG_TO_BLEND, LayerMethod.FG_BLEND_TO_BG]:
@@ -166,10 +166,10 @@ class LayerDiffuse:
         alpha = []
         if layer_diffusion_method is not None:
             sd_version = get_sd_version(model)
-            if sd_version not in ['sdxl', 'sd15']:
+            if sd_version not in ['sdxl', 'sd1']:
                 raise Exception(f"Only SDXL and SD1.5 model supported for Layer Diffusion")
             method = self.get_layer_diffusion_method(layer_diffusion_method, blend_samples is not None)
-            sd15_allow = True if sd_version == 'sd15' and method in [LayerMethod.FG_ONLY_ATTN, LayerMethod.EVERYTHING, LayerMethod.BG_TO_BLEND, LayerMethod.BG_BLEND_TO_FG] else False
+            sd15_allow = True if sd_version == 'sd1' and method in [LayerMethod.FG_ONLY_ATTN, LayerMethod.EVERYTHING, LayerMethod.BG_TO_BLEND, LayerMethod.BG_BLEND_TO_FG] else False
             sdxl_allow = True if sd_version == 'sdxl' and method in [LayerMethod.FG_ONLY_CONV, LayerMethod.FG_ONLY_ATTN, LayerMethod.BG_BLEND_TO_FG] else False
             if sdxl_allow or sd15_allow:
                 if self.vae_transparent_decoder is None:

py/layer_diffuse/model.py

@@ -7,18 +7,17 @@ import comfy.model_management
 from comfy.model_patcher import ModelPatcher
 from tqdm import tqdm
 from typing import Optional, Tuple
-
+from ..libs.utils import install_package
+from packaging import version
 
 try:
+    install_package("diffusers", "0.27.2", True, "0.25.0")
+
     from diffusers.configuration_utils import ConfigMixin, register_to_config
     from diffusers.models.modeling_utils import ModelMixin
     from diffusers import __version__
     if __version__:
-        try:
-            diffusers_version = float(__version__.replace('.', '').replace('dev','.'))
-        except ValueError:
-            diffusers_version = 270
-        if diffusers_version < 270:
+        if version.parse(__version__) < version.parse("0.27.0"):
             from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
         else:
             from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block

py/libs/cache.py

@@ -1,12 +1,86 @@
-cache = {}
+import itertools
+from typing import Optional
+
+class TaggedCache:
+    def __init__(self, tag_settings: Optional[dict]=None):
+        self._tag_settings = tag_settings or {}  # tag cache size
+        self._data = {}
+
+    def __getitem__(self, key):
+        for tag_data in self._data.values():
+            if key in tag_data:
+                return tag_data[key]
+        raise KeyError(f'Key `{key}` does not exist')
+
+    def __setitem__(self, key, value: tuple):
+        # value: (tag: str, (islist: bool, data: *))
+
+        # if key already exists, pop old value
+        for tag_data in self._data.values():
+            if key in tag_data:
+                tag_data.pop(key, None)
+                break
+
+        tag = value[0]
+        if tag not in self._data:
+
+            try:
+                from cachetools import LRUCache
+
+                default_size = 20
+                if 'ckpt' in tag:
+                    default_size = 5
+                elif tag in ['latent', 'image']:
+                    default_size = 100
+
+                self._data[tag] = LRUCache(maxsize=self._tag_settings.get(tag, default_size))
+
+            except (ImportError, ModuleNotFoundError):
+                # TODO: implement a simple lru dict
+                self._data[tag] = {}
+        self._data[tag][key] = value
+
+    def __delitem__(self, key):
+        for tag_data in self._data.values():
+            if key in tag_data:
+                del tag_data[key]
+                return
+        raise KeyError(f'Key `{key}` does not exist')
+
+    def __contains__(self, key):
+        return any(key in tag_data for tag_data in self._data.values())
+
+    def items(self):
+        yield from itertools.chain(*map(lambda x :x.items(), self._data.values()))
+
+    def get(self, key, default=None):
+        """D.get(k[,d]) -> D[k] if k in D, else d.  d defaults to None."""
+        for tag_data in self._data.values():
+            if key in tag_data:
+                return tag_data[key]
+        return default
+
+    def clear(self):
+        # clear all cache
+        self._data = {}
+
+cache_settings = {}
+cache = TaggedCache(cache_settings)
 cache_count = {}
 
-
-def update_cache(k, v):
-    cache[k] = v
+def update_cache(k, tag, v):
+    cache[k] = (tag, v)
     cnt = cache_count.get(k)
     if cnt is None:
         cnt = 0
         cache_count[k] = cnt
     else:
-        cache_count[k] += 1
+        cache_count[k] += 1
+def remove_cache(key):
+    global cache
+    if key == '*':
+        cache = TaggedCache(cache_settings)
+    elif key in cache:
+        del cache[key]
+    else:
+        print(f"invalid {key}")

py/libs/chooser.py

@@ -0,0 +1,52 @@
+from server import PromptServer
+from aiohttp import web
+import time
+
+class ChooserCancelled(Exception):
+    pass
+
+class ChooserMessage:
+    stash = {}
+    messages = {}
+    cancelled = False
+
+    @classmethod
+    def addMessage(cls, id, message):
+        if message == '__cancel__':
+            cls.messages = {}
+            cls.cancelled = True
+        elif message == '__start__':
+            cls.messages = {}
+            cls.stash = {}
+            cls.cancelled = False
+        else:
+            cls.messages[str(id)] = message
+
+    @classmethod
+    def waitForMessage(cls, id, period=0.1, asList=False):
+        sid = str(id)
+        while not (sid in cls.messages) and not ("-1" in cls.messages):
+            if cls.cancelled:
+                cls.cancelled = False
+                raise ChooserCancelled()
+            time.sleep(period)
+        if cls.cancelled:
+            cls.cancelled = False
+            raise ChooserCancelled()
+        message = cls.messages.pop(str(id), None) or cls.messages.pop("-1")
+        try:
+            if asList:
+                return [int(x.strip()) for x in message.split(",")]
+            else:
+                return int(message.strip())
+        except ValueError:
+            print(
+                f"ERROR IN IMAGE_CHOOSER - failed to parse '${message}' as ${'comma separated list of ints' if asList else 'int'}")
+            return [1] if asList else 1
+
+
+@PromptServer.instance.routes.post('/easyuse/image_chooser_message')
+async def make_image_selection(request):
+    post = await request.post()
+    ChooserMessage.addMessage(post.get("id"), post.get("message"))
+    return web.json_response({})

py/libs/colorfix.py

@@ -0,0 +1,115 @@
+import torch
+from PIL import Image
+from torch import Tensor
+from torch.nn import functional as F
+
+from torchvision.transforms import ToTensor, ToPILImage
+
+def adain_color_fix(target: Image, source: Image):
+    # Convert images to tensors
+    to_tensor = ToTensor()
+    target_tensor = to_tensor(target).unsqueeze(0)
+    source_tensor = to_tensor(source).unsqueeze(0)
+
+    # Apply adaptive instance normalization
+    result_tensor = adaptive_instance_normalization(target_tensor, source_tensor)
+
+    # Convert tensor back to image
+    to_image = ToPILImage()
+    result_image = to_image(result_tensor.squeeze(0).clamp_(0.0, 1.0))
+
+    return result_image
+
+def wavelet_color_fix(target: Image, source: Image):
+    source = source.resize(target.size, resample=Image.Resampling.LANCZOS)
+
+    # Convert images to tensors
+    to_tensor = ToTensor()
+    target_tensor = to_tensor(target).unsqueeze(0)
+    source_tensor = to_tensor(source).unsqueeze(0)
+
+    # Apply wavelet reconstruction
+    result_tensor = wavelet_reconstruction(target_tensor, source_tensor)
+
+    # Convert tensor back to image
+    to_image = ToPILImage()
+    result_image = to_image(result_tensor.squeeze(0).clamp_(0.0, 1.0))
+
+    return result_image
+
+def calc_mean_std(feat: Tensor, eps=1e-5):
+    """Calculate mean and std for adaptive_instance_normalization.
+    Args:
+        feat (Tensor): 4D tensor.
+        eps (float): A small value added to the variance to avoid
+            divide-by-zero. Default: 1e-5.
+    """
+    size = feat.size()
+    assert len(size) == 4, 'The input feature should be 4D tensor.'
+    b, c = size[:2]
+    feat_var = feat.view(b, c, -1).var(dim=2) + eps
+    feat_std = feat_var.sqrt().view(b, c, 1, 1)
+    feat_mean = feat.view(b, c, -1).mean(dim=2).view(b, c, 1, 1)
+    return feat_mean, feat_std
+
+def adaptive_instance_normalization(content_feat:Tensor, style_feat:Tensor):
+    """Adaptive instance normalization.
+    Adjust the reference features to have the similar color and illuminations
+    as those in the degradate features.
+    Args:
+        content_feat (Tensor): The reference feature.
+        style_feat (Tensor): The degradate features.
+    """
+    size = content_feat.size()
+    style_mean, style_std = calc_mean_std(style_feat)
+    content_mean, content_std = calc_mean_std(content_feat)
+    normalized_feat = (content_feat - content_mean.expand(size)) / content_std.expand(size)
+    return normalized_feat * style_std.expand(size) + style_mean.expand(size)
+
+def wavelet_blur(image: Tensor, radius: int):
+    """
+    Apply wavelet blur to the input tensor.
+    """
+    # input shape: (1, 3, H, W)
+    # convolution kernel
+    kernel_vals = [
+        [0.0625, 0.125, 0.0625],
+        [0.125, 0.25, 0.125],
+        [0.0625, 0.125, 0.0625],
+    ]
+    kernel = torch.tensor(kernel_vals, dtype=image.dtype, device=image.device)
+    # add channel dimensions to the kernel to make it a 4D tensor
+    kernel = kernel[None, None]
+    # repeat the kernel across all input channels
+    kernel = kernel.repeat(3, 1, 1, 1)
+    image = F.pad(image, (radius, radius, radius, radius), mode='replicate')
+    # apply convolution
+    output = F.conv2d(image, kernel, groups=3, dilation=radius)
+    return output
+
+def wavelet_decomposition(image: Tensor, levels=5):
+    """
+    Apply wavelet decomposition to the input tensor.
+    This function only returns the low frequency & the high frequency.
+    """
+    high_freq = torch.zeros_like(image)
+    for i in range(levels):
+        radius = 2 ** i
+        low_freq = wavelet_blur(image, radius)
+        high_freq += (image - low_freq)
+        image = low_freq
+
+    return high_freq, low_freq
+
+def wavelet_reconstruction(content_feat:Tensor, style_feat:Tensor):
+    """
+    Apply wavelet decomposition, so that the content will have the same color as the style.
+    """
+    # calculate the wavelet decomposition of the content feature
+    content_high_freq, content_low_freq = wavelet_decomposition(content_feat)
+    del content_low_freq
+    # calculate the wavelet decomposition of the style feature
+    style_high_freq, style_low_freq = wavelet_decomposition(style_feat)
+    del style_high_freq
+    # reconstruct the content feature with the style's high frequency
+    return content_high_freq + style_low_freq

py/libs/image.py

@@ -1,7 +1,13 @@
+import os
+import base64
 import torch
 import numpy as np
 from enum import Enum
 from PIL import Image
+from io import BytesIO
+
+import folder_paths
+from .utils import install_package
 
 # PIL to Tensor
 def pil2tensor(image):
@@ -10,6 +16,17 @@ def pil2tensor(image):
 def tensor2pil(image):
     return Image.fromarray(np.clip(255. * image.cpu().numpy().squeeze(), 0, 255).astype(np.uint8))
 
+def pil2byte(pil_image, format='PNG'):
+  byte_arr = BytesIO()
+  pil_image.save(byte_arr, format=format)
+  byte_arr.seek(0)
+  return byte_arr
+
+def image2base64(image_base64):
+    image_bytes = base64.b64decode(image_base64)
+    image_data = Image.open(BytesIO(image_bytes))
+    return image_data
+
 # Get new bounds
 def get_new_bounds(width, height, left, right, top, bottom):
   """Returns the new bounds for an image with inset crop data."""
@@ -19,6 +36,56 @@ def get_new_bounds(width, height, left, right, top, bottom):
   bottom = height - bottom
   return (left, right, top, bottom)
 
+def RGB2RGBA(image: Image, mask: Image) -> Image:
+  (R, G, B) = image.convert('RGB').split()
+  return Image.merge('RGBA', (R, G, B, mask.convert('L')))
+
+def image2mask(image: Image) -> torch.Tensor:
+  _image = image.convert('RGBA')
+  alpha = _image.split()[0]
+  bg = Image.new("L", _image.size)
+  _image = Image.merge('RGBA', (bg, bg, bg, alpha))
+  ret_mask = torch.tensor([pil2tensor(_image)[0, :, :, 3].tolist()])
+  return ret_mask
+
+
+# 图像融合
+class blendImage:
+  def g(self, x):
+    return torch.where(x <= 0.25, ((16 * x - 12) * x + 4) * x, torch.sqrt(x))
+
+  def blend_mode(self, img1, img2, mode):
+    if mode == "normal":
+      return img2
+    elif mode == "multiply":
+      return img1 * img2
+    elif mode == "screen":
+      return 1 - (1 - img1) * (1 - img2)
+    elif mode == "overlay":
+      return torch.where(img1 <= 0.5, 2 * img1 * img2, 1 - 2 * (1 - img1) * (1 - img2))
+    elif mode == "soft_light":
+      return torch.where(img2 <= 0.5, img1 - (1 - 2 * img2) * img1 * (1 - img1),
+                         img1 + (2 * img2 - 1) * (self.g(img1) - img1))
+    elif mode == "difference":
+      return img1 - img2
+    else:
+      raise ValueError(f"Unsupported blend mode: {mode}")
+
+  def blend_images(self, image1: torch.Tensor, image2: torch.Tensor, blend_factor: float, blend_mode: str = 'normal'):
+    image2 = image2.to(image1.device)
+    if image1.shape != image2.shape:
+      image2 = image2.permute(0, 3, 1, 2)
+      image2 = comfy.utils.common_upscale(image2, image1.shape[2], image1.shape[1], upscale_method='bicubic',
+                                          crop='center')
+      image2 = image2.permute(0, 2, 3, 1)
+
+    blended_image = self.blend_mode(image1, image2, blend_mode)
+    blended_image = image1 * (1 - blend_factor) + blended_image * blend_factor
+    blended_image = torch.clamp(blended_image, 0, 1)
+    return blended_image
+
+
+
 
 class ResizeMode(Enum):
   RESIZE = "Just Resize"
@@ -33,3 +100,75 @@ class ResizeMode(Enum):
       return 2
     assert False, "NOTREACHED"
 
+
+
+# CLIP反推
+import comfy.utils
+from torchvision import transforms
+Config, Interrogator = None, None
+class CI_Inference:
+  ci_model = None
+  cache_path: str
+
+  def __init__(self):
+    self.ci_model = None
+    self.low_vram = False
+    self.cache_path = os.path.join(folder_paths.models_dir, "clip_interrogator")
+
+  def _load_model(self, model_name, low_vram=False):
+    if not (self.ci_model and model_name == self.ci_model.config.clip_model_name and self.low_vram == low_vram):
+      self.low_vram = low_vram
+      print(f"Load model: {model_name}")
+
+      config = Config(
+        device="cuda" if torch.cuda.is_available() else "cpu",
+        download_cache=True,
+        clip_model_name=model_name,
+        clip_model_path=self.cache_path,
+        cache_path=self.cache_path,
+        caption_model_name='blip-large'
+      )
+
+      if low_vram:
+        config.apply_low_vram_defaults()
+
+      self.ci_model = Interrogator(config)
+
+  def _interrogate(self, image, mode, caption=None):
+    if mode == 'best':
+      prompt = self.ci_model.interrogate(image, caption=caption)
+    elif mode == 'classic':
+      prompt = self.ci_model.interrogate_classic(image, caption=caption)
+    elif mode == 'fast':
+      prompt = self.ci_model.interrogate_fast(image, caption=caption)
+    elif mode == 'negative':
+      prompt = self.ci_model.interrogate_negative(image)
+    else:
+      raise Exception(f"Unknown mode {mode}")
+    return prompt
+
+  def image_to_prompt(self, image, mode, model_name='ViT-L-14/openai', low_vram=False):
+    try:
+      from clip_interrogator import Config, Interrogator
+      global Config, Interrogator
+    except:
+      install_package("clip_interrogator", "0.6.0")
+      from clip_interrogator import Config, Interrogator
+
+    pbar = comfy.utils.ProgressBar(len(image))
+
+    self._load_model(model_name, low_vram)
+    prompt = []
+    for i in range(len(image)):
+      im = image[i]
+
+      im = tensor2pil(im)
+      im = im.convert('RGB')
+
+      _prompt = self._interrogate(im, mode)
+      pbar.update(1)
+      prompt.append(_prompt)
+
+    return prompt
+
+ci = CI_Inference()

py/libs/loader.py

@@ -276,6 +276,32 @@ class easyLoader:
                                             lbw_a, lbw_b, "", lbw)
             else:
                 _lora = comfy.utils.load_torch_file(lora_path, safe_load=True)
+                keys = _lora.keys()
+                if "down_blocks.0.resnets.0.norm1.bias" in keys:
+                    print('Using LORA for Resadapter')
+                    key_map = {}
+                    key_map = comfy.lora.model_lora_keys_unet(model.model, key_map)
+                    mapping_norm = {}
+
+                    for key in keys:
+                        if ".weight" in key:
+                            key_name_in_ori_sd = key_map[key.replace(".weight", "")]
+                            mapping_norm[key_name_in_ori_sd] = _lora[key]
+                        elif ".bias" in key:
+                            key_name_in_ori_sd = key_map[key.replace(".bias", "")]
+                            mapping_norm[key_name_in_ori_sd.replace(".weight", ".bias")] = _lora[
+                                key
+                            ]
+                        else:
+                            print("===>Unexpected key", key)
+                            mapping_norm[key] = _lora[key]
+
+                    for k in mapping_norm.keys():
+                        if k not in model.model.state_dict():
+                            print("===>Missing key:", k)
+                    model.model.load_state_dict(mapping_norm, strict=False)
+                    return (model, clip)
+
                 model, clip = comfy.sd.load_lora_for_models(model, clip, _lora, model_strength, clip_strength)
 
             self.add_to_cache("lora", unique_id, (model, clip))

py/libs/sampler.py

@@ -5,7 +5,7 @@ import latent_preview
 from nodes import MAX_RESOLUTION
 from PIL import Image
 from typing import Dict, List, Optional, Tuple, Union, Any
-
+from .utils import get_sd_version
 class easySampler:
     def __init__(self):
         self.last_helds: dict[str, list] = {
@@ -49,16 +49,10 @@ class easySampler:
 
     def common_ksampler(self, model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, denoise=1.0,
                         disable_noise=False, start_step=None, last_step=None, force_full_denoise=False,
-                        preview_latent=True, disable_pbar=False):
+                        preview_latent=True, disable_pbar=False, custom=None):
         device = comfy.model_management.get_torch_device()
         latent_image = latent["samples"]
 
-        if disable_noise:
-            noise = torch.zeros(latent_image.size(), dtype=latent_image.dtype, layout=latent_image.layout, device="cpu")
-        else:
-            batch_inds = latent["batch_index"] if "batch_index" in latent else None
-            noise = comfy.sample.prepare_noise(latent_image, seed, batch_inds)
-
         noise_mask = None
         if "noise_mask" in latent:
             noise_mask = latent["noise_mask"]
@@ -80,12 +74,29 @@ class easySampler:
                 preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0)
             pbar.update_absolute(step + 1, total_steps, preview_bytes)
 
-        samples = comfy.sample.sample(model, noise, steps, cfg, sampler_name, scheduler, positive, negative,
-                                      latent_image,
-                                      denoise=denoise, disable_noise=disable_noise, start_step=start_step,
-                                      last_step=last_step,
-                                      force_full_denoise=force_full_denoise, noise_mask=noise_mask, callback=callback,
-                                      disable_pbar=disable_pbar, seed=seed)
+        if custom is not None:
+            guider = custom['guider'] if 'guider' in custom else None
+            sampler = custom['sampler'] if 'sampler' in custom else None
+            sigmas = custom['sigmas'] if 'sigmas' in custom else None
+            noise = custom['noise'] if 'noise' in custom else None
+            samples = guider.sample(noise.generate_noise(latent), latent_image, sampler, sigmas,
+                                    denoise_mask=noise_mask, callback=callback, disable_pbar=disable_pbar,
+                                    seed=noise.seed)
+            samples = samples.to(comfy.model_management.intermediate_device())
+        else:
+            if disable_noise:
+                noise = torch.zeros(latent_image.size(), dtype=latent_image.dtype, layout=latent_image.layout,
+                                    device="cpu")
+            else:
+                batch_inds = latent["batch_index"] if "batch_index" in latent else None
+                noise = comfy.sample.prepare_noise(latent_image, seed, batch_inds)
+
+            samples = comfy.sample.sample(model, noise, steps, cfg, sampler_name, scheduler, positive, negative,
+                                          latent_image,
+                                          denoise=denoise, disable_noise=disable_noise, start_step=start_step,
+                                          last_step=last_step,
+                                          force_full_denoise=force_full_denoise, noise_mask=noise_mask, callback=callback,
+                                          disable_pbar=disable_pbar, seed=seed)
 
         out = latent.copy()
         out["samples"] = samples
@@ -209,4 +220,43 @@ class easySampler:
             sdxl_pipe.get("clip"),
             sdxl_pipe.get("images"),
             sdxl_pipe.get("seed")
-        )
+        )
+
+class alignYourStepsScheduler:
+
+    NOISE_LEVELS = {
+        "SD1": [14.6146412293, 6.4745760956, 3.8636745985, 2.6946151520, 1.8841921177, 1.3943805092, 0.9642583904,
+                0.6523686016, 0.3977456272, 0.1515232662, 0.0291671582],
+        "SDXL": [14.6146412293, 6.3184485287, 3.7681790315, 2.1811480769, 1.3405244945, 0.8620721141, 0.5550693289,
+                 0.3798540708, 0.2332364134, 0.1114188177, 0.0291671582],
+        "SVD": [700.00, 54.5, 15.886, 7.977, 4.248, 1.789, 0.981, 0.403, 0.173, 0.034, 0.002]}
+
+
+    def loglinear_interp(self, t_steps, num_steps):
+        """
+        Performs log-linear interpolation of a given array of decreasing numbers.
+        """
+        xs = np.linspace(0, 1, len(t_steps))
+        ys = np.log(t_steps[::-1])
+
+        new_xs = np.linspace(0, 1, num_steps)
+        new_ys = np.interp(new_xs, xs, ys)
+
+        interped_ys = np.exp(new_ys)[::-1].copy()
+        return interped_ys
+
+    def get_sigmas(self, model_type, steps, denoise):
+
+        total_steps = steps
+        if denoise < 1.0:
+            if denoise <= 0.0:
+                return (torch.FloatTensor([]),)
+            total_steps = round(steps * denoise)
+
+        sigmas = self.NOISE_LEVELS[model_type][:]
+        if (steps + 1) != len(sigmas):
+            sigmas = self.loglinear_interp(sigmas, steps + 1)
+
+        sigmas = sigmas[-(total_steps + 1):]
+        sigmas[-1] = 0
+        return (torch.FloatTensor(sigmas),)

py/libs/stability.py

@@ -0,0 +1,201 @@
+import json
+import os
+import yaml
+import requests
+import pathlib
+from aiohttp import web
+from server import PromptServer
+from .image import tensor2pil, pil2tensor, image2base64, pil2byte
+from ..log import log_node_error
+
+
+root_path = pathlib.Path(__file__).parent.parent.parent
+config_path = os.path.join(root_path,'config.yaml')
+default_key = [{'name':'Default', 'key':''}]
+
+class StabilityAPI:
+    def __init__(self):
+        self.api_url = "https://api.stability.ai"
+        self.api_keys = None
+        self.api_current = 0
+        self.user_info = {}
+        self.getAPIKeys()
+
+    def getErrors(self, code):
+        errors = {
+            400: "Bad Request",
+            403: "ApiKey Forbidden",
+            413: "Your request was larger than 10MiB.",
+            429: "You have made more than 150 requests in 10 seconds.",
+            500: "Internal Server Error",
+        }
+        return errors.get(code, "Unknown Error")
+
+    def getAPIKeys(self):
+        if os.path.isfile(config_path):
+            with open(config_path, 'r') as f:
+                data = yaml.load(f, Loader=yaml.FullLoader)
+                if not data:
+                    data = {'STABILITY_API_KEY': default_key, 'STABILITY_API_DEFAULT':0}
+                    with open(config_path, 'w') as f:
+                        yaml.dump(data, f)
+                if 'STABILITY_API_KEY' not in data:
+                    data['STABILITY_API_KEY'] = default_key
+                    data['STABILITY_API_DEFAULT'] = 0
+                    with open(config_path, 'w') as f:
+                        yaml.dump(data, f)
+                api_keys = data['STABILITY_API_KEY']
+                self.api_current = data['STABILITY_API_DEFAULT']
+                self.api_keys = api_keys
+                return api_keys
+        else:
+            # create a yaml file
+            with open(config_path, 'w') as f:
+                data = {'STABILITY_API_KEY': default_key, 'STABILITY_API_DEFAULT':0}
+                yaml.dump(data, f)
+                return data['STABILITY_API_KEY']
+        pass
+
+    def setAPIKeys(self, api_keys):
+        if len(api_keys) > 0:
+            self.api_keys = api_keys
+            # load and save the yaml file
+            with open(config_path, 'r') as f:
+                data = yaml.load(f, Loader=yaml.FullLoader)
+                data['STABILITY_API_KEY'] = api_keys
+                with open(config_path, 'w') as f:
+                    yaml.dump(data, f)
+        return True
+
+    def setAPIDefault(self, current):
+        if current is not None:
+            self.api_current = current
+            # load and save the yaml file
+            with open(config_path, 'r') as f:
+                data = yaml.load(f, Loader=yaml.FullLoader)
+                data['STABILITY_API_DEFAULT'] = current
+                with open(config_path, 'w') as f:
+                    yaml.dump(data, f)
+        return True
+
+    def generate_sd3_image(self, prompt, negative_prompt, aspect_ratio,  model, seed, mode='text-to-image', image=None, strength=1, output_format='png', node_name='easy stableDiffusion3API'):
+        url = f"{self.api_url}/v2beta/stable-image/generate/sd3"
+        api_key = self.api_keys[self.api_current]['key']
+        files = None
+        data = {
+            "prompt": prompt,
+            "mode": mode,
+            "model": model,
+            "seed": seed,
+            "output_format": output_format,
+        }
+        if model == 'sd3':
+            data['negative_prompt'] = negative_prompt
+
+        if mode == 'text-to-image':
+            files = {"none": ''}
+            data['aspect_ratio'] = aspect_ratio
+        elif mode == 'image-to-image':
+            pil_image = tensor2pil(image)
+            image_byte = pil2byte(pil_image)
+            files = {"image": ("output.png", image_byte, 'image/png')}
+            data['strength'] = strength
+
+        response = requests.post(url,
+            headers={"authorization": f"{api_key}", "accept": "application/json"},
+            files=files,
+            data=data,
+        )
+        if response.status_code == 200:
+            PromptServer.instance.send_sync('stable-diffusion-api-generate-succeed',{"model":model})
+            json_data = response.json()
+            image_base64 = json_data['image']
+            image_data = image2base64(image_base64)
+            output_t = pil2tensor(image_data)
+            return output_t
+        else:
+            if 'application/json' in response.headers['Content-Type']:
+                error_info = response.json()
+                log_node_error(node_name,  error_info.get('name', 'No name provided'))
+                log_node_error(node_name, error_info.get('errors', ['No details provided']))
+            error_status_text = self.getErrors(response.status_code)
+            PromptServer.instance.send_sync('easyuse-toast',{"type": "error", "content": error_status_text})
+            raise Exception(f"Failed to generate image: {error_status_text}")
+
+    # get user account
+    async def getUserAccount(self, cache=True):
+        url = f"{self.api_url}/v1/user/account"
+        api_key = self.api_keys[self.api_current]['key']
+        name = self.api_keys[self.api_current]['name']
+        if cache and name in self.user_info:
+            return self.user_info[name]
+        else:
+            response = requests.get(url, headers={"Authorization": f"Bearer {api_key}"})
+            if response.status_code == 200:
+                user_info = response.json()
+                self.user_info[name] = user_info
+                return user_info
+            else:
+                PromptServer.instance.send_sync('easyuse-toast',{'type': 'error', 'content': self.getErrors(response.status_code)})
+                return None
+
+    # get user balance
+    async def getUserBalance(self):
+        url = f"{self.api_url}/v1/user/balance"
+        api_key = self.api_keys[self.api_current]['key']
+        response = requests.get(url, headers={
+            "Authorization": f"Bearer {api_key}"
+        })
+        if response.status_code == 200:
+            return response.json()
+        else:
+            PromptServer.instance.send_sync('easyuse-toast', {'type': 'error', 'content': self.getErrors(response.status_code)})
+            return None
+
+stableAPI = StabilityAPI()
+
+
+@PromptServer.instance.routes.get("/easyuse/stability/api_keys")
+async def get_stability_api_keys(request):
+    stableAPI.getAPIKeys()
+    return web.json_response({"keys": stableAPI.api_keys, "current": stableAPI.api_current})
+
+@PromptServer.instance.routes.post("/easyuse/stability/set_api_keys")
+async def set_stability_api_keys(request):
+    post = await request.post()
+    api_keys = post.get("api_keys")
+    current = post.get('current')
+    if api_keys is not None:
+        api_keys = json.loads(api_keys)
+        stableAPI.setAPIKeys(api_keys)
+        if current is not None:
+            print(current)
+            stableAPI.setAPIDefault(int(current))
+            account = await stableAPI.getUserAccount()
+            balance = await stableAPI.getUserBalance()
+            return web.json_response({'account': account, 'balance': balance})
+        else:
+            return web.json_response({'status': 'ok'})
+    else:
+        return web.Response(status=400)
+
+@PromptServer.instance.routes.post("/easyuse/stability/set_apikey_default")
+async def set_stability_api_default(request):
+    post = await request.post()
+    current = post.get("current")
+    if current is not None and current < len(stableAPI.api_keys):
+        stableAPI.api_current = current
+        return web.json_response({'status': 'ok'})
+    else:
+        return web.Response(status=400)
+
+@PromptServer.instance.routes.get("/easyuse/stability/user_info")
+async def get_account_info(request):
+    account = await stableAPI.getUserAccount()
+    balance = await stableAPI.getUserBalance()
+    return web.json_response({'account': account, 'balance': balance})
+
+@PromptServer.instance.routes.get("/easyuse/stability/balance")
+async def get_balance_info(request):
+    balance = await stableAPI.getUserBalance()
+    return web.json_response({'balance': balance})

py/libs/styleAlign.py

@@ -0,0 +1,148 @@
+import torch
+import torch.nn as nn
+from comfy.model_patcher import ModelPatcher
+from typing import Union
+
+T = torch.Tensor
+
+
+def exists(val):
+    return val is not None
+
+
+def default(val, d):
+    if exists(val):
+        return val
+    return d
+
+
+class StyleAlignedArgs:
+    def __init__(self, share_attn: str) -> None:
+        self.adain_keys = "k" in share_attn
+        self.adain_values = "v" in share_attn
+        self.adain_queries = "q" in share_attn
+
+    share_attention: bool = True
+    adain_queries: bool = True
+    adain_keys: bool = True
+    adain_values: bool = True
+
+
+def expand_first(
+    feat: T,
+    scale=1.0,
+) -> T:
+    """
+    Expand the first element so it has the same shape as the rest of the batch.
+    """
+    b = feat.shape[0]
+    feat_style = torch.stack((feat[0], feat[b // 2])).unsqueeze(1)
+    if scale == 1:
+        feat_style = feat_style.expand(2, b // 2, *feat.shape[1:])
+    else:
+        feat_style = feat_style.repeat(1, b // 2, 1, 1, 1)
+        feat_style = torch.cat([feat_style[:, :1], scale * feat_style[:, 1:]], dim=1)
+    return feat_style.reshape(*feat.shape)
+
+
+def concat_first(feat: T, dim=2, scale=1.0) -> T:
+    """
+    concat the the feature and the style feature expanded above
+    """
+    feat_style = expand_first(feat, scale=scale)
+    return torch.cat((feat, feat_style), dim=dim)
+
+
+def calc_mean_std(feat, eps: float = 1e-5) -> "tuple[T, T]":
+    feat_std = (feat.var(dim=-2, keepdims=True) + eps).sqrt()
+    feat_mean = feat.mean(dim=-2, keepdims=True)
+    return feat_mean, feat_std
+
+def adain(feat: T) -> T:
+    feat_mean, feat_std = calc_mean_std(feat)
+    feat_style_mean = expand_first(feat_mean)
+    feat_style_std = expand_first(feat_std)
+    feat = (feat - feat_mean) / feat_std
+    feat = feat * feat_style_std + feat_style_mean
+    return feat
+
+class SharedAttentionProcessor:
+    def __init__(self, args: StyleAlignedArgs, scale: float):
+        self.args = args
+        self.scale = scale
+
+    def __call__(self, q, k, v, extra_options):
+        if self.args.adain_queries:
+            q = adain(q)
+        if self.args.adain_keys:
+            k = adain(k)
+        if self.args.adain_values:
+            v = adain(v)
+        if self.args.share_attention:
+            k = concat_first(k, -2, scale=self.scale)
+            v = concat_first(v, -2)
+
+        return q, k, v
+
+
+def get_norm_layers(
+    layer: nn.Module,
+    norm_layers_: "dict[str, list[Union[nn.GroupNorm, nn.LayerNorm]]]",
+    share_layer_norm: bool,
+    share_group_norm: bool,
+):
+    if isinstance(layer, nn.LayerNorm) and share_layer_norm:
+        norm_layers_["layer"].append(layer)
+    if isinstance(layer, nn.GroupNorm) and share_group_norm:
+        norm_layers_["group"].append(layer)
+    else:
+        for child_layer in layer.children():
+            get_norm_layers(
+                child_layer, norm_layers_, share_layer_norm, share_group_norm
+            )
+
+
+def register_norm_forward(
+    norm_layer: Union[nn.GroupNorm, nn.LayerNorm],
+) -> Union[nn.GroupNorm, nn.LayerNorm]:
+    if not hasattr(norm_layer, "orig_forward"):
+        setattr(norm_layer, "orig_forward", norm_layer.forward)
+    orig_forward = norm_layer.orig_forward
+
+    def forward_(hidden_states: T) -> T:
+        n = hidden_states.shape[-2]
+        hidden_states = concat_first(hidden_states, dim=-2)
+        hidden_states = orig_forward(hidden_states)  # type: ignore
+        return hidden_states[..., :n, :]
+
+    norm_layer.forward = forward_  # type: ignore
+    return norm_layer
+
+
+def register_shared_norm(
+    model: ModelPatcher,
+    share_group_norm: bool = True,
+    share_layer_norm: bool = True,
+):
+    norm_layers = {"group": [], "layer": []}
+    get_norm_layers(model.model, norm_layers, share_layer_norm, share_group_norm)
+    print(
+        f"Patching {len(norm_layers['group'])} group norms, {len(norm_layers['layer'])} layer norms."
+    )
+    return [register_norm_forward(layer) for layer in norm_layers["group"]] + [
+        register_norm_forward(layer) for layer in norm_layers["layer"]
+    ]
+
+
+SHARE_NORM_OPTIONS = ["both", "group", "layer", "disabled"]
+SHARE_ATTN_OPTIONS = ["q+k", "q+k+v", "disabled"]
+
+
+def styleAlignBatch(model, share_norm, share_attn, scale=1.0):
+    m = model.clone()
+    share_group_norm = share_norm in ["group", "both"]
+    share_layer_norm = share_norm in ["layer", "both"]
+    register_shared_norm(model, share_group_norm, share_layer_norm)
+    args = StyleAlignedArgs(share_attn)
+    m.set_model_attn1_patch(SharedAttentionProcessor(args, scale))
+    return m

py/libs/utils.py

@@ -1,3 +1,10 @@
+class AlwaysEqualProxy(str):
+    def __eq__(self, _):
+        return True
+
+    def __ne__(self, _):
+        return False
+
 comfy_ui_revision = None
 def get_comfyui_revision():
     try:
@@ -10,6 +17,51 @@ def get_comfyui_revision():
         comfy_ui_revision = "Unknown"
     return comfy_ui_revision
 
+
+import sys
+import importlib.util
+import importlib.metadata
+from packaging import version
+from server import PromptServer
+def is_package_installed(package):
+    try:
+        module = importlib.util.find_spec(package)
+        return module is not None
+    except ImportError as e:
+        print(e)
+        return False
+
+def install_package(package, v=None, compare=True, compare_version=None):
+    run_install = True
+    if is_package_installed(package):
+        try:
+            installed_version = importlib.metadata.version(package)
+            if v is not None:
+                if compare_version is None:
+                    compare_version = v
+                if not compare or version.parse(installed_version) >= version.parse(compare_version):
+                    run_install = False
+            else:
+                run_install = False
+        except:
+            run_install = False
+
+    if run_install:
+        import subprocess
+        package_command = package + '==' + v if v is not None else package
+        PromptServer.instance.send_sync("easyuse-toast", {'content': f"Installing {package_command}...", 'duration': 5000})
+        result = subprocess.run([sys.executable, '-s', '-m', 'pip', 'install', package_command], capture_output=True, text=True)
+        if result.returncode == 0:
+            PromptServer.instance.send_sync("easyuse-toast", {'content': f"{package} installed successfully", 'type': 'success', 'duration': 5000})
+            print(f"Package {package} installed successfully")
+            return True
+        else:
+            PromptServer.instance.send_sync("easyuse-toast", {'content': f"{package} installed failed", 'type': 'error', 'duration': 5000})
+            print(f"Package {package} installed failed")
+            return False
+    else:
+        return False
+
 def compare_revision(num):
     global comfy_ui_revision
     if not comfy_ui_revision:
@@ -51,7 +103,11 @@ def get_sd_version(model):
     elif isinstance(
             model_config, (comfy.supported_models.SD15, comfy.supported_models.SD20)
     ):
-        return 'sd15'
+        return 'sd1'
+    elif isinstance(
+            model_config, (comfy.supported_models.SVD_img2vid)
+    ):
+        return 'svd'
     else:
         return 'unknown'
 
@@ -121,11 +177,14 @@ def is_linked_styles_selector(prompt, my_unique_id, prompt_type='positive'):
     else:
         return False
 
+use_mirror = False
 def get_local_filepath(url, dirname, local_file_name=None):
     """Get local file path when is already downloaded or download it"""
     import os
+    from server import PromptServer
     from urllib.parse import urlparse
     from torch.hub import download_url_to_file
+    global use_mirror
     if not os.path.exists(dirname):
         os.makedirs(dirname)
     if not local_file_name:
@@ -133,8 +192,23 @@ def get_local_filepath(url, dirname, local_file_name=None):
         local_file_name = os.path.basename(parsed_url.path)
     destination = os.path.join(dirname, local_file_name)
     if not os.path.exists(destination):
-        print(f'downloading {url} to {destination}')
-        download_url_to_file(url, destination)
+        try:
+            if use_mirror:
+                url = url.replace('huggingface.co', 'hf-mirror.com')
+            print(f'downloading {url} to {destination}')
+            PromptServer.instance.send_sync("easyuse-toast", {'content': f'Downloading model to {destination}, please wait...', 'duration': 10000})
+            download_url_to_file(url, destination)
+        except Exception as e:
+            use_mirror = True
+            url = url.replace('huggingface.co', 'hf-mirror.com')
+            print(f'无法从huggingface下载，正在尝试从 {url} 下载...')
+            PromptServer.instance.send_sync("easyuse-toast", {'content': f'无法连接huggingface，正在尝试从 {url} 下载...', 'duration': 10000})
+            try:
+                download_url_to_file(url, destination)
+            except Exception as err:
+                PromptServer.instance.send_sync("easyuse-toast",
+                                                {'content': f'无法从 {url} 下载模型', 'type':'error'})
+                raise Exception(f'无法从 {url} 下载，错误信息：{str(err.args[0])}')
     return destination
 
 def to_lora_patch_dict(state_dict: dict) -> dict:
@@ -161,10 +235,32 @@ def easySave(images, filename_prefix, output_type, prompt=None, extra_pnginfo=No
     from nodes import PreviewImage, SaveImage
     if output_type == "Hide":
         return list()
-    if output_type == "Preview":
+    if output_type in ["Preview", "Preview&Choose"]:
         filename_prefix = 'easyPreview'
         results = PreviewImage().save_images(images, filename_prefix, prompt, extra_pnginfo)
         return results['ui']['images']
     else:
         results = SaveImage().save_images(images, filename_prefix, prompt, extra_pnginfo)
-        return results['ui']['images']
+        return results['ui']['images']
+
+def getMetadata(filepath):
+    with open(filepath, "rb") as file:
+        # https://github.com/huggingface/safetensors#format
+        # 8 bytes: N, an unsigned little-endian 64-bit integer, containing the size of the header
+        header_size = int.from_bytes(file.read(8), "little", signed=False)
+
+        if header_size <= 0:
+            raise BufferError("Invalid header size")
+
+        header = file.read(header_size)
+        if header_size <= 0:
+            raise BufferError("Invalid header")
+
+        return header
+
+def cleanGPUUsedForce():
+    import torch.cuda
+    import comfy.model_management
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+    comfy.model_management.unload_all_models()

py/logic.py

@@ -1,7 +1,7 @@
 from typing import Iterator, List, Tuple, Dict, Any, Union, Optional
 from _decimal import Context, getcontext
 from decimal import Decimal
-import torch
+from .libs.utils import AlwaysEqualProxy, cleanGPUUsedForce
 import numpy as np
 import json
 
@@ -277,6 +277,30 @@ class imageSwitch:
         else:
             return (image_b, )
 
+class textSwitch:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "input": ("INT", {"default": 1, "min": 1, "max": 2}),
+            },
+            "optional": {
+                "text1": ("STRING", {"forceInput": True}),
+                "text2": ("STRING", {"forceInput": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("STRING",)
+    CATEGORY = "EasyUse/Logic/Switch"
+    FUNCTION = "switch"
+
+    def switch(self, input, text1=None, text2=None,):
+        if input == 1:
+            return (text1,)
+        else:
+            return (text2,)
+
 # ---------------------------------------------------------------运算 开始----------------------------------------------------------------------#
 
 COMPARE_FUNCTIONS = {
@@ -287,12 +311,6 @@ COMPARE_FUNCTIONS = {
     "a <= b": lambda a, b: a <= b,
     "a >= b": lambda a, b: a >= b,
 }
-class AlwaysEqualProxy(str):
-    def __eq__(self, _):
-        return True
-
-    def __ne__(self, _):
-        return False
 
 # 比较
 class Compare:
@@ -512,11 +530,52 @@ class cleanGPUUsed:
     CATEGORY = "EasyUse/Logic"
 
     def empty_cache(self, anything, unique_id=None, extra_pnginfo=None):
-        if torch.cuda.is_available():
-            torch.cuda.empty_cache()
-
+        cleanGPUUsedForce()
         return ()
 
+from .libs.cache import remove_cache
+class clearCacheKey:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {
+            "anything": (AlwaysEqualProxy("*"), {}),
+            "cache_key": ("STRING", {"default": "*"}),
+        }, "optional": {},
+            "hidden": {"unique_id": "UNIQUE_ID", "extra_pnginfo": "EXTRA_PNGINFO",}
+        }
+
+    RETURN_TYPES = ()
+    RETURN_NAMES = ()
+    OUTPUT_NODE = True
+    FUNCTION = "empty_cache"
+    CATEGORY = "EasyUse/Logic"
+
+    def empty_cache(self, anything, cache_name, unique_id=None, extra_pnginfo=None):
+        remove_cache(cache_name)
+        return ()
+
+class clearCacheAll:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {
+            "anything": (AlwaysEqualProxy("*"), {}),
+        }, "optional": {},
+            "hidden": {"unique_id": "UNIQUE_ID", "extra_pnginfo": "EXTRA_PNGINFO",}
+        }
+
+    RETURN_TYPES = ()
+    RETURN_NAMES = ()
+    OUTPUT_NODE = True
+    FUNCTION = "empty_cache"
+    CATEGORY = "EasyUse/Logic"
+
+    def empty_cache(self, anything, unique_id=None, extra_pnginfo=None):
+        remove_cache('*')
+        return ()
+
+
+
+
 NODE_CLASS_MAPPINGS = {
   "easy string": String,
   "easy int": Int,
@@ -526,13 +585,16 @@ NODE_CLASS_MAPPINGS = {
   "easy boolean": Boolean,
   "easy compare": Compare,
   "easy imageSwitch": imageSwitch,
+  "easy textSwitch": textSwitch,
   "easy if": If,
   "easy isSDXL": isSDXL,
   "easy xyAny": xyAny,
   "easy convertAnything": ConvertAnything,
   "easy showAnything": showAnything,
   "easy showTensorShape": showTensorShape,
-  "easy cleanGpuUsed": cleanGPUUsed
+  "easy clearCacheKey": clearCacheKey,
+  "easy clearCacheAll": clearCacheAll,
+  "easy cleanGpuUsed": cleanGPUUsed,
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
   "easy string": "String",
@@ -543,11 +605,14 @@ NODE_DISPLAY_NAME_MAPPINGS = {
   "easy boolean": "Boolean",
   "easy compare": "Compare",
   "easy imageSwitch": "Image Switch",
+  "easy textSwitch": "Text Switch",
   "easy if": "If",
   "easy isSDXL": "Is SDXL",
   "easy xyAny": "XYAny",
   "easy convertAnything": "Convert Any",
   "easy showAnything": "Show Any",
   "easy showTensorShape": "Show Tensor Shape",
+  "easy clearCacheKey": "Clear Cache Key",
+  "easy clearCacheAll": "Clear Cache All",
   "easy cleanGpuUsed": "Clean GPU Used"
 }

py/wildcards.py

@@ -156,9 +156,8 @@ def process(text, seed=None):
 
     def replace_wildcard(string):
         global easy_wildcard_dict
-        pattern = r"__([\w.\-+/*\\]+)__"
+        pattern = r"__([\w\s.\-+/*\\]+?)__"
         matches = re.findall(pattern, string)
-
         replacements_found = False
 
         for match in matches:

requirements.txt

@@ -1,2 +1,4 @@
 diffusers>=0.25.0
+clip_interrogator>=0.6.0
+onnxruntime
 aiohttp

web/css/account.css

@@ -0,0 +1,93 @@
+.easyuse-account{
+
+}
+.easyuse-account-user{
+    font-size: 10px;
+    color:var(--descrip-text);
+    text-align: center;
+}
+.easyuse-account-user-info{
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    padding-bottom:10px;
+    cursor: pointer;
+}
+.easyuse-account-user-info .user{
+    display: flex;
+    align-items: center;
+}
+.easyuse-account-user-info .edit{
+    padding:5px 10px;
+    background: var(--comfy-menu-bg);
+    border-radius:4px;
+}
+.easyuse-account-user-info:hover{
+    filter:brightness(110%);
+}
+.easyuse-account-user-info h5{
+    margin:0;
+    font-size: 10px;
+    text-align: left;
+}
+.easyuse-account-user-info h6{
+    margin:0;
+    font-size: 8px;
+    text-align: left;
+    font-weight: 300;
+}
+.easyuse-account-user-info .remark{
+    margin-top: 4px;
+}
+.easyuse-account-user-info .avatar{
+    width: 36px;
+    height: 36px;
+    background: var(--comfy-input-bg);
+    border-radius: 50%;
+    margin-right: 5px;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    font-size: 16px;
+    overflow: hidden;
+}
+.easyuse-account-user-info .avatar img{
+    width: 100%;
+    height: 100%;
+}
+.easyuse-account-dialog{
+    width: 600px;
+}
+.easyuse-account-dialog-main a, .easyuse-account-dialog-main a:visited{
+    font-weight: 400;
+    color: var(--theme-color-light);
+}
+.easyuse-account-dialog-item{
+    display: flex;
+    justify-content: flex-start;
+    align-items: center;
+    padding: 10px 0;
+    border-bottom: 1px solid var(--border-color);
+}
+.easyuse-account-dialog-item input{
+    padding:5px;
+    margin-right:5px;
+}
+.easyuse-account-dialog-item input.key{
+    flex:1;
+}
+.easyuse-account-dialog-item button{
+    cursor: pointer;
+    margin-left:5px!important;
+    padding:5px!important;
+    font-size: 16px!important;
+}
+.easyuse-account-dialog-item button:hover{
+    filter:brightness(120%);
+}
+.easyuse-account-dialog-item button.choose {
+    background: var(--theme-color);
+}
+.easyuse-account-dialog-item button.delete{
+    background: var(--error-color);
+}

web/css/chooser.css

@@ -0,0 +1,30 @@
+.easyuse-chooser-dialog{
+    max-width: 600px;
+}
+.easyuse-chooser-dialog-title{
+    font-size: 18px;
+    font-weight: 700;
+    text-align: center;
+    color:var(--input-text);
+    margin:0;
+}
+.easyuse-chooser-dialog-images{
+    margin-top:10px;
+    display: flex;
+    flex-wrap: wrap;
+    width: 100%;
+    box-sizing: border-box;
+}
+.easyuse-chooser-dialog-images img{
+    width: 50%;
+    height: auto;
+    cursor: pointer;
+    box-sizing: border-box;
+    filter:brightness(80%);
+}
+.easyuse-chooser-dialog-images img:hover{
+    filter:brightness(100%);
+}
+.easyuse-chooser-dialog-images img.selected{
+    border: 4px solid var(--success-color);
+}

web/css/index.css

@@ -2,4 +2,9 @@
 @import "dropdown.css";
 @import "selector.css";
 @import "groupmap.css";
-@import "contextmenu.css";
+@import "contextmenu.css";
+@import "modelinfo.css";
+@import "toast.css";
+@import "account.css";
+@import "chooser.css";
+@import "toolbar.css";

web/css/modelinfo.css

@@ -0,0 +1,265 @@
+.easyuse-model-info {
+	color: white;
+	max-width: 90vw;
+	font-family: var(--font-family);
+}
+.easyuse-model-content {
+	display: flex;
+	flex-direction: column;
+	overflow: hidden;
+}
+.easyuse-model-header{
+	margin:0 0 15px 0;
+}
+.easyuse-model-header-remark{
+	display: flex;
+	align-items: center;
+	margin-top:5px;
+}
+.easyuse-model-info h2 {
+	text-align: left;
+	margin:0;
+}
+.easyuse-model-info h5 {
+	text-align: left;
+	margin:0 15px 0 0px;
+	font-weight: 400;
+	color:var(--descrip-text);
+}
+.easyuse-model-info p {
+	margin: 5px 0;
+}
+.easyuse-model-info a {
+	color: var(--theme-color-light);
+}
+.easyuse-model-info a:hover {
+	text-decoration: underline;
+}
+.easyuse-model-tags-list {
+	display: flex;
+	flex-wrap: wrap;
+	list-style: none;
+	gap: 10px;
+	max-height: 200px;
+	overflow: auto;
+	margin: 10px 0;
+	padding: 0;
+}
+.easyuse-model-tag {
+	background-color: var(--comfy-input-bg);
+	border: 2px solid var(--border-color);
+	color: var(--input-text);
+	display: flex;
+	align-items: center;
+	gap: 5px;
+	border-radius: 5px;
+	padding: 2px 5px;
+	cursor: pointer;
+}
+.easyuse-model-tag--selected span::before {
+	content: "✅";
+	position: absolute;
+	background-color: var(--theme-color-light);
+	left: 0;
+	top: 0;
+	right: 0;
+	bottom: 0;
+	text-align: center;
+}
+.easyuse-model-tag:hover {
+	border: 2px solid var(--theme-color-light);
+}
+.easyuse-model-tag p {
+	margin: 0;
+}
+.easyuse-model-tag span {
+	text-align: center;
+	border-radius: 5px;
+	background-color: var(--theme-color-light);
+	padding: 2px;
+	position: relative;
+	min-width: 20px;
+	overflow: hidden;
+	color: #fff;
+}
+
+.easyuse-model-metadata .comfy-modal-content {
+	max-width: 100%;
+}
+.easyuse-model-metadata label {
+	margin-right: 1ch;
+	color: #ccc;
+}
+
+.easyuse-model-metadata span {
+	color: var(--theme-color-light);
+}
+
+.easyuse-preview {
+	max-width:660px;
+	margin-right: 15px;
+	position: relative;
+}
+.easyuse-preview-group{
+	position: relative;
+	overflow: hidden;
+	border-radius:.5rem;
+	width: 660px;
+}
+.easyuse-preview-list{
+	display: flex;
+	flex-wrap: nowrap;
+	width: 100%;
+	transition: all .5s ease-in-out;
+}
+.easyuse-preview-list.no-transition{
+	transition: none;
+}
+.easyuse-preview-slide{
+	display: flex;
+	flex-basis: calc(50% - 5px);
+	flex-grow: 0;
+	flex-shrink: 0;
+	position: relative;
+	justify-content: center;
+	align-items: center;
+	padding-right:5px;
+	padding-left:0;
+}
+.easyuse-preview-slide:nth-child(even){
+	padding-left:5px;
+	padding-right:0;
+}
+.easyuse-preview-slide-content{
+	position: relative;
+	min-height:150px;
+	width: 100%;
+}
+.easyuse-preview-slide-content .save{
+    position: absolute;
+    right: 6px;
+    z-index: 12;
+    bottom: 6px;
+    display: flex;
+    align-items: center;
+    height: 26px;
+    padding: 0 9px;
+    color: var(--input-text);
+    font-size: 12px;
+    line-height: 26px;
+    background: rgba(0, 0, 0, .5);
+    border-radius: 13px;
+    cursor: pointer;
+	min-width:80px;
+	text-align: center;
+}
+.easyuse-preview-slide-content .save:hover{
+	filter: brightness(120%);
+	will-change: auto;
+}
+
+.easyuse-preview-slide-content img {
+	border-radius: 14px;
+	object-position: center center;
+	max-width: 100%;
+	max-height:700px;
+	border-style: none;
+    vertical-align: middle;
+}
+.easyuse-preview button {
+	position: absolute;
+	z-index:10;
+	top: 50%;
+	display: flex;
+	align-items: center;
+	justify-content: center;
+	width:30px;
+	height:30px;
+	border-radius:15px;
+	border:1px solid rgba(66, 63, 78, .15);
+	background-color: rgba(66, 63, 78, .5);
+	color:hsla(0, 0%, 100%, .8);
+	transition-property: color, background-color, border-color, text-decoration-color, fill, stroke;
+    transition-timing-function: cubic-bezier(.4,0,.2,1);
+    transition-duration: .15s;
+	transform: translateY(-50%);
+}
+.easyuse-preview button.left{
+	left:10px;
+}
+.easyuse-preview button.right{
+	right:10px;
+}
+
+.easyuse-model-detail{
+	margin-top: 16px;
+    overflow: hidden;
+    border: 1px solid var(--border-color);
+    border-radius: 8px;
+	width:300px;
+}
+.easyuse-model-detail-head{
+	height: 40px;
+    padding: 0 10px;
+    font-weight: 500;
+    font-size: 14px;
+    font-style: normal;
+    line-height: 40px;
+}
+.easyuse-model-detail-body{
+	box-sizing: border-box;
+    font-size: 12px;
+}
+.easyuse-model-detail-item{
+	display: flex;
+    justify-content: flex-start;
+    border-top: 1px solid var(--border-color);
+}
+.easyuse-model-detail-item-label{
+	flex-shrink: 0;
+    width: 88px;
+    padding-top: 5px;
+    padding-bottom: 5px;
+    padding-left: 10px;
+    border-right: 1px solid var(--border-color);
+    color: var(--input-text);
+    font-weight: 400;
+}
+.easyuse-model-detail-item-value{
+	display: flex;
+    flex-wrap: wrap;
+    padding: 5px 10px 5px 10px;
+	color: var(--input-text);
+}
+.easyuse-model-detail-textarea{
+	border-top:1px solid var(--border-color);
+	padding:10px;
+	height:100px;
+	overflow-y: auto;
+	font-size: 12px;
+}
+.easyuse-model-detail-textarea textarea{
+	width:100%;
+	height:100%;
+	border:0;
+	background-color:transparent;
+	color: var(--input-text);
+}
+.easyuse-model-detail-textarea textarea::placeholder{
+	color:var(--descrip-text);
+}
+.easyuse-model-detail-textarea.empty{
+	display: flex;
+	justify-content: center;
+	align-items: center;
+	color: var(--descrip-text);
+}
+
+.easyuse-model-notes {
+	background-color: rgba(0, 0, 0, 0.25);
+	padding: 5px;
+	margin-top: 5px;
+}
+.easyuse-model-notes:empty {
+	display: none;
+}

web/css/selector.css

@@ -1,3 +1,6 @@
+.easyuse-prompt-styles{
+    overflow: auto;
+}
 .easyuse-prompt-styles .tools{
     display:flex;
     justify-content:space-between;
@@ -41,9 +44,13 @@
     min-height: 150px;
     height: calc(100% - 40px);
     overflow: auto;
-    // display: flex;
-    // flex-wrap: wrap;
+    /*display: flex;*/
+    /*flex-wrap: wrap;*/
 }
+.easyuse-prompt-styles-list.no-top{
+    height: auto;
+}
+
 .easyuse-prompt-styles-tag{
     display: inline-block;
     vertical-align: middle;

web/css/theme.css

@@ -1,5 +1,8 @@
 :root {
    --theme-color:#3f3eed;
-   --theme-color-light:#006691;
+   --theme-color-light: #008ecb;
+   --success-color: #52c41a;
+   --error-color: #ff4d4f;
+   --warning-color: #faad14;
    --font-family: Inter, -apple-system, BlinkMacSystemFont, Helvetica Neue, sans-serif;
 }

web/css/toast.css

@@ -0,0 +1,110 @@
+.easyuse-toast-container{
+    position: fixed;
+    z-index: 99999;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 0;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    justify-content: start;
+    padding:10px 0;
+}
+.easyuse-toast-container > div {
+  position: relative;
+  height: fit-content;
+  padding: 4px;
+  margin-top: -100px; /* re-set by JS */
+  opacity: 0;
+  transition: all 0.33s ease-in-out;
+  z-index: 3;
+}
+
+.easyuse-toast-container > div:last-child {
+  z-index: 2;
+}
+
+.easyuse-toast-container > div:not(.-show) {
+  z-index: 1;
+}
+
+.easyuse-toast-container > div.-show {
+  opacity: 1;
+  margin-top: 0px !important;
+}
+
+.easyuse-toast-container > div.-show {
+  opacity: 1;
+  transform: translateY(0%);
+}
+
+.easyuse-toast-container > div > div {
+  position: relative;
+  background: var(--comfy-menu-bg);
+  color: var(--input-text);
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  justify-content: center;
+  height: fit-content;
+  box-shadow: 0 0 10px rgba(0, 0, 0, 0.88);
+  padding: 9px 12px;
+  border-radius: 8px;
+  font-family: Arial, sans-serif;
+  font-size: 14px;
+  pointer-events: all;
+}
+
+.easyuse-toast-container > div > div > span {
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  justify-content: center;
+}
+
+.easyuse-toast-container > div > div > span svg {
+  width: 16px;
+  height: auto;
+  margin-right: 8px;
+}
+
+.easyuse-toast-container > div > div > span svg[data-icon=info-circle]{
+  fill: var(--theme-color-light);
+}
+.easyuse-toast-container > div > div > span svg[data-icon=check-circle]{
+  fill: var(--success-color);
+}
+.easyuse-toast-container > div > div > span svg[data-icon=close-circle]{
+  fill: var(--error-color);
+}
+.easyuse-toast-container > div > div > span svg[data-icon=exclamation-circle]{
+  fill: var(--warning-color);
+}
+/*rotate animation*/
+@keyframes rotate {
+  0% {
+    transform: rotate(0deg);
+  }
+  100% {
+    transform: rotate(360deg);
+  }
+}
+.easyuse-toast-container > div > div > span svg[data-icon=loading]{
+  fill: var(--theme-color);
+  animation: rotate 1s linear infinite;
+}
+
+.easyuse-toast-container a {
+  cursor: pointer;
+  text-decoration: underline;
+  color: var(--theme-color-light);
+  margin-left: 4px;
+  display: inline-block;
+  line-height: 1;
+}
+
+.easyuse-toast-container a:hover {
+  color: var(--theme-color-light);
+  text-decoration: none;
+}

web/css/toolbar.css

@@ -0,0 +1,160 @@
+.easyuse-toolbar{
+    background: rgba(35,35,35,.5);
+    backdrop-filter: blur(4px) brightness(120%);
+    border-radius:0 15px 15px 0;
+    min-width:50px;
+    height:24px;
+    position: fixed;
+    bottom:85px;
+    left:0px;
+    display: flex;
+    align-items: center;
+    z-index:10000;
+}
+.easyuse-toolbar.disable-render-info{
+    bottom: 55px;
+}
+.easyuse-toolbar-item{
+    border-radius:20px;
+    height: 20px;
+    width:20px;
+    cursor: pointer;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    transition: all 0.3s ease-in-out;
+    margin-left:2.5px;
+}
+.easyuse-toolbar-icon{
+    width: 14px;
+    height: 14px;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    font-size: 12px;
+    color:white;
+    transition: all 0.3s ease-in-out;
+}
+.easyuse-toolbar-tips{
+    visibility: hidden;
+    opacity: 0;
+    position: absolute;
+    top: -25px;
+    left: 0;
+    color: var(--descrip-text);
+    padding: 2px 5px;
+    border-radius: 5px;
+    font-size: 11px;
+    min-width:100px;
+    transition: all 0.3s ease-in-out;
+}
+.easyuse-toolbar-item:hover{
+    background:rgba(12,12,12,1);
+}
+.easyuse-toolbar-item:hover .easyuse-toolbar-tips{
+    opacity: 1;
+    visibility: visible;
+}
+.easyuse-toolbar-item:hover .easyuse-toolbar-icon.group{
+    color:var(--warning-color);
+}
+.easyuse-toolbar-item:hover .easyuse-toolbar-icon.rocket{
+  color:var(--theme-color-light);
+}
+.easyuse-toolbar-item:hover .easyuse-toolbar-icon.question{
+  color:var(--success-color);
+}
+
+
+.easyuse-guide-dialog{
+    min-width: 600px;
+    font-family: var(--font-family);
+}
+.easyuse-guide-dialog-title{
+    color:var(--input-text);
+    font-size: 20px;
+    font-weight: bold;
+    margin-bottom: 5px;
+}
+.easyuse-guide-dialog-remark{
+    color: var(--input-text);
+    font-size: 14px;
+}
+.easyuse-guide-dialog-content{
+    max-height: 600px;
+    overflow: auto;
+}
+.easyuse-guide-dialog a, .easyuse-guide-dialog a:visited{
+    color: var(--theme-color-light);
+    cursor: pointer;
+}
+.easyuse-guide-dialog-note{
+    margin-top: 20px;
+    color:white;
+    font-size: 14px;
+}
+.easyuse-guide-dialog p{
+    margin:4px 0;
+    font-size: 12px;
+    font-weight: 300;
+}
+.markdown-body h1, .markdown-body h2, .markdown-body h3, .markdown-body h4, .markdown-body h5, .markdown-body h6 {
+    margin-top: 24px;
+    margin-bottom: 16px;
+    font-weight: 600;
+    line-height: 1.25;
+    padding-bottom: 10px;
+    border-bottom: 1px solid var(--border-color);
+    color: var(--input-text);
+}
+.markdown-body table {
+    display: block;
+    width: 100%;
+    width: max-content;
+    max-width: 100%;
+    overflow: auto;
+    color:var(--input-text);
+    box-sizing: border-box;
+    border: 1px solid var(--border-color);
+    text-align: left;
+    width: 100%;
+}
+.markdown-body table th, .markdown-body table td {
+    padding: 6px 13px;
+    font-size: 14px;
+    margin:0;
+    border-right: 1px solid var(--border-color);
+    border-bottom: 1px solid var(--border-color);
+}
+.markdown-body table td {
+    font-size: 12px;
+}
+.markdown-body table th:last-child, .markdown-body table td:last-child{
+    border-right: none;
+}
+.markdown-body table tr:last-child td{
+    border-bottom: none;
+}
+.markdown-body table th{
+    font-weight: bold;
+    width: auto;
+    min-width: 100px;
+}
+.markdown-body table th:last-child{
+    width:100%;
+}
+.markdown-body a{
+    margin-right: 10px;
+}
+.markdown-body .warning{
+    color:var(--warning-color)
+}
+.markdown-body .error{
+    color:var(--error-color)
+}
+.markdown-body .success{
+    color:var(--success-color)
+}
+.markdown-body .link{
+    color:var(--theme-color-light)
+}

web/js/bookmark.js

@@ -1,4 +1,4 @@
-import { app } from "/scripts/app.js";
+import { app } from "../../../scripts/app.js";
 
 
 app.registerExtension({

web/js/common/i18n.js

@@ -2,8 +2,44 @@ import {getLocale} from './utils.js'
 const locale = getLocale()
 
 const zhCN = {
+    "Workflow created by": "工作流创建者",
+    "Watch more video content": "观看更多视频内容",
+    "Workflow Guide":"工作流指南",
+    // ExtraMenu
+    "💎 View Checkpoint Info...": "💎 查看 Checkpoint 信息...",
+    "💎 View Lora Info...": "💎 查看 Lora 信息...",
+    "🔃 Reload Node": "🔃 刷新节点",
+    // ModelInfo
+    "Updated At:": "最近更新：",
+    "Created At:": "首次发布：",
+    "✏️ Edit": "✏️ 编辑",
+    "💾 Save": "💾 保存",
+    "No notes": "当前还没有备注内容",
+    "Saving Notes...": "正在保存备注...",
+    "Type your notes here":"在这里输入备注内容",
+    "ModelName":"模型名称",
+    "Models Required":"所需模型",
+    "Download Model": "下载模型",
+    "Source Url": "模型源地址",
+    "Notes": "备注",
+    "Type": "类型",
+    "Trained Words": "训练词",
+    "BaseModel": "基础算法",
+    "Details": "详情",
+    "Download": "下载量",
+    "Source": "来源",
+    "Saving Preview...": "正在保存预览图...",
+    "Saving Succeed":"保存成功",
+    "Clean SuccessFully":"清理成功",
+    "Clean Failed": "清理失败",
+    "Saving Failed":"保存失败",
+    "No COMBO link": "沒有找到COMBO连接",
+    "Reboot ComfyUI":"重启ComfyUI",
+    "Are you sure you'd like to reboot the server?": "是否要重启ComfyUI？",
     // GroupMap
-    "Groups Map (EasyUse)": "管理组 (EasyUse)",
+    "Groups Map": "管理组",
+    "Cleanup Of GPU Usage": "清理GPU占用",
+    "Please stop all running tasks before cleaning GPU": "请在清理GPU之前停止所有运行中的任务",
     "Always": "启用中",
     "Bypass": "已忽略",
     "Never": "已停用",
@@ -12,9 +48,53 @@ const zhCN = {
     // Quick
     "Enable ALT+1~9 to paste nodes from nodes template (ComfyUI-Easy-Use)": "启用ALT1~9从节点模板粘贴到工作流 (ComfyUI-Easy-Use)",
     "Enable process bar in queue button (ComfyUI-Easy-Use)": "启用提示词队列进度显示条 (ComfyUI-Easy-Use）",
-    "Enable ContextMenu Auto Nest Subdirectories (ComfyUI-Easy-Use)": "启用上下文菜单自动嵌套子目录 (ComfyUI-Easy-Use)"
+    "Enable ContextMenu Auto Nest Subdirectories (ComfyUI-Easy-Use)": "启用上下文菜单自动嵌套子目录 (ComfyUI-Easy-Use)",
+    "Enable tool bar fixed on the left-bottom (ComfyUI-Easy-Use)": "启用工具栏固定在左下角 (ComfyUI-Easy-Use)",
+    "Too many thumbnails, have closed the display": "模型缩略图太多啦，为您关闭了显示",
+    // selector
+    "Empty All": "清空所有",
+    "🔎 Type here to search styles ...": "🔎 在此处输入以搜索样式 ...",
+    // account
+    "Loading UserInfo...": "正在获取用户信息...",
+    "Please set the APIKEY first": "请先设置APIKEY",
+    "Setting APIKEY": "设置APIKEY",
+    "Save Account Info": "保存账号信息",
+    "Choose": "选择",
+    "Delete": "删除",
+    "Edit": "编辑",
+    "At least one account is required": "删除失败: 至少需要一个账户",
+    "APIKEY is not Empty": "APIKEY 不能为空",
+    "Add Account": "添加账号",
+    "Getting Your APIKEY": "获取您的APIKEY",
+    // choosers
+    "Choose Selected Images": "选择选中的图片",
+    "Choose images to continue": "选择图片以继续",
+    // seg
+    "Background": "背景",
+    "Hat": "帽子",
+    "Hair": "头发",
+    "Body": "身体",
+    "Face": "脸部",
+    "Clothes": "衣服",
+    "Others": "其他",
+    "Glove": "手套",
+    "Sunglasses": "太阳镜",
+    "Upper-clothes": "上衣",
+    "Dress": "连衣裙",
+    "Coat": "外套",
+    "Socks": "袜子",
+    "Pants": "裤子",
+    "Jumpsuits": "连体衣",
+    "Scarf": "围巾",
+    "Skirt": "裙子",
+    "Left-arm": "左臂",
+    "Right-arm": "右臂",
+    "Left-leg": "左腿",
+    "Right-leg": "右腿",
+    "Left-shoe": "左鞋",
+    "Right-shoe": "右鞋",
 }
-
 export const $t = (key) => {
-    return locale === 'zh-CN' ? zhCN[key] : key
+    const cn = zhCN[key]
+    return locale === 'zh-CN' && cn ? cn : key
 }

web/js/common/icon.js

@@ -0,0 +1,4 @@
+export const quesitonIcon = `<svg t="1714564780771" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="1489" width="200" height="200" data-spm-anchor-id="a313x.search_index.0.i2.5a663a81pw6qup"><path d="M514.048 54.272q95.232 0 178.688 36.352t145.92 98.304 98.304 145.408 35.84 178.688-35.84 178.176-98.304 145.408-145.92 98.304-178.688 35.84-178.176-35.84-145.408-98.304-98.304-145.408-35.84-178.176 35.84-178.688 98.304-145.408 145.408-98.304 178.176-36.352zM515.072 826.368q26.624 0 44.544-17.92t17.92-43.52q0-26.624-17.92-44.544t-44.544-17.92-44.544 17.92-17.92 44.544q0 25.6 17.92 43.52t44.544 17.92zM567.296 574.464q-1.024-16.384 20.48-34.816t48.128-40.96 49.152-50.688 24.576-65.024q2.048-39.936-8.192-74.752t-33.792-59.904-60.928-39.936-87.552-14.848q-62.464 0-103.936 22.016t-67.072 53.248-35.84 64.512-9.216 55.808q1.024 26.624 16.896 38.912t34.304 12.8 33.792-10.24 15.36-31.232q0-12.288 7.68-30.208t20.992-34.304 32.256-27.648 42.496-11.264q46.08 0 73.728 23.04t25.6 57.856q0 17.408-10.24 32.256t-26.112 28.672-33.792 27.648-33.792 28.672-26.624 32.256-11.776 37.888l1.024 38.912q0 15.36 14.336 29.184t37.888 14.848q23.552-1.024 37.376-15.36t12.8-32.768l0-24.576z" p-id="1490" fill="currentColor"></path></svg>`
+export const rocketIcon = `<svg t="1714565020764" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="7999" width="200" height="200"><path d="M810.438503 379.664884l-71.187166-12.777183C737.426025 180.705882 542.117647 14.602496 532.991087 7.301248c-12.777184-10.951872-32.855615-10.951872-47.45811 0-9.12656 7.301248-204.434938 175.229947-206.26025 359.586453l-67.536542 10.951871c-18.253119 3.650624-31.030303 18.253119-31.030303 36.506239v189.832442c0 10.951872 5.475936 21.903743 12.777184 27.379679 7.301248 5.475936 14.602496 9.12656 23.729055 9.12656h5.475936l133.247772-23.729055c40.156863 47.458111 91.265597 73.012478 151.500891 73.012477 60.235294 0 111.344029-27.379679 151.500891-74.837789l136.898396 23.729055h5.475936c9.12656 0 16.427807-3.650624 23.729055-9.12656 9.12656-7.301248 12.777184-16.427807 12.777184-27.379679V412.520499c1.825312-14.602496-10.951872-29.204991-27.379679-32.855615zM620.606061 766.631016H401.568627c-20.078431 0-36.506239 16.427807-36.506238 36.506239v109.518716c0 14.602496 9.12656 29.204991 23.729055 34.680927 14.602496 5.475936 31.030303 1.825312 40.156863-9.126559l16.427807-18.25312 32.855615 80.313726c5.475936 14.602496 18.253119 23.729055 34.680927 23.729055 16.427807 0 27.379679-9.12656 34.680927-23.729055l32.855615-80.313726 16.427807 18.25312c10.951872 10.951872 25.554367 14.602496 40.156863 9.126559 14.602496-5.475936 23.729055-18.253119 23.729055-34.680927v-109.518716c-3.650624-20.078431-20.078431-36.506239-40.156862-36.506239z" fill="currentColor" p-id="8000"></path></svg>`
+export const groupIcon = `<svg t="1714565543756" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="22538" width="200" height="200"><path d="M871.616 64H152.384c-31.488 0-60.416 25.28-60.416 58.24v779.52c0 32.896 26.24 58.24 60.352 58.24h719.232c34.112 0 60.352-25.344 60.352-58.24V122.24c0.128-32.96-28.8-58.24-60.288-58.24zM286.272 512c-23.616 0-44.672-20.224-44.672-43.008 0-22.784 20.992-43.008 44.608-43.008 23.616 0 44.608 20.224 44.608 43.008A43.328 43.328 0 0 1 286.272 512z m0-202.496c-23.616 0-44.608-20.224-44.608-43.008 0-22.784 20.992-43.008 44.608-43.008 23.616 0 44.608 20.224 44.608 43.008a43.456 43.456 0 0 1-44.608 43.008zM737.728 512H435.904c-23.68 0-44.672-20.224-44.672-43.008 0-22.784 20.992-43.008 44.608-43.008h299.264c23.616 0 44.608 20.224 44.608 43.008a42.752 42.752 0 0 1-41.984 43.008z m0-202.496H435.904c-23.616 0-44.608-20.224-44.608-43.008 0-22.784 20.992-43.008 44.608-43.008h299.264c23.616 0 44.608 20.224 44.608 43.008a42.88 42.88 0 0 1-42.048 43.008z" p-id="22539" fill="currentColor"></path></svg>`
+export const rebootIcon = `<svg t="1714568501931" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="4275" width="200" height="200"><path d="M511.721751 0.000278a511.999861 511.999861 0 1 0 512.277971 511.721751A511.721751 511.721751 0 0 0 511.721751 0.000278zM184.386696 511.722029A36.988583 36.988583 0 0 1 222.487718 475.011556h92.888622a36.710473 36.710473 0 0 1 0 73.420947H222.487718a36.710473 36.710473 0 0 1-38.101022-36.710474z m201.351385 158.522499l-65.911986 65.911987a36.988583 36.988583 0 0 1-62.852781-25.864197 38.101022 38.101022 0 0 1 10.846276-26.142307L333.731577 618.238024a36.710473 36.710473 0 1 1 52.006504 52.006504z m29.201513-256.138985a36.710473 36.710473 0 0 1-52.006504 0l-65.633877-65.633877a36.988583 36.988583 0 0 1 26.142307-62.85278 36.154254 36.154254 0 0 1 25.864197 10.846276L414.939594 361.54282a36.988583 36.988583 0 0 1 0 52.562723z m135.439398 373.779366a37.266693 37.266693 0 0 1-36.988583 36.988583 36.988583 36.988583 0 0 1-36.710473-36.988583V695.274397a36.988583 36.988583 0 0 1 36.710473-36.988583A37.266693 37.266693 0 0 1 550.378992 695.274397z m0-459.437137a37.266693 37.266693 0 0 1-36.988583 36.988583 36.988583 36.988583 0 0 1-36.710473-36.988583V235.559149a36.988583 36.988583 0 0 1 36.710473-36.988583 37.544802 37.544802 0 0 1 36.988583 36.988583z m63.965219 15.85225L679.978088 278.109926a36.710473 36.710473 0 0 1 52.006504 51.728394L667.463154 396.584635a37.544802 37.544802 0 0 1-52.284614 0 36.988583 36.988583 0 0 1-10.568166-26.142306 36.432364 36.432364 0 0 1 9.733837-26.142307z m122.090135 397.974905a37.544802 37.544802 0 0 1-52.284613 0l-65.355767-65.911986a36.154254 36.154254 0 0 1 0-51.728395 36.710473 36.710473 0 0 1 25.864197-10.846276 35.876145 35.876145 0 0 1 25.864197 10.846276l65.911986 65.633877a36.988583 36.988583 0 0 1 0 52.006504z m66.468206-194.676753h-92.888622a36.710473 36.710473 0 0 1 0-73.420947h92.888622a36.710473 36.710473 0 0 1 0 73.420947z" fill="currentColor" p-id="4276"></path></svg>`

web/js/common/model.js

@@ -0,0 +1,683 @@
+import { $el, ComfyDialog } from "../../../../scripts/ui.js";
+import { api } from "../../../../scripts/api.js";
+import {formatTime} from './utils.js';
+import {$t} from "./i18n.js";
+import {toast} from "./toast.js";
+
+class MetadataDialog extends ComfyDialog {
+	constructor() {
+		super();
+		this.element.classList.add("easyuse-model-metadata");
+	}
+	show(metadata) {
+		super.show(
+			$el(
+				"div",
+				Object.keys(metadata).map((k) =>
+					$el("div", [$el("label", { textContent: k }), $el("span", { textContent: metadata[k] })])
+				)
+			)
+		);
+	}
+}
+
+export class ModelInfoDialog extends ComfyDialog {
+	constructor(name) {
+		super();
+		this.name = name;
+		this.element.classList.add("easyuse-model-info");
+	}
+
+	get customNotes() {
+		return this.metadata["easyuse.notes"];
+	}
+
+	set customNotes(v) {
+		this.metadata["easyuse.notes"] = v;
+	}
+
+	get hash() {
+		return this.metadata["easyuse.sha256"];
+	}
+
+	async show(type, value) {
+		this.type = type;
+
+		const req = api.fetchApi("/easyuse/metadata/" + encodeURIComponent(`${type}/${value}`));
+		this.info = $el("div", { style: { flex: "auto" } });
+		// this.img = $el("img", { style: { display: "none" } });
+		this.imgCurrent = 0
+		this.imgList = $el("div.easyuse-preview-list",{
+			style: { display: "none" }
+		})
+		this.imgWrapper = $el("div.easyuse-preview", [
+			$el("div.easyuse-preview-group",[
+				this.imgList
+			]),
+		]);
+		this.main = $el("main", { style: { display: "flex" } }, [this.imgWrapper, this.info]);
+		this.content = $el("div.easyuse-model-content", [
+			$el("div.easyuse-model-header",[$el("h2", { textContent: this.name })])
+			, this.main]);
+
+		const loading = $el("div", { textContent: "ℹ️ Loading...", parent: this.content });
+
+		super.show(this.content);
+
+		this.metadata = await (await req).json();
+		this.viewMetadata.style.cursor = this.viewMetadata.style.opacity = "";
+		this.viewMetadata.removeAttribute("disabled");
+
+		loading.remove();
+		this.addInfo();
+	}
+
+	createButtons() {
+		const btns = super.createButtons();
+		this.viewMetadata = $el("button", {
+			type: "button",
+			textContent: "View raw metadata",
+			disabled: "disabled",
+			style: {
+				opacity: 0.5,
+				cursor: "not-allowed",
+			},
+			onclick: (e) => {
+				if (this.metadata) {
+					new MetadataDialog().show(this.metadata);
+				}
+			},
+		});
+
+		btns.unshift(this.viewMetadata);
+		return btns;
+	}
+
+	parseNote() {
+		if (!this.customNotes) return [];
+
+		let notes = [];
+		// Extract links from notes
+		const r = new RegExp("(\\bhttps?:\\/\\/[^\\s]+)", "g");
+		let end = 0;
+		let m;
+		do {
+			m = r.exec(this.customNotes);
+			let pos;
+			let fin = 0;
+			if (m) {
+				pos = m.index;
+				fin = m.index + m[0].length;
+			} else {
+				pos = this.customNotes.length;
+			}
+
+			let pre = this.customNotes.substring(end, pos);
+			if (pre) {
+				pre = pre.replaceAll("\n", "<br>");
+				notes.push(
+					$el("span", {
+						innerHTML: pre,
+					})
+				);
+			}
+			if (m) {
+				notes.push(
+					$el("a", {
+						href: m[0],
+						textContent: m[0],
+						target: "_blank",
+					})
+				);
+			}
+
+			end = fin;
+		} while (m);
+		return notes;
+	}
+
+	addInfoEntry(name, value) {
+		return $el(
+			"p",
+			{
+				parent: this.info,
+			},
+			[
+				typeof name === "string" ? $el("label", { textContent: name + ": " }) : name,
+				typeof value === "string" ? $el("span", { textContent: value }) : value,
+			]
+		);
+	}
+
+	async getCivitaiDetails() {
+		const req = await fetch("https://civitai.com/api/v1/model-versions/by-hash/" + this.hash);
+		if (req.status === 200) {
+			return await req.json();
+		} else if (req.status === 404) {
+			throw new Error("Model not found");
+		} else {
+			throw new Error(`Error loading info (${req.status}) ${req.statusText}`);
+		}
+	}
+
+	addCivitaiInfo() {
+		const promise = this.getCivitaiDetails();
+		const content = $el("span", { textContent: "ℹ️ Loading..." });
+
+		this.addInfoEntry(
+			$el("label", [
+				$el("img", {
+					style: {
+						width: "18px",
+						position: "relative",
+						top: "3px",
+						margin: "0 5px 0 0",
+					},
+					src: "https://civitai.com/favicon.ico",
+				}),
+				$el("span", { textContent: "Civitai: " }),
+			]),
+			content
+		);
+
+		return promise
+			.then((info) => {
+				this.imgWrapper.style.display = 'block'
+				// 变更标题信息
+				let header = this.element.querySelector('.easyuse-model-header')
+				if(header){
+					header.replaceChildren(
+						$el("h2", { textContent: this.name }),
+						$el("div.easyuse-model-header-remark",[
+							$el("h5", { textContent:  $t("Updated At:") + formatTime(new Date(info.updatedAt),'yyyy/MM/dd')}),
+							$el("h5", { textContent: $t("Created At:") + formatTime(new Date(info.updatedAt),'yyyy/MM/dd')}),
+						])
+					)
+				}
+				// 替换内容
+				let textarea = null
+				let notes = this.parseNote.call(this)
+				let editText = $t("✏️ Edit")
+				console.log(notes)
+				let textarea_div = $el("div.easyuse-model-detail-textarea",[
+					$el("p",notes?.length>0 ? notes : {textContent:$t('No notes')}),
+				])
+				if(!notes || notes.length == 0) textarea_div.classList.add('empty')
+				else textarea_div.classList.remove('empty')
+				this.info.replaceChildren(
+					$el("div.easyuse-model-detail",[
+						$el("div.easyuse-model-detail-head.flex-b",[
+							$el('span',$t("Notes")),
+							$el("a", {
+								textContent: editText,
+								href: "#",
+								style: {
+									fontSize: "12px",
+									float: "right",
+									color: "var(--warning-color)",
+									textDecoration: "none",
+								},
+								onclick: async (e) => {
+									e.preventDefault();
+
+									if (textarea) {
+										if(textarea.value != this.customNotes){
+											toast.showLoading($t('Saving Notes...'))
+											this.customNotes = textarea.value;
+											const resp = await api.fetchApi(
+												"/easyuse/metadata/notes/" + encodeURIComponent(`${this.type}/${this.name}`),
+												{
+													method: "POST",
+													body: this.customNotes,
+												}
+											);
+											toast.hideLoading()
+											if (resp.status !== 200) {
+												toast.error($t('Saving Failed'))
+												console.error(resp);
+												alert(`Error saving notes (${resp.status}) ${resp.statusText}`);
+												return;
+											}
+											toast.success($t('Saving Succeed'))
+											notes = this.parseNote.call(this)
+											console.log(notes)
+											textarea_div.replaceChildren($el("p",notes?.length>0 ? notes : {textContent:$t('No notes')}));
+											if(textarea.value) textarea_div.classList.remove('empty')
+											else textarea_div.classList.add('empty')
+										}else {
+											textarea_div.replaceChildren($el("p",{textContent:$t('No notes')}));
+											textarea_div.classList.add('empty')
+										}
+										e.target.textContent = editText;
+										textarea.remove();
+										textarea = null;
+
+									} else {
+										e.target.textContent = "💾 Save";
+										textarea = $el("textarea", {
+											placeholder: $t("Type your notes here"),
+											style: {
+												width: "100%",
+												minWidth: "200px",
+												minHeight: "50px",
+												height:"100px"
+											},
+											textContent: this.customNotes,
+										});
+										textarea_div.replaceChildren(textarea);
+										textarea.focus()
+									}
+								}
+							})
+						]),
+						textarea_div
+					]),
+					$el("div.easyuse-model-detail",[
+						$el("div.easyuse-model-detail-head",{textContent:$t("Details")}),
+						$el("div.easyuse-model-detail-body",[
+							$el("div.easyuse-model-detail-item",[
+								$el("div.easyuse-model-detail-item-label",{textContent:$t("Type")}),
+								$el("div.easyuse-model-detail-item-value",{textContent:info.model.type}),
+							]),
+							$el("div.easyuse-model-detail-item",[
+								$el("div.easyuse-model-detail-item-label",{textContent:$t("BaseModel")}),
+								$el("div.easyuse-model-detail-item-value",{textContent:info.baseModel}),
+							]),
+							$el("div.easyuse-model-detail-item",[
+								$el("div.easyuse-model-detail-item-label",{textContent:$t("Download")}),
+								$el("div.easyuse-model-detail-item-value",{textContent:info.stats?.downloadCount || 0}),
+							]),
+							$el("div.easyuse-model-detail-item",[
+								$el("div.easyuse-model-detail-item-label",{textContent:$t("Trained Words")}),
+								$el("div.easyuse-model-detail-item-value",{textContent:info?.trainedWords.join(',') || '-'}),
+							]),
+							$el("div.easyuse-model-detail-item",[
+								$el("div.easyuse-model-detail-item-label",{textContent:$t("Source")}),
+								$el("div.easyuse-model-detail-item-value",[
+									$el("label", [
+										$el("img", {
+											style: {
+												width: "14px",
+												position: "relative",
+												top: "3px",
+												margin: "0 5px 0 0",
+											},
+											src: "https://civitai.com/favicon.ico",
+										}),
+										$el("a", {
+											href: "https://civitai.com/models/" + info.modelId,
+											textContent: "View " + info.model.name,
+											target: "_blank",
+										})
+									])
+								]),
+							])
+						]),
+					])
+				);
+
+				if (info.images?.length) {
+					this.imgCurrent = 0
+					this.isSaving = false
+					info.images.map(cate=>
+						cate.url &&
+						this.imgList.appendChild(
+							$el('div.easyuse-preview-slide',[
+								$el('div.easyuse-preview-slide-content',[
+									$el('img',{src:(cate.url)}),
+									$el("div.save", {
+										textContent: "Save as preview",
+										onclick: async () => {
+											if(this.isSaving) return
+											this.isSaving = true
+											toast.showLoading($t('Saving Preview...'))
+											// Convert the preview to a blob
+											const blob = await (await fetch(cate.url)).blob();
+
+											// Store it in temp
+											const name = "temp_preview." + new URL(cate.url).pathname.split(".")[1];
+											const body = new FormData();
+											body.append("image", new File([blob], name));
+											body.append("overwrite", "true");
+											body.append("type", "temp");
+
+											const resp = await api.fetchApi("/upload/image", {
+												method: "POST",
+												body,
+											});
+
+											if (resp.status !== 200) {
+												this.isSaving = false
+												toast.error($t('Saving Failed'))
+												toast.hideLoading()
+												console.error(resp);
+												alert(`Error saving preview (${req.status}) ${req.statusText}`);
+												return;
+											}
+
+											// Use as preview
+											await api.fetchApi("/easyuse/save/" + encodeURIComponent(`${this.type}/${this.name}`), {
+												method: "POST",
+												body: JSON.stringify({
+													filename: name,
+													type: "temp",
+												}),
+												headers: {
+													"content-type": "application/json",
+												},
+											}).then(_=>{
+												toast.success($t('Saving Succeed'))
+												toast.hideLoading()
+											});
+											this.isSaving = false
+											app.refreshComboInNodes();
+										},
+									})
+								])
+							])
+						)
+					)
+					let _this = this
+					this.imgDistance = (-660 * this.imgCurrent).toString()
+					this.imgList.style.display = ''
+					this.imgList.style.transform = 'translate3d(' + this.imgDistance +'px, 0px, 0px)'
+					this.slides = this.imgList.querySelectorAll('.easyuse-preview-slide')
+					// 添加按钮
+					this.slideLeftButton = $el("button.left",{
+						parent: this.imgWrapper,
+						style:{
+							display:info.images.length <= 2 ? 'none' : 'block'
+						},
+						innerHTML:`<svg viewBox="0 0 15 15" fill="none" xmlns="http://www.w3.org/2000/svg" width="16" height="16" style="transform: rotate(90deg);"><path d="M3.13523 6.15803C3.3241 5.95657 3.64052 5.94637 3.84197 6.13523L7.5 9.56464L11.158 6.13523C11.3595 5.94637 11.6759 5.95657 11.8648 6.15803C12.0536 6.35949 12.0434 6.67591 11.842 6.86477L7.84197 10.6148C7.64964 10.7951 7.35036 10.7951 7.15803 10.6148L3.15803 6.86477C2.95657 6.67591 2.94637 6.35949 3.13523 6.15803Z" fill="currentColor" fill-rule="evenodd" clip-rule="evenodd"></path></svg>`,
+						onclick: ()=>{
+							if(info.images.length <= 2) return
+							_this.imgList.classList.remove("no-transition")
+							if(_this.imgCurrent == 0){
+								_this.imgCurrent = (info.images.length/2)-1
+								this.slides[this.slides.length-1].style.transform = 'translate3d(' + (-660 * (this.imgCurrent+1)).toString()+'px, 0px, 0px)'
+								this.slides[this.slides.length-2].style.transform = 'translate3d(' + (-660 * (this.imgCurrent+1)).toString()+'px, 0px, 0px)'
+								_this.imgList.style.transform = 'translate3d(660px, 0px, 0px)'
+								setTimeout(_=>{
+									this.slides[this.slides.length-1].style.transform = 'translate3d(0px, 0px, 0px)'
+									this.slides[this.slides.length-2].style.transform = 'translate3d(0px, 0px, 0px)'
+									_this.imgDistance = (-660 * this.imgCurrent).toString()
+									_this.imgList.style.transform = 'translate3d(' + _this.imgDistance +'px, 0px, 0px)'
+									_this.imgList.classList.add("no-transition")
+								},500)
+							}
+							else {
+								_this.imgCurrent = _this.imgCurrent-1
+								_this.imgDistance = (-660 * this.imgCurrent).toString()
+								_this.imgList.style.transform = 'translate3d(' + _this.imgDistance +'px, 0px, 0px)'
+							}
+						}
+					})
+					this.slideRightButton = $el("button.right",{
+						parent: this.imgWrapper,
+						style:{
+							display:info.images.length <= 2 ? 'none' : 'block'
+						},
+						innerHTML:`<svg viewBox="0 0 15 15" fill="none" xmlns="http://www.w3.org/2000/svg" width="16" height="16" style="transform: rotate(-90deg);"><path d="M3.13523 6.15803C3.3241 5.95657 3.64052 5.94637 3.84197 6.13523L7.5 9.56464L11.158 6.13523C11.3595 5.94637 11.6759 5.95657 11.8648 6.15803C12.0536 6.35949 12.0434 6.67591 11.842 6.86477L7.84197 10.6148C7.64964 10.7951 7.35036 10.7951 7.15803 10.6148L3.15803 6.86477C2.95657 6.67591 2.94637 6.35949 3.13523 6.15803Z" fill="currentColor" fill-rule="evenodd" clip-rule="evenodd"></path></svg>`,
+						onclick: ()=>{
+							if(info.images.length <= 2) return
+							_this.imgList.classList.remove("no-transition")
+
+							if( _this.imgCurrent >= (info.images.length/2)-1){
+								_this.imgCurrent = 0
+								const max = info.images.length/2
+								this.slides[0].style.transform = 'translate3d(' + (660 * max).toString()+'px, 0px, 0px)'
+								this.slides[1].style.transform = 'translate3d(' + (660 * max).toString()+'px, 0px, 0px)'
+								_this.imgList.style.transform = 'translate3d(' + (-660 * max).toString()+'px, 0px, 0px)'
+								setTimeout(_=>{
+									this.slides[0].style.transform = 'translate3d(0px, 0px, 0px)'
+									this.slides[1].style.transform = 'translate3d(0px, 0px, 0px)'
+									_this.imgDistance = (-660 * this.imgCurrent).toString()
+									_this.imgList.style.transform = 'translate3d(' + _this.imgDistance +'px, 0px, 0px)'
+									_this.imgList.classList.add("no-transition")
+								},500)
+							}
+							else  {
+								_this.imgCurrent = _this.imgCurrent+1
+								_this.imgDistance = (-660 * this.imgCurrent).toString()
+								_this.imgList.style.transform = 'translate3d(' + _this.imgDistance +'px, 0px, 0px)'
+							}
+
+						}
+					})
+
+				}
+
+				if(info.description){
+					$el("div", {
+						parent: this.content,
+						innerHTML: info.description,
+						style: {
+							marginTop: "10px",
+						},
+					});
+				}
+
+				return info;
+			})
+			.catch((err) => {
+				this.imgWrapper.style.display = 'none'
+				content.textContent = "⚠️ " + err.message;
+			})
+			.finally(_=>{
+			})
+	}
+}
+
+
+export class CheckpointInfoDialog extends ModelInfoDialog {
+    async addInfo() {
+        // super.addInfo();
+        await this.addCivitaiInfo();
+    }
+}
+
+const MAX_TAGS = 500
+export class LoraInfoDialog extends ModelInfoDialog {
+	getTagFrequency() {
+		if (!this.metadata.ss_tag_frequency) return [];
+
+		const datasets = JSON.parse(this.metadata.ss_tag_frequency);
+		const tags = {};
+		for (const setName in datasets) {
+			const set = datasets[setName];
+			for (const t in set) {
+				if (t in tags) {
+					tags[t] += set[t];
+				} else {
+					tags[t] = set[t];
+				}
+			}
+		}
+
+		return Object.entries(tags).sort((a, b) => b[1] - a[1]);
+	}
+
+	getResolutions() {
+		let res = [];
+		if (this.metadata.ss_bucket_info) {
+			const parsed = JSON.parse(this.metadata.ss_bucket_info);
+			if (parsed?.buckets) {
+				for (const { resolution, count } of Object.values(parsed.buckets)) {
+					res.push([count, `${resolution.join("x")} * ${count}`]);
+				}
+			}
+		}
+		res = res.sort((a, b) => b[0] - a[0]).map((a) => a[1]);
+		let r = this.metadata.ss_resolution;
+		if (r) {
+			const s = r.split(",");
+			const w = s[0].replace("(", "");
+			const h = s[1].replace(")", "");
+			res.push(`${w.trim()}x${h.trim()} (Base res)`);
+		} else if ((r = this.metadata["modelspec.resolution"])) {
+			res.push(r + " (Base res");
+		}
+		if (!res.length) {
+			res.push("⚠️ Unknown");
+		}
+		return res;
+	}
+
+	getTagList(tags) {
+		return tags.map((t) =>
+			$el(
+				"li.easyuse-model-tag",
+				{
+					dataset: {
+						tag: t[0],
+					},
+					$: (el) => {
+						el.onclick = () => {
+							el.classList.toggle("easyuse-model-tag--selected");
+						};
+					},
+				},
+				[
+					$el("p", {
+						textContent: t[0],
+					}),
+					$el("span", {
+						textContent: t[1],
+					}),
+				]
+			)
+		);
+	}
+
+	addTags() {
+		let tags = this.getTagFrequency();
+		let hasMore;
+		if (tags?.length) {
+			const c = tags.length;
+			let list;
+			if (c > MAX_TAGS) {
+				tags = tags.slice(0, MAX_TAGS);
+				hasMore = $el("p", [
+					$el("span", { textContent: `⚠️ Only showing first ${MAX_TAGS} tags ` }),
+					$el("a", {
+						href: "#",
+						textContent: `Show all ${c}`,
+						onclick: () => {
+							list.replaceChildren(...this.getTagList(this.getTagFrequency()));
+							hasMore.remove();
+						},
+					}),
+				]);
+			}
+			list = $el("ol.easyuse-model-tags-list", this.getTagList(tags));
+			this.tags = $el("div", [list]);
+		} else {
+			this.tags = $el("p", { textContent: "⚠️ No tag frequency metadata found" });
+		}
+
+		this.content.append(this.tags);
+
+		if (hasMore) {
+			this.content.append(hasMore);
+		}
+	}
+
+	async addInfo() {
+		// this.addInfoEntry("Name", this.metadata.ss_output_name || "⚠️ Unknown");
+		// this.addInfoEntry("Base Model", this.metadata.ss_sd_model_name || "⚠️ Unknown");
+		// this.addInfoEntry("Clip Skip", this.metadata.ss_clip_skip || "⚠️ Unknown");
+		//
+		// this.addInfoEntry(
+		// 	"Resolution",
+		// 	$el(
+		// 		"select",
+		// 		this.getResolutions().map((r) => $el("option", { textContent: r }))
+		// 	)
+		// );
+
+		// super.addInfo();
+		const p = this.addCivitaiInfo();
+		this.addTags();
+
+		const info = await p;
+		if (info) {
+			// $el(
+			// 	"p",
+			// 	{
+			// 		parent: this.content,
+			// 		textContent: "Trained Words: ",
+			// 	},
+			// 	[
+			// 		$el("pre", {
+			// 			textContent: info.trainedWords.join(", "),
+			// 			style: {
+			// 				whiteSpace: "pre-wrap",
+			// 				margin: "10px 0",
+			// 				background: "#222",
+			// 				padding: "5px",
+			// 				borderRadius: "5px",
+			// 				maxHeight: "250px",
+			// 				overflow: "auto",
+			// 			},
+			// 		}),
+			// 	]
+			// );
+			$el("div", {
+				parent: this.content,
+				innerHTML: info.description,
+				style: {
+					maxHeight: "250px",
+					overflow: "auto",
+				},
+			});
+		}
+	}
+
+	createButtons() {
+		const btns = super.createButtons();
+
+		function copyTags(e, tags) {
+			const textarea = $el("textarea", {
+				parent: document.body,
+				style: {
+					position: "fixed",
+				},
+				textContent: tags.map((el) => el.dataset.tag).join(", "),
+			});
+			textarea.select();
+			try {
+				document.execCommand("copy");
+				if (!e.target.dataset.text) {
+					e.target.dataset.text = e.target.textContent;
+				}
+				e.target.textContent = "Copied " + tags.length + " tags";
+				setTimeout(() => {
+					e.target.textContent = e.target.dataset.text;
+				}, 1000);
+			} catch (ex) {
+				prompt("Copy to clipboard: Ctrl+C, Enter", text);
+			} finally {
+				document.body.removeChild(textarea);
+			}
+		}
+
+		btns.unshift(
+			$el("button", {
+				type: "button",
+				textContent: "Copy Selected",
+				onclick: (e) => {
+					copyTags(e, [...this.tags.querySelectorAll(".easyuse-model-tag--selected")]);
+				},
+			}),
+			$el("button", {
+				type: "button",
+				textContent: "Copy All",
+				onclick: (e) => {
+					copyTags(e, [...this.tags.querySelectorAll(".easyuse-model-tag")]);
+				},
+			})
+		);
+
+		return btns;
+	}
+}

web/js/common/toast.js

@@ -0,0 +1,127 @@
+import {sleep} from "./utils.js";
+import {$t} from "./i18n.js";
+
+class Toast{
+
+    constructor() {
+        this.info_icon = `<svg focusable="false" data-icon="info-circle" width="1em" height="1em" fill="currentColor" aria-hidden="true" viewBox="64 64 896 896"><path d="M512 64C264.6 64 64 264.6 64 512s200.6 448 448 448 448-200.6 448-448S759.4 64 512 64zm32 664c0 4.4-3.6 8-8 8h-48c-4.4 0-8-3.6-8-8V456c0-4.4 3.6-8 8-8h48c4.4 0 8 3.6 8 8v272zm-32-344a48.01 48.01 0 010-96 48.01 48.01 0 010 96z"></path></svg>`
+        this.success_icon = `<svg focusable="false" data-icon="check-circle" width="1em" height="1em" fill="currentColor" aria-hidden="true" viewBox="64 64 896 896"><path d="M512 64C264.6 64 64 264.6 64 512s200.6 448 448 448 448-200.6 448-448S759.4 64 512 64zm193.5 301.7l-210.6 292a31.8 31.8 0 01-51.7 0L318.5 484.9c-3.8-5.3 0-12.7 6.5-12.7h46.9c10.2 0 19.9 4.9 25.9 13.3l71.2 98.8 157.2-218c6-8.3 15.6-13.3 25.9-13.3H699c6.5 0 10.3 7.4 6.5 12.7z"></path></svg>`
+        this.error_icon = `<svg focusable="false" data-icon="close-circle" width="1em" height="1em" fill="currentColor" aria-hidden="true" fill-rule="evenodd" viewBox="64 64 896 896"><path d="M512 64c247.4 0 448 200.6 448 448S759.4 960 512 960 64 759.4 64 512 264.6 64 512 64zm127.98 274.82h-.04l-.08.06L512 466.75 384.14 338.88c-.04-.05-.06-.06-.08-.06a.12.12 0 00-.07 0c-.03 0-.05.01-.09.05l-45.02 45.02a.2.2 0 00-.05.09.12.12 0 000 .07v.02a.27.27 0 00.06.06L466.75 512 338.88 639.86c-.05.04-.06.06-.06.08a.12.12 0 000 .07c0 .03.01.05.05.09l45.02 45.02a.2.2 0 00.09.05.12.12 0 00.07 0c.02 0 .04-.01.08-.05L512 557.25l127.86 127.87c.04.04.06.05.08.05a.12.12 0 00.07 0c.03 0 .05-.01.09-.05l45.02-45.02a.2.2 0 00.05-.09.12.12 0 000-.07v-.02a.27.27 0 00-.05-.06L557.25 512l127.87-127.86c.04-.04.05-.06.05-.08a.12.12 0 000-.07c0-.03-.01-.05-.05-.09l-45.02-45.02a.2.2 0 00-.09-.05.12.12 0 00-.07 0z"></path></svg>`
+        this.warn_icon = `<svg focusable="false" data-icon="exclamation-circle" width="1em" height="1em" fill="currentColor" aria-hidden="true" viewBox="64 64 896 896"><path d="M512 64C264.6 64 64 264.6 64 512s200.6 448 448 448 448-200.6 448-448S759.4 64 512 64zm-32 232c0-4.4 3.6-8 8-8h48c4.4 0 8 3.6 8 8v272c0 4.4-3.6 8-8 8h-48c-4.4 0-8-3.6-8-8V296zm32 440a48.01 48.01 0 010-96 48.01 48.01 0 010 96z"></path></svg>`
+        this.loading_icon = `<svg focusable="false" data-icon="loading" width="1em" height="1em" fill="currentColor" aria-hidden="true" viewBox="0 0 1024 1024"><path d="M988 548c-19.9 0-36-16.1-36-36 0-59.4-11.6-117-34.6-171.3a440.45 440.45 0 00-94.3-139.9 437.71 437.71 0 00-139.9-94.3C629 83.6 571.4 72 512 72c-19.9 0-36-16.1-36-36s16.1-36 36-36c69.1 0 136.2 13.5 199.3 40.3C772.3 66 827 103 874 150c47 47 83.9 101.8 109.7 162.7 26.7 63.1 40.2 130.2 40.2 199.3.1 19.9-16 36-35.9 36z"></path></svg>`
+    }
+
+    async showToast(data){
+        let container = document.querySelector(".easyuse-toast-container");
+        if (!container) {
+            container = document.createElement("div");
+            container.classList.add("easyuse-toast-container");
+            document.body.appendChild(container);
+        }
+        await this.hideToast(data.id);
+        const toastContainer = document.createElement("div");
+        const content = document.createElement("span");
+        content.innerHTML = data.content;
+        toastContainer.appendChild(content);
+        for (let a = 0; a < (data.actions || []).length; a++) {
+            const action = data.actions[a];
+            if (a > 0) {
+                const sep = document.createElement("span");
+                sep.innerHTML = "&nbsp;|&nbsp;";
+                toastContainer.appendChild(sep);
+            }
+            const actionEl = document.createElement("a");
+            actionEl.innerText = action.label;
+            if (action.href) {
+                actionEl.target = "_blank";
+                actionEl.href = action.href;
+            }
+            if (action.callback) {
+                actionEl.onclick = (e) => {
+                    return action.callback(e);
+                };
+            }
+            toastContainer.appendChild(actionEl);
+        }
+        const animContainer = document.createElement("div");
+        animContainer.setAttribute("toast-id", data.id);
+        animContainer.appendChild(toastContainer);
+        container.appendChild(animContainer);
+        await sleep(64);
+        animContainer.style.marginTop = `-${animContainer.offsetHeight}px`;
+        await sleep(64);
+        animContainer.classList.add("-show");
+        if (data.duration) {
+            await sleep(data.duration);
+            this.hideToast(data.id);
+        }
+    }
+    async hideToast(id) {
+        const msg = document.querySelector(`.easyuse-toast-container > [toast-id="${id}"]`);
+        if (msg === null || msg === void 0 ? void 0 : msg.classList.contains("-show")) {
+            msg.classList.remove("-show");
+            await sleep(750);
+        }
+        msg && msg.remove();
+    }
+    async clearAllMessages() {
+        let container = document.querySelector(".easyuse-toast-container");
+        container && (container.innerHTML = "");
+    }
+
+    async copyright(duration = 5000, actions = []) {
+        this.showToast({
+            id: `toast-info`,
+            content: `${this.info_icon} ${$t('Workflow created by')} <a href="https://github.com/yolain/">Yolain</a> , ${$t('Watch more video content')} <a href="https://space.bilibili.com/1840885116">B站乱乱呀</a>`,
+            duration,
+            actions
+        });
+    }
+    async info(content, duration = 3000, actions = []) {
+        this.showToast({
+            id: `toast-info`,
+            content: `${this.info_icon} ${content}`,
+            duration,
+            actions
+        });
+    }
+    async success(content, duration = 3000, actions = []) {
+        this.showToast({
+            id: `toast-success`,
+            content: `${this.success_icon} ${content}`,
+            duration,
+            actions
+        });
+    }
+    async error(content, duration = 3000, actions = []) {
+        this.showToast({
+            id: `toast-error`,
+            content: `${this.error_icon} ${content}`,
+            duration,
+            actions
+        });
+    }
+    async warn(content, duration = 3000, actions = []) {
+        this.showToast({
+            id: `toast-warn`,
+            content: `${this.warn_icon} ${content}`,
+            duration,
+            actions
+        });
+    }
+    async showLoading(content, duration = 0, actions = []) {
+        this.showToast({
+            id: `toast-loading`,
+            content: `${this.loading_icon} ${content}`,
+            duration,
+            actions
+        });
+    }
+
+    async hideLoading() {
+        this.hideToast("toast-loading");
+    }
+
+}
+
+export const toast = new Toast();

web/js/common/utils.js

@@ -1,3 +1,10 @@
+export function sleep(ms = 100, value) {
+    return new Promise((resolve) => {
+        setTimeout(() => {
+            resolve(value);
+        }, ms);
+    });
+}
 export function addPreconnect(href, crossorigin=false){
     const preconnect = document.createElement("link");
     preconnect.rel = 'preconnect'
@@ -5,7 +12,6 @@ export function addPreconnect(href, crossorigin=false){
     if(crossorigin) preconnect.crossorigin = ''
     document.head.appendChild(preconnect);
 }
-
 export function addCss(href, base=true) {
     const link = document.createElement("link");
     link.rel = "stylesheet";
@@ -45,4 +51,130 @@ export function spliceExtension(fileName){
 }
 export function getExtension(fileName){
    return fileName.substring(fileName.lastIndexOf('.') + 1)
+}
+
+export function formatTime(time, format) {
+  time = typeof (time) === "number" ? time : (time instanceof Date ? time.getTime() : parseInt(time));
+  if (isNaN(time)) return null;
+  if (typeof (format) !== 'string' || !format) format = 'yyyy-MM-dd hh:mm:ss';
+  let _time = new Date(time);
+  time = _time.toString().split(/[\s\:]/g).slice(0, -2);
+  time[1] = ['01', '02', '03', '04', '05', '06', '07', '08', '09', '10', '11', '12'][_time.getMonth()];
+  let _mapping = {
+    MM: 1,
+    dd: 2,
+    yyyy: 3,
+    hh: 4,
+    mm: 5,
+    ss: 6
+  };
+  return format.replace(/([Mmdhs]|y{2})\1/g, (key) => time[_mapping[key]]);
+}
+
+
+let origProps = {};
+export const findWidgetByName = (node, name) => node.widgets.find((w) => w.name === name);
+
+export const doesInputWithNameExist = (node, name) => node.inputs ? node.inputs.some((input) => input.name === name) : false;
+
+export function updateNodeHeight(node) {node.setSize([node.size[0], node.computeSize()[1]]);}
+
+export function toggleWidget(node, widget, show = false, suffix = "") {
+	if (!widget || doesInputWithNameExist(node, widget.name)) return;
+	if (!origProps[widget.name]) {
+		origProps[widget.name] = { origType: widget.type, origComputeSize: widget.computeSize };
+	}
+	const origSize = node.size;
+
+	widget.type = show ? origProps[widget.name].origType : "easyHidden" + suffix;
+	widget.computeSize = show ? origProps[widget.name].origComputeSize : () => [0, -4];
+
+	widget.linkedWidgets?.forEach(w => toggleWidget(node, w, ":" + widget.name, show));
+
+	const height = show ? Math.max(node.computeSize()[1], origSize[1]) : node.size[1];
+	node.setSize([node.size[0], height]);
+}
+
+export function isLocalNetwork(ip) {
+  const localNetworkRanges = [
+    '192.168.',
+    '10.',
+    '127.',
+    /^172\.((1[6-9]|2[0-9]|3[0-1])\.)/
+  ];
+
+  return localNetworkRanges.some(range => {
+    if (typeof range === 'string') {
+      return ip.startsWith(range);
+    } else {
+      return range.test(ip);
+    }
+  });
+}
+
+
+/**
+* accAdd 高精度加法
+* @since 1.0.10
+* @param {Number} arg1
+* @param {Number} arg2
+* @return {Number}
+*/
+export function accAdd(arg1, arg2) {
+  let r1, r2, s1, s2,max;
+  s1 = typeof arg1 == 'string' ? arg1 : arg1.toString()
+  s2 = typeof arg2 == 'string' ? arg2 : arg2.toString()
+  try { r1 = s1.split(".")[1].length } catch (e) { r1 = 0 }
+  try { r2 = s2.split(".")[1].length } catch (e) { r2 = 0 }
+  max = Math.pow(10, Math.max(r1, r2))
+  return (arg1 * max + arg2 * max) / max
+}
+/**
+ * accSub 高精度减法
+ * @since 1.0.10
+ * @param {Number} arg1
+ * @param {Number} arg2
+ * @return {Number}
+ */
+export function accSub(arg1, arg2) {
+  let r1, r2, max, min,s1,s2;
+  s1 = typeof arg1 == 'string' ? arg1 : arg1.toString()
+  s2 = typeof arg2 == 'string' ? arg2 : arg2.toString()
+  try { r1 = s1.split(".")[1].length } catch (e) { r1 = 0 }
+  try { r2 = s2.split(".")[1].length } catch (e) { r2 = 0 }
+  max = Math.pow(10, Math.max(r1, r2));
+  //动态控制精度长度
+  min = (r1 >= r2) ? r1 : r2;
+  return ((arg1 * max - arg2 * max) / max).toFixed(min)
+}
+/**
+ * accMul 高精度乘法
+ * @since 1.0.10
+ * @param {Number} arg1
+ * @param {Number} arg2
+ * @return {Number}
+ */
+export function accMul(arg1, arg2) {
+  let max = 0, s1 =  typeof arg1 == 'string' ? arg1 : arg1.toString(), s2 = typeof arg2 == 'string' ? arg2 : arg2.toString();
+  try { max += s1.split(".")[1].length } catch (e) { }
+  try { max += s2.split(".")[1].length } catch (e) { }
+  return Number(s1.replace(".", "")) * Number(s2.replace(".", "")) / Math.pow(10, max)
+}
+/**
+ * accDiv 高精度除法
+ * @since 1.0.10
+ * @param {Number} arg1
+ * @param {Number} arg2
+ * @return {Number}
+ */
+export function accDiv(arg1, arg2) {
+  let t1 = 0, t2 = 0, r1, r2,s1 =  typeof arg1 == 'string' ? arg1 : arg1.toString(), s2 = typeof arg2 == 'string' ? arg2 : arg2.toString();
+  try { t1 = s1.toString().split(".")[1].length } catch (e) { }
+  try { t2 = s2.toString().split(".")[1].length } catch (e) { }
+  r1 = Number(s1.toString().replace(".", ""))
+  r2 = Number(s2.toString().replace(".", ""))
+  return (r1 / r2) * Math.pow(10, t2 - t1)
+}
+Number.prototype.div = function (arg) {
+  return accDiv(this, arg);
 }

web/js/easy/easy.js

@@ -1,9 +1,394 @@
-import { app } from "/scripts/app.js";
-import {deepEqual,addCss} from "../common/utils.js";
+import { api } from "../../../../scripts/api.js";
+import { app } from "../../../../scripts/app.js";
+import {deepEqual, addCss, isLocalNetwork} from "../common/utils.js";
+import {quesitonIcon, rocketIcon, groupIcon, rebootIcon} from "../common/icon.js";
 import {$t} from '../common/i18n.js';
+import {toast} from "../common/toast.js";
+import {$el, ComfyDialog} from "../../../../scripts/ui.js";
+
 
 addCss('css/index.css')
 
+api.addEventListener("easyuse-toast",event=>{
+    const content = event.detail.content
+    const type = event.detail.type
+    const duration = event.detail.duration
+    if(!type){
+        toast.info(content, duration)
+    }
+    else{
+      toast.showToast({
+        id: `toast-${type}`,
+        content: `${toast[type+"_icon"]} ${content}`,
+        duration: duration || 3000,
+      })
+    }
+})
+
+
+let draggerEl = null
+let isGroupMapcanMove = true
+function createGroupMap(){
+    let groups = app.canvas.graph._groups
+    let nodes = app.canvas.graph._nodes
+    let old_nodes = groups.length
+    let div =
+        document.querySelector('#easyuse_groups_map') ||
+        document.createElement('div')
+    div.id = 'easyuse_groups_map'
+    div.innerHTML = ''
+    let btn = document.createElement('div')
+    btn.style = `display: flex;
+        width: calc(100% - 8px);
+        justify-content: space-between;
+        align-items: center;
+        padding: 0 6px;
+        height: 44px;`
+    let hideBtn = document.createElement('button')
+    let textB = document.createElement('p')
+    btn.appendChild(textB)
+    btn.appendChild(hideBtn)
+    textB.style.fontSize = '11px'
+    textB.innerHTML =  `<b>${$t('Groups Map')} (EasyUse)</b>`
+    hideBtn.style = `float: right;color: var(--input-text);border-radius:6px;font-size:9px;
+        background-color: var(--comfy-input-bg); border: 1px solid var(--border-color);cursor: pointer;padding: 5px;aspect-ratio: 1 / 1;`
+    hideBtn.addEventListener('click', () => {
+        div.style.display = 'none'
+    })
+    hideBtn.innerText = '❌'
+    div.appendChild(btn)
+
+    div.addEventListener('mousedown', function (e) {
+        var startX = e.clientX
+        var startY = e.clientY
+        var offsetX = div.offsetLeft
+        var offsetY = div.offsetTop
+
+        function moveBox (e) {
+          var newX = e.clientX
+          var newY = e.clientY
+          var deltaX = newX - startX
+          var deltaY = newY - startY
+          div.style.left = offsetX + deltaX + 'px'
+          div.style.top = offsetY + deltaY + 'px'
+        }
+
+        function stopMoving () {
+          document.removeEventListener('mousemove', moveBox)
+          document.removeEventListener('mouseup', stopMoving)
+        }
+
+        if(isGroupMapcanMove){
+            document.addEventListener('mousemove', moveBox)
+            document.addEventListener('mouseup', stopMoving)
+        }
+    })
+
+    function updateGroups(groups, groupsDiv, autoSortDiv){
+        if(groups.length>0){
+            autoSortDiv.style.display = 'block'
+        }else autoSortDiv.style.display = 'none'
+        for (let index in groups) {
+            const group = groups[index]
+            const title = group.title
+            const show_text = $t('Always')
+            const hide_text = $t('Bypass')
+            const mute_text = $t('Never')
+            let group_item = document.createElement('div')
+            let group_item_style = `justify-content: space-between;display:flex;background-color: var(--comfy-input-bg);border-radius: 5px;border:1px solid var(--border-color);margin-top:5px;`
+            group_item.addEventListener("mouseover",event=>{
+                event.preventDefault()
+                group_item.style = group_item_style + "filter:brightness(1.2);"
+            })
+            group_item.addEventListener("mouseleave",event=>{
+                event.preventDefault()
+                group_item.style = group_item_style + "filter:brightness(1);"
+            })
+            group_item.addEventListener("dragstart",e=>{
+                draggerEl = e.currentTarget;
+                e.currentTarget.style.opacity = "0.6";
+                e.currentTarget.style.border = "1px dashed yellow";
+                e.dataTransfer.effectAllowed = 'move';
+                e.dataTransfer.setDragImage(emptyImg, 0, 0);
+            })
+            group_item.addEventListener("dragend",e=>{
+                e.target.style.opacity = "1";
+                e.currentTarget.style.border = "1px dashed transparent";
+                e.currentTarget.removeAttribute("draggable");
+                document.querySelectorAll('.easyuse-group-item').forEach((el,i) => {
+                    var prev_i = el.dataset.id;
+                    if (el == draggerEl && prev_i != i ) {
+                        groups.splice(i, 0, groups.splice(prev_i, 1)[0]);
+                    }
+                    el.dataset.id = i;
+                });
+                 isGroupMapcanMove = true
+            })
+            group_item.addEventListener("dragover",e=>{
+                e.preventDefault();
+                if (e.currentTarget == draggerEl) return;
+                let rect = e.currentTarget.getBoundingClientRect();
+                if (e.clientY > rect.top + rect.height / 2) {
+                    e.currentTarget.parentNode.insertBefore(draggerEl, e.currentTarget.nextSibling);
+                } else {
+                    e.currentTarget.parentNode.insertBefore(draggerEl, e.currentTarget);
+                }
+                isGroupMapcanMove = true
+            })
+
+
+            group_item.setAttribute('data-id',index)
+            group_item.className = 'easyuse-group-item'
+            group_item.style = group_item_style
+            // 标题
+            let text_group_title = document.createElement('div')
+            text_group_title.style = `flex:1;font-size:12px;color:var(--input-text);padding:4px;white-space: nowrap;overflow: hidden;text-overflow: ellipsis;cursor:pointer`
+            text_group_title.innerHTML = `${title}`
+            text_group_title.addEventListener('mousedown',e=>{
+                isGroupMapcanMove = false
+                e.currentTarget.parentNode.draggable = 'true';
+            })
+            text_group_title.addEventListener('mouseleave',e=>{
+                setTimeout(_=>{
+                    isGroupMapcanMove = true
+                },150)
+            })
+            group_item.append(text_group_title)
+            // 按钮组
+            let buttons = document.createElement('div')
+            group.recomputeInsideNodes();
+            const nodesInGroup = group._nodes;
+            let isGroupShow = nodesInGroup && nodesInGroup.length>0 && nodesInGroup[0].mode == 0
+            let isGroupMute = nodesInGroup && nodesInGroup.length>0 && nodesInGroup[0].mode == 2
+            let go_btn = document.createElement('button')
+            go_btn.style = "margin-right:6px;cursor:pointer;font-size:10px;padding:2px 4px;color:var(--input-text);background-color: var(--comfy-input-bg);border: 1px solid var(--border-color);border-radius:4px;"
+            go_btn.innerText = "Go"
+            go_btn.addEventListener('click', () => {
+                app.canvas.ds.offset[0] =  -group.pos[0] - group.size[0] * 0.5 + (app.canvas.canvas.width * 0.5) / app.canvas.ds.scale;
+                app.canvas.ds.offset[1] = -group.pos[1] - group.size[1] * 0.5 + (app.canvas.canvas.height * 0.5) / app.canvas.ds.scale;
+                app.canvas.setDirty(true, true);
+                app.canvas.setZoom(1)
+            })
+            buttons.append(go_btn)
+            let see_btn = document.createElement('button')
+            let defaultStyle = `cursor:pointer;font-size:10px;;padding:2px;border: 1px solid var(--border-color);border-radius:4px;width:36px;`
+            see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
+            see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
+            let pressTimer
+            let firstTime =0, lastTime =0
+            let isHolding = false
+            see_btn.addEventListener('click', () => {
+                if(isHolding){
+                    isHolding = false
+                    return
+                }
+                for (const node of nodesInGroup) {
+                    node.mode = isGroupShow ? 4 : 0;
+                    node.graph.change();
+                }
+                isGroupShow = nodesInGroup[0].mode == 0 ? true : false
+                isGroupMute = nodesInGroup[0].mode == 2 ? true : false
+                see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
+                see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
+            })
+            see_btn.addEventListener('mousedown', () => {
+                firstTime = new Date().getTime();
+                clearTimeout(pressTimer);
+                pressTimer = setTimeout(_=>{
+                    for (const node of nodesInGroup) {
+                        node.mode = isGroupMute ? 0 : 2;
+                        node.graph.change();
+                    }
+                    isGroupShow = nodesInGroup[0].mode == 0 ? true : false
+                    isGroupMute = nodesInGroup[0].mode == 2 ? true : false
+                    see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
+                    see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
+                },500)
+            })
+            see_btn.addEventListener('mouseup', () => {
+                lastTime = new Date().getTime();
+                if(lastTime - firstTime > 500) isHolding = true
+                clearTimeout(pressTimer);
+            })
+            buttons.append(see_btn)
+            group_item.append(buttons)
+
+            groupsDiv.append(group_item)
+        }
+
+    }
+
+    let groupsDiv =  document.createElement('div')
+    groupsDiv.id = 'easyuse-groups-items'
+    groupsDiv.style = `overflow-y: auto;max-height: 400px;height:100%;width: 100%;`
+
+    let autoSortDiv = document.createElement('button')
+    autoSortDiv.style = `cursor:pointer;font-size:10px;padding:2px 4px;color:var(--input-text);background-color: var(--comfy-input-bg);border: 1px solid var(--border-color);border-radius:4px;`
+    autoSortDiv.innerText =  $t('Auto Sorting')
+    autoSortDiv.addEventListener('click',e=>{
+        e.preventDefault()
+        groupsDiv.innerHTML = ``
+        let new_groups = groups.sort((a,b)=> a['pos'][0] - b['pos'][0]).sort((a,b)=> a['pos'][1] - b['pos'][1])
+        updateGroups(new_groups, groupsDiv, autoSortDiv)
+    })
+
+    updateGroups(groups, groupsDiv, autoSortDiv)
+
+    div.appendChild(groupsDiv)
+
+    let remarkDiv =  document.createElement('p')
+    remarkDiv.style = `text-align:center; font-size:10px; padding:0 10px;color:var(--descrip-text)`
+    remarkDiv.innerText =  $t('Toggle `Show/Hide` can set mode of group, LongPress can set group nodes to never')
+    div.appendChild(groupsDiv)
+    div.appendChild(remarkDiv)
+    div.appendChild(autoSortDiv)
+
+    let graphDiv = document.getElementById("graph-canvas")
+    graphDiv.addEventListener('mouseover', async () => {
+      groupsDiv.innerHTML = ``
+      let new_groups = app.canvas.graph._groups
+      updateGroups(new_groups, groupsDiv, autoSortDiv)
+      old_nodes = nodes
+    })
+
+    if (!document.querySelector('#easyuse_groups_map')){
+        document.body.appendChild(div)
+    }else{
+        div.style.display = 'flex'
+    }
+
+}
+
+async function cleanup(){
+    try {
+       const {Running, Pending} = await api.getQueue()
+       if(Running.length>0 || Pending.length>0){
+           toast.error($t("Clean Failed")+ ":"+ $t("Please stop all running tasks before cleaning GPU"))
+           return
+       }
+        api.fetchApi("/easyuse/cleangpu",{
+            method:"POST"
+        }).then(res=>{
+            if(res.status == 200){
+                toast.success($t("Clean SuccessFully"))
+            }else{
+                toast.error($t("Clean Failed"))
+            }
+        })
+
+    } catch (exception) {}
+}
+
+
+let guideDialog = null
+let isDownloading = false
+function download_model(url,local_dir){
+    if(isDownloading || !url || !local_dir) return
+    isDownloading = true
+    let body =  new FormData();
+    body.append('url', url);
+    body.append('local_dir', local_dir);
+    api.fetchApi("/easyuse/model/download",{
+        method:"POST",
+        body
+    }).then(res=>{
+        if(res.status == 200){
+            toast.success($t("Download SuccessFully"))
+        }else{
+            toast.error($t("Download Failed"))
+        }
+        isDownloading = false
+    })
+
+}
+class GuideDialog extends ComfyDialog {
+    show(note, need_models){
+
+        let modelsDiv = null
+        if(need_models?.length>0){
+            let tbody = []
+
+            for(let i=0;i<need_models.length;i++){
+                tbody.push($el('tr',[
+                    $el('td',{innerHTML:need_models[i].title || need_models[i].name || ''}),
+                    $el('td',[
+                        need_models[i]['download_url'] ? $el('a',{onclick:_=>download_model(need_models[i]['download_url'],need_models[i]['local_dir']), target:"_blank", textContent:$t('Download Model')}) : '',
+                         need_models[i]['source_url'] ? $el('a',{href:need_models[i]['source_url'], target:"_blank", textContent:$t('Source Url')}) : '',
+                    ]),
+                    $el('td',{innerHTML:need_models[i].description || ''}),
+                ]))
+            }
+            modelsDiv = $el('div.easyuse-guide-dialog-models.markdown-body',[
+                $el('h3',{textContent:$t('Models Required')}),
+                $el('table',{cellpadding:0,cellspacing:0},[
+                    $el('thead',[
+                        $el('tr',[
+                            $el('th',{innerHTML:$t('ModelName')}),
+                            $el('th',{innerHTML:$t('Details')}),
+                            $el('th',{innerHTML:$t('Description')}),
+                        ])
+                    ]),
+                    $el('tbody',tbody)
+                ])
+            ])
+        }
+
+        super.show(
+           $el('div.easyuse-guide-dialog',[
+               $el('div.easyuse-guide-dialog-header',[
+                     $el('div.easyuse-guide-dialog-title',{
+                        innerHTML:$t('Workflow Guide')
+                     }),
+                     $el('div.easyuse-guide-dialog-remark',{
+                        innerHTML:`${$t('Workflow created by')} <a href="https://github.com/yolain/" target="_blank">Yolain</a> , ${$t('Watch more video content')} <a href="https://space.bilibili.com/1840885116" target="_blank">B站乱乱呀</a>`
+                     })
+               ]),
+               $el('div.easyuse-guide-dialog-content.markdown-body',[
+                   $el('div.easyuse-guide-dialog-note',{
+                       innerHTML:note
+                   }),
+                   modelsDiv
+               ])
+           ])
+        )
+    }
+
+    close(){
+        guideDialog = null
+        super.close()
+    }
+}
+
+
+const toolBarId = "Comfy.EasyUse.toolBar"
+
+let enableToolBar = true
+let disableRenderInfo = localStorage['Comfy.Settings.Comfy.EasyUse.disableRenderInfo'] ? true : false
+export function addToolBar(app) {
+	app.ui.settings.addSetting({
+		id: toolBarId,
+		name: $t("Enable tool bar fixed on the left-bottom (ComfyUI-Easy-Use)"),
+		type: "boolean",
+		defaultValue: enableToolBar,
+		onChange(value) {
+			enableToolBar = !!value;
+            if(enableToolBar){
+                showToolBar()
+            }else hideToolBar()
+		},
+	});
+}
+const getEnableToolBar = _ => app.ui.settings.getSettingValue(toolBarId, true)
+
+let note = null
+let toolbar = null
+function showToolBar(){
+    toolbar.style.display = 'flex'
+}
+function hideToolBar(){
+    toolbar.style.display = 'none'
+}
+
 app.registerExtension({
     name: "comfy.easyUse",
     init() {
@@ -11,249 +396,106 @@ app.registerExtension({
         const getCanvasMenuOptions = LGraphCanvas.prototype.getCanvasMenuOptions;
         LGraphCanvas.prototype.getCanvasMenuOptions = function () {
             const options = getCanvasMenuOptions.apply(this, arguments);
-            let draggerEl = null
-            let isGroupMapcanMove = true
-            let old_groups = []
             let emptyImg = new Image()
             emptyImg.src = "data:image/gif;base64,R0lGODlhAQABAIAAAAUEBAAAACwAAAAAAQABAAACAkQBADs=";
 
             options.push(null,
+                // Groups Map
                 {
-                    content: '📜 '+ $t('Groups Map (EasyUse)'),
+                    content: groupIcon.replace('currentColor','var(--warning-color)') + ' '+ $t('Groups Map') + ' (EasyUse)',
                     callback: async() => {
-                        let groups = app.canvas.graph._groups
-                        let nodes = app.canvas.graph._nodes
-                        let old_nodes = groups.length
-                        let div =
-                            document.querySelector('#easyuse_groups_map') ||
-                            document.createElement('div')
-                        div.id = 'easyuse_groups_map'
-                        div.innerHTML = ''
-                        let btn = document.createElement('div')
-                        btn.style = `display: flex;
-                            width: calc(100% - 8px);
-                            justify-content: space-between;
-                            align-items: center;
-                            padding: 0 6px;
-                            height: 44px;`
-                        let hideBtn = document.createElement('button')
-                        let textB = document.createElement('p')
-                        btn.appendChild(textB)
-                        btn.appendChild(hideBtn)
-                        textB.style.fontSize = '11px'
-                        textB.innerHTML =  `<b>${$t('Groups Map (EasyUse)')}</b>`
-                        hideBtn.style = `float: right;color: var(--input-text);border-radius:6px;font-size:9px;
-                            background-color: var(--comfy-input-bg); border: 1px solid var(--border-color);cursor: pointer;padding: 5px;aspect-ratio: 1 / 1;`
-                        hideBtn.addEventListener('click', () => {
-                            div.style.display = 'none'
-                        })
-                        hideBtn.innerText = '❌'
-                        div.appendChild(btn)
-
-                        div.addEventListener('mousedown', function (e) {
-                            var startX = e.clientX
-                            var startY = e.clientY
-                            var offsetX = div.offsetLeft
-                            var offsetY = div.offsetTop
-
-                            function moveBox (e) {
-                              var newX = e.clientX
-                              var newY = e.clientY
-                              var deltaX = newX - startX
-                              var deltaY = newY - startY
-                              div.style.left = offsetX + deltaX + 'px'
-                              div.style.top = offsetY + deltaY + 'px'
-                            }
-
-                            function stopMoving () {
-                              document.removeEventListener('mousemove', moveBox)
-                              document.removeEventListener('mouseup', stopMoving)
-                            }
-
-                            if(isGroupMapcanMove){
-                                document.addEventListener('mousemove', moveBox)
-                                document.addEventListener('mouseup', stopMoving)
-                            }
-                        })
-
-                        function updateGroups(groups, groupsDiv, autoSortDiv){
-                            if(groups.length>0){
-                                autoSortDiv.style.display = 'block'
-                            }else autoSortDiv.style.display = 'none'
-                            for (let index in groups) {
-                                const group = groups[index]
-                                const title = group.title
-                                const show_text = $t('Always')
-                                const hide_text = $t('Bypass')
-                                const mute_text = $t('Never')
-                                let group_item = document.createElement('div')
-                                let group_item_style = `justify-content: space-between;display:flex;background-color: var(--comfy-input-bg);border-radius: 5px;border:1px solid var(--border-color);margin-top:5px;`
-                                group_item.addEventListener("mouseover",event=>{
-                                    event.preventDefault()
-                                    group_item.style = group_item_style + "filter:brightness(1.2);"
-                                })
-                                group_item.addEventListener("mouseleave",event=>{
-                                    event.preventDefault()
-                                    group_item.style = group_item_style + "filter:brightness(1);"
-                                })
-                                group_item.addEventListener("dragstart",e=>{
-                                    draggerEl = e.currentTarget;
-                                    e.currentTarget.style.opacity = "0.6";
-									e.currentTarget.style.border = "1px dashed yellow";
-									e.dataTransfer.effectAllowed = 'move';
-									e.dataTransfer.setDragImage(emptyImg, 0, 0);
-                                })
-                                group_item.addEventListener("dragend",e=>{
-                                    e.target.style.opacity = "1";
-                                    e.currentTarget.style.border = "1px dashed transparent";
-                                    e.currentTarget.removeAttribute("draggable");
-									document.querySelectorAll('.easyuse-group-item').forEach((el,i) => {
-										var prev_i = el.dataset.id;
-										if (el == draggerEl && prev_i != i ) {
-											groups.splice(i, 0, groups.splice(prev_i, 1)[0]);
-										}
-										el.dataset.id = i;
-									});
-                                     isGroupMapcanMove = true
-                                })
-                                group_item.addEventListener("dragover",e=>{
-                                    e.preventDefault();
-									if (e.currentTarget == draggerEl) return;
-									let rect = e.currentTarget.getBoundingClientRect();
-									if (e.clientY > rect.top + rect.height / 2) {
-										e.currentTarget.parentNode.insertBefore(draggerEl, e.currentTarget.nextSibling);
-									} else {
-										e.currentTarget.parentNode.insertBefore(draggerEl, e.currentTarget);
-									}
-                                    isGroupMapcanMove = true
-                                })
-
-
-                                group_item.setAttribute('data-id',index)
-                                group_item.className = 'easyuse-group-item'
-                                group_item.style = group_item_style
-                                // 标题
-                                let text_group_title = document.createElement('div')
-                                text_group_title.style = `flex:1;font-size:12px;color:var(--input-text);padding:4px;white-space: nowrap;overflow: hidden;text-overflow: ellipsis;cursor:pointer`
-                                text_group_title.innerHTML = `${title}`
-                                text_group_title.addEventListener('mousedown',e=>{
-                                    isGroupMapcanMove = false
-                                    e.currentTarget.parentNode.draggable = 'true';
-                                })
-                                text_group_title.addEventListener('mouseleave',e=>{
-                                    setTimeout(_=>{
-                                        isGroupMapcanMove = true
-                                    },150)
-                                })
-                                group_item.append(text_group_title)
-                                // 按钮组
-                                let buttons = document.createElement('div')
-                                group.recomputeInsideNodes();
-                                const nodesInGroup = group._nodes;
-                                let isGroupShow = nodesInGroup && nodesInGroup.length>0 && nodesInGroup[0].mode == 0
-                                let isGroupMute = nodesInGroup && nodesInGroup.length>0 && nodesInGroup[0].mode == 2
-                                let go_btn = document.createElement('button')
-                                go_btn.style = "margin-right:6px;cursor:pointer;font-size:10px;padding:2px 4px;color:var(--input-text);background-color: var(--comfy-input-bg);border: 1px solid var(--border-color);border-radius:4px;"
-                                go_btn.innerText = "Go"
-                                go_btn.addEventListener('click', () => {
-                                    app.canvas.ds.offset[0] =  -group.pos[0] - group.size[0] * 0.5 + (app.canvas.canvas.width * 0.5) / app.canvas.ds.scale;
-                                    app.canvas.ds.offset[1] = -group.pos[1] - group.size[1] * 0.5 + (app.canvas.canvas.height * 0.5) / app.canvas.ds.scale;
-                                    app.canvas.setDirty(true, true);
-                                    app.canvas.setZoom(1)
-                                })
-                                buttons.append(go_btn)
-                                let see_btn = document.createElement('button')
-                                let defaultStyle = `cursor:pointer;font-size:10px;;padding:2px;border: 1px solid var(--border-color);border-radius:4px;width:36px;`
-                                see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
-                                see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
-                                let pressTimer
-                                let firstTime =0, lastTime =0
-                                let isHolding = false
-                                see_btn.addEventListener('click', () => {
-                                    if(isHolding){
-                                        isHolding = false
-                                        return
-                                    }
-                                    for (const node of nodesInGroup) {
-                                        node.mode = isGroupShow ? 4 : 0;
-                                        node.graph.change();
-                                    }
-                                    isGroupShow = nodesInGroup[0].mode == 0 ? true : false
-                                    isGroupMute = nodesInGroup[0].mode == 2 ? true : false
-                                    see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
-                                    see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
-                                })
-                                see_btn.addEventListener('mousedown', () => {
-                                    firstTime = new Date().getTime();
-                                    clearTimeout(pressTimer);
-                                    pressTimer = setTimeout(_=>{
-                                        for (const node of nodesInGroup) {
-                                            node.mode = isGroupMute ? 0 : 2;
-                                            node.graph.change();
-                                        }
-                                        isGroupShow = nodesInGroup[0].mode == 0 ? true : false
-                                        isGroupMute = nodesInGroup[0].mode == 2 ? true : false
-                                        see_btn.style = isGroupMute ? `background-color:var(--error-text);color:var(--input-text);` + defaultStyle : (isGroupShow ? `background-color:#006691;color:var(--input-text);` + defaultStyle : `background-color: var(--comfy-input-bg);color:var(--descrip-text);` + defaultStyle)
-                                        see_btn.innerText = isGroupMute ? mute_text : (isGroupShow ? show_text : hide_text)
-                                    },500)
-                                })
-                                see_btn.addEventListener('mouseup', () => {
-                                    lastTime = new Date().getTime();
-                                    if(lastTime - firstTime > 500) isHolding = true
-                                    clearTimeout(pressTimer);
-                                })
-                                buttons.append(see_btn)
-                                group_item.append(buttons)
-
-                                groupsDiv.append(group_item)
-                            }
-
-                        }
-
-                        let groupsDiv =  document.createElement('div')
-                        groupsDiv.id = 'easyuse-groups-items'
-                        groupsDiv.style = `overflow-y: auto;max-height: 400px;height:100%;width: 100%;`
-
-                        let autoSortDiv = document.createElement('button')
-                        autoSortDiv.style = `cursor:pointer;font-size:10px;padding:2px 4px;color:var(--input-text);background-color: var(--comfy-input-bg);border: 1px solid var(--border-color);border-radius:4px;`
-                        autoSortDiv.innerText =  $t('Auto Sorting')
-                        autoSortDiv.addEventListener('click',e=>{
-                            e.preventDefault()
-                            groupsDiv.innerHTML = ``
-                            let new_groups = groups.sort((a,b)=> a['pos'][0] - b['pos'][0]).sort((a,b)=> a['pos'][1] - b['pos'][1])
-                            updateGroups(new_groups, groupsDiv, autoSortDiv)
-                        })
-
-                        updateGroups(groups, groupsDiv, autoSortDiv)
-
-                        div.appendChild(groupsDiv)
-
-                        let remarkDiv =  document.createElement('p')
-                        remarkDiv.style = `text-align:center; font-size:10px; padding:0 10px;color:var(--descrip-text)`
-                        remarkDiv.innerText =  $t('Toggle `Show/Hide` can set mode of group, LongPress can set group nodes to never')
-                        div.appendChild(groupsDiv)
-                        div.appendChild(remarkDiv)
-                        div.appendChild(autoSortDiv)
-
-                        let graphDiv = document.getElementById("graph-canvas")
-                        graphDiv.addEventListener('mouseover', async () => {
-                          groupsDiv.innerHTML = ``
-                          let new_groups = app.canvas.graph._groups
-                          updateGroups(new_groups, groupsDiv, autoSortDiv)
-                          old_nodes = nodes
-                        })
-
-                        if (!document.querySelector('#easyuse_groups_map')){
-                            document.body.appendChild(div)
-                        }else{
-                            div.style.display = 'flex'
-                        }
-
+                        createGroupMap()
                     }
                 },
+                // Force clean ComfyUI GPU Used 强制卸载模型GPU占用
+                {
+                    content: rocketIcon.replace('currentColor','var(--theme-color-light)') + ' '+ $t('Cleanup Of GPU Usage') + ' (EasyUse)',
+                    callback: async() =>{
+                       await cleanup()
+                    }
+                },
+                // Only show the reboot option if the server is running on a local network 仅在本地或局域网环境可重启服务
+                isLocalNetwork(window.location.host) ? {
+                    content: rebootIcon.replace('currentColor','var(--error-color)') + ' '+ $t('Reboot ComfyUI') + ' (EasyUse)',
+                    callback: _ =>{
+                        if (confirm($t("Are you sure you'd like to reboot the server?"))){
+                            try {
+                                api.fetchApi("/easyuse/reboot");
+                            } catch (exception) {}
+                        }
+                    }
+                } : null,
             );
             return options;
         };
+
+        let renderInfoEvent = LGraphCanvas.prototype.renderInfo
+        if(disableRenderInfo){
+            LGraphCanvas.prototype.renderInfo = function (ctx, x, y) {}
+        }
+
+        if(!toolbar){
+            toolbar = $el('div.easyuse-toolbar',[
+                $el('div.easyuse-toolbar-item',{
+                    onclick:_=>{
+                        createGroupMap()
+                    }
+                },[
+                    $el('div.easyuse-toolbar-icon.group', {innerHTML:groupIcon}),
+                    $el('div.easyuse-toolbar-tips',$t('Groups Map'))
+                ]),
+                $el('div.easyuse-toolbar-item',{
+                    onclick:async()=>{
+                        await cleanup()
+                    }
+                },[
+                    $el('div.easyuse-toolbar-icon.rocket',{innerHTML:rocketIcon}),
+                    $el('div.easyuse-toolbar-tips',$t('Cleanup Of GPU Usage'))
+                ]),
+            ])
+            if(disableRenderInfo){
+                toolbar.classList.add('disable-render-info')
+            }else{
+                toolbar.classList.remove('disable-render-info')
+            }
+            document.body.appendChild(toolbar)
+        }
+
+        // rewrite handleFile
+        let loadGraphDataEvent = app.loadGraphData
+        app.loadGraphData = async function (data, clean=true) {
+            // if(data?.extra?.cpr){
+            //     toast.copyright()
+            // }
+            if(data?.extra?.note){
+                 if(guideDialog) {
+                     guideDialog.close()
+                     guideDialog = null
+                 }
+                 if(note && toolbar) toolbar.removeChild(note)
+                 note = $el('div.easyuse-toolbar-item',{
+                    onclick:async()=>{
+                        if(guideDialog) return
+                        guideDialog = new GuideDialog()
+                        const need_models = data.extra?.need_models || null
+                        guideDialog.show(data.extra.note, need_models)
+                    }
+                },[
+                    $el('div.easyuse-toolbar-icon.question',{innerHTML:quesitonIcon}),
+                    $el('div.easyuse-toolbar-tips',$t('Workflow Guide'))
+                ])
+                if(toolbar) toolbar.insertBefore(note, toolbar.firstChild)
+            }
+            else{
+                if(note) {
+                    toolbar.removeChild(note)
+                    note = null
+                }
+            }
+            return loadGraphDataEvent.apply(this, [...arguments])
+        }
+
+        addToolBar(app)
     },
     beforeRegisterNodeDef(nodeType, nodeData, app) {
         if (nodeData.name.startsWith("easy")) {

web/js/easy/easyAccount.js

@@ -0,0 +1,283 @@
+import { app } from "../../../../scripts/app.js";
+import { api } from "../../../../scripts/api.js";
+import { $el, ComfyDialog } from "../../../../scripts/ui.js";
+import { $t } from '../common/i18n.js'
+import { toast } from "../common/toast.js";
+import {sleep, accSub} from "../common/utils.js";
+
+let api_keys = []
+let api_current = 0
+let user_info = {}
+
+const api_cost = {
+    'sd3': 6.5,
+    'sd3-turbo': 4,
+}
+
+class AccountDialog extends ComfyDialog {
+    constructor() {
+		super();
+        this.lists = []
+        this.dialog_div = null
+        this.user_div = null
+	}
+
+    addItem(index, user_div){
+        return $el('div.easyuse-account-dialog-item',[
+          $el('input',{type:'text',placeholder:'Enter name',oninput: e=>{
+              const dataIndex = Array.prototype.indexOf.call(this.dialog_div.querySelectorAll('.easyuse-account-dialog-item'), e.target.parentNode)
+              api_keys[dataIndex]['name'] = e.target.value
+          },value:api_keys[index]['name']}),
+          $el('input.key',{type:'text',oninput: e=>{
+              const dataIndex = Array.prototype.indexOf.call(this.dialog_div.querySelectorAll('.easyuse-account-dialog-item'), e.target.parentNode)
+              api_keys[dataIndex]['key'] = e.target.value
+          },placeholder:'Enter APIKEY', value:api_keys[index]['key']}),
+          $el('button.choose',{textContent:$t('Choose'),onclick:async(e)=>{
+                const dataIndex = Array.prototype.indexOf.call(this.dialog_div.querySelectorAll('.easyuse-account-dialog-item'), e.target.parentNode)
+                let name = api_keys[dataIndex]['name']
+                let key = api_keys[dataIndex]['key']
+                if(!name){
+                    toast.error($t('Please enter the account name'))
+                    return
+                }
+                else if(!key){
+                    toast.error($t('Please enter the APIKEY'))
+                    return
+                }
+                let missing = true
+                for(let i=0;i<api_keys.length;i++){
+                    if(!api_keys[i].key) {
+                        missing = false
+                        break
+                    }
+                }
+                if(!missing){
+                    toast.error($t('APIKEY is not Empty'))
+                    return
+                }
+                // 保存记录
+                api_current = dataIndex
+                const body = new FormData();
+                body.append('api_keys', JSON.stringify(api_keys));
+                body.append('current',api_current)
+                const res = await api.fetchApi('/easyuse/stability/set_api_keys', {
+                    method: 'POST',
+                    body
+                })
+                 if (res.status == 200) {
+                    const data = await res.json()
+                    if(data?.account && data?.balance){
+                        const avatar = data.account?.profile_picture || null
+                        const email = data.account?.email || null
+                        const credits = data.balance?.credits || 0
+                        user_div.replaceChildren(
+                            $el('div.easyuse-account-user-info', {
+                                onclick:_=>{
+                                    new AccountDialog().show(user_div);
+                                }
+                            },[
+                                $el('div.user',[
+                                   $el('div.avatar', avatar ? [$el('img',{src:avatar})] : '😀'),
+                                   $el('div.info', [
+                                    $el('h5.name', email),
+                                    $el('h6.remark','Credits: '+ credits)
+                                   ])
+                                ]),
+                                $el('div.edit', {textContent:$t('Edit')})
+                            ])
+                        )
+                        toast.success($t('Save Succeed'))
+                    }
+                    else toast.success($t('Save Succeed'))
+                    this.close()
+                } else {
+                    toast.error($t('Save Failed'))
+                }
+          }}),
+          $el('button.delete',{textContent:$t('Delete'),onclick:e=>{
+              const dataIndex = Array.prototype.indexOf.call(this.dialog_div.querySelectorAll('.easyuse-account-dialog-item'), e.target.parentNode)
+              if(api_keys.length<=1){
+                toast.error($t('At least one account is required'))
+                return
+              }
+              api_keys.splice(dataIndex,1)
+              this.dialog_div.removeChild(e.target.parentNode)
+          }}),
+      ])
+    }
+
+    show(userdiv) {
+        api_keys.forEach((item,index)=>{
+          this.lists.push(this.addItem(index,userdiv))
+        })
+        this.dialog_div = $el("div.easyuse-account-dialog", this.lists)
+		super.show(
+            $el('div.easyuse-account-dialog-main',[
+               $el('div',[
+                    $el('a',{href:'https://platform.stability.ai/account/keys',target:'_blank',textContent:$t('Getting Your APIKEY')}),
+                ]),
+                this.dialog_div,
+            ])
+        );
+	}
+
+    createButtons() {
+		const btns = super.createButtons();
+        btns.unshift($el('button',{
+            type:'button',
+            textContent:$t('Save Account Info'),
+            onclick:_=>{
+                let missing = true
+                for(let i=0;i<api_keys.length;i++){
+                    if(!api_keys[i].key) {
+                        missing = false
+                        break
+                    }
+                }
+                if(!missing){
+                    toast.error($t('APIKEY is not Empty'))
+                }
+                else {
+                    const body = new FormData();
+                    body.append('api_keys', JSON.stringify(api_keys));
+                    api.fetchApi('/easyuse/stability/set_api_keys', {
+                        method: 'POST',
+                        body
+                    }).then(res => {
+                        if (res.status == 200) {
+                            toast.success($t('Save Succeed'))
+
+                        } else {
+                            toast.error($t('Save Failed'))
+                        }
+                    })
+                }
+            }
+        }))
+        btns.unshift($el('button',{
+            type:'button',
+            textContent:$t('Add Account'),
+            onclick:_=>{
+                const name = 'Account '+(api_keys.length).toString()
+                api_keys.push({name,key:''})
+                const item = this.addItem(api_keys.length - 1)
+                this.lists.push(item)
+                this.dialog_div.appendChild(item)
+            }
+        }))
+        return btns
+    }
+}
+
+
+app.registerExtension({
+    name: 'comfy.easyUse.account',
+    async beforeRegisterNodeDef(nodeType, nodeData, app) {
+        if(nodeData.name == 'easy stableDiffusion3API'){
+
+            const onNodeCreated = nodeType.prototype.onNodeCreated;
+            nodeType.prototype.onNodeCreated = async function() {
+                onNodeCreated ? onNodeCreated?.apply(this, arguments) : undefined;
+                const seed_widget = this.widgets.find(w => ['seed_num','seed'].includes(w.name))
+				const seed_control = this.widgets.find(w=> ['control_before_generate','control_after_generate'].includes(w.name))
+                let model_widget = this.widgets.find(w => w.name == 'model')
+                model_widget.callback = value =>{
+                    cost_widget.value = '-'+api_cost[value]
+                }
+                const cost_widget = this.addWidget('text', 'cost_credit', '0', _=>{
+                },{
+                    serialize:false,
+                })
+                cost_widget.disabled = true
+                setTimeout(_=>{
+                    if(seed_control.name == 'control_before_generate' && seed_widget.value === 0){
+                        seed_widget.value = Math.floor(Math.random() * 4294967294)
+				    }
+                    cost_widget.value = '-'+api_cost[model_widget.value]
+                },100)
+                let user_div = $el('div.easyuse-account-user', [$t('Loading UserInfo...')])
+                let account = this.addDOMWidget('account',"btn",$el('div.easyuse-account',user_div));
+                // 更新balance信息
+                api.addEventListener('stable-diffusion-api-generate-succeed', async ({detail}) => {
+                    let remarkDiv = user_div.querySelectorAll('.remark')
+                    if(remarkDiv && remarkDiv[0]){
+                        const credits = detail?.model ? api_cost[detail.model] : 0
+                        if(credits) {
+                            let balance = accSub(parseFloat(remarkDiv[0].innerText.replace(/Credits: /g,'')),credits)
+                            if(balance>0){
+                                remarkDiv[0].innerText = 'Credits: '+ balance.toString()
+                            }
+                        }
+                    }
+                    await sleep(10000)
+                    const res = await api.fetchApi('/easyuse/stability/balance')
+                    if(res.status == 200){
+                        const data = await res.json()
+                        if(data?.balance){
+                            const credits = data.balance?.credits || 0
+                            if(remarkDiv && remarkDiv[0]){
+                                remarkDiv[0].innerText = 'Credits: ' + credits
+                            }
+                        }
+                    }
+                })
+                // 获取api_keys
+                const res = await api.fetchApi('/easyuse/stability/api_keys')
+                if (res.status == 200){
+                    let data = await res.json()
+                    api_keys = data.keys
+                    api_current = data.current
+                    if (api_keys.length > 0 && api_current!==undefined){
+                        const api_key = api_keys[api_current]['key']
+                        const api_name = api_keys[api_current]['name']
+                        if(!api_key){
+                            user_div.replaceChildren(
+                                $el('div.easyuse-account-user-info', {
+                                    onclick:_=>{
+                                        new AccountDialog().show(user_div);
+                                    }
+                                },[
+                                    $el('div.user',[
+                                        $el('div.avatar', '😀'),
+                                        $el('div.info', [
+                                            $el('h5.name', api_name),
+                                            $el('h6.remark',$t('Click to set the APIKEY first'))
+                                        ])
+                                    ]),
+                                    $el('div.edit', {textContent:$t('Edit')})
+                                ])
+                            )
+                        }else{
+                            // 获取账号信息
+                            const res = await  api.fetchApi('/easyuse/stability/user_info')
+                            if(res.status == 200){
+                                const data = await res.json()
+                                if(data?.account && data?.balance){
+                                    const avatar = data.account?.profile_picture || null
+                                    const email = data.account?.email || null
+                                    const credits = data.balance?.credits || 0
+                                    user_div.replaceChildren(
+                                        $el('div.easyuse-account-user-info', {
+                                            onclick:_=>{
+                                                new AccountDialog().show(user_div);
+                                            }
+                                        },[
+                                            $el('div.user',[
+                                                $el('div.avatar', avatar ? [$el('img',{src:avatar})] : '😀'),
+                                                $el('div.info', [
+                                                    $el('h5.name', email),
+                                                    $el('h6.remark','Credits: '+ credits)
+                                                ])
+                                            ]),
+                                            $el('div.edit', {textContent:$t('Edit')})
+                                        ])
+                                    )
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+})

web/js/easy/easyContextMenu.js

@@ -1,11 +1,12 @@
-import {app} from "/scripts/app.js";
-import {api} from "/scripts/api.js";
-import {$el} from "/scripts/ui.js";
+import {app} from "../../../../scripts/app.js";
+import {api} from "../../../../scripts/api.js";
+import {$el} from "../../../../scripts/ui.js";
 import {$t} from "../common/i18n.js";
 import {getExtension, spliceExtension} from '../common/utils.js'
+import {toast} from "../common/toast.js";
 
 const setting_id = "Comfy.EasyUse.MenuNestSub"
-let enableMenuNestSub = true
+let enableMenuNestSub = false
 let thumbnails = []
 
 export function addMenuNestSubSetting(app) {
@@ -20,7 +21,8 @@ export function addMenuNestSubSetting(app) {
 	});
 }
 
-const getEnableMenuNestSub = _ => app.ui.settings.getSettingValue(setting_id, true)
+const getEnableMenuNestSub = _ => app.ui.settings.getSettingValue(setting_id, enableMenuNestSub)
+
 
 const Loaders = ['easy fullLoader','easy a1111Loader','easy comfyLoader']
 app.registerExtension({
@@ -33,9 +35,12 @@ app.registerExtension({
             let data = await imgRes.json();
             thumbnails = data
         }
+        else if(getEnableMenuNestSub()){
+            toast.error($t("Too many thumbnails, have closed the display"))
+        }
         const existingContextMenu = LiteGraph.ContextMenu;
         LiteGraph.ContextMenu = function(values,options){
-            const threshold = 15;
+            const threshold = 10;
             const enabled = getEnableMenuNestSub();
             if(!enabled || (values?.length || 0) <= threshold || !(options?.callback) || values.some(i => typeof i !== 'string')){
                 if(enabled){

web/js/easy/easyDynamicWidgets.js

@@ -1,37 +1,14 @@
-import { app } from "/scripts/app.js";
-import { api } from "/scripts/api.js";
-import { ComfyWidgets } from "/scripts/widgets.js";
+import { app } from "../../../../scripts/app.js";
+import { api } from "../../../../scripts/api.js";
+import { ComfyWidgets } from "../../../../scripts/widgets.js";
+import { toast} from "../common/toast.js";
+import { $t } from '../common/i18n.js';
 
-let origProps = {};
+import { findWidgetByName, toggleWidget, updateNodeHeight} from "../common/utils.js";
 
 const seedNodes = ["easy seed", "easy latentNoisy", "easy wildcards", "easy preSampling", "easy preSamplingAdvanced", "easy preSamplingNoiseIn", "easy preSamplingSdTurbo", "easy preSamplingCascade", "easy preSamplingDynamicCFG", "easy preSamplingLayerDiffusion", "easy fullkSampler", "easy fullCascadeKSampler"]
 const loaderNodes = ["easy fullLoader", "easy a1111Loader", "easy comfyLoader"]
-const findWidgetByName = (node, name) => node.widgets.find((w) => w.name === name);
 
-const doesInputWithNameExist = (node, name) => node.inputs ? node.inputs.some((input) => input.name === name) : false;
-
-function updateNodeHeight(node) {node.setSize([node.size[0], node.computeSize()[1]]);}
-
-function toggleWidget(node, widget, show = false, suffix = "") {
-	if (!widget || doesInputWithNameExist(node, widget.name)) return;
-	if (!origProps[widget.name]) {
-		origProps[widget.name] = { origType: widget.type, origComputeSize: widget.computeSize };	
-	}
-	const origSize = node.size;
-
-	widget.type = show ? origProps[widget.name].origType : "easyHidden" + suffix;
-	widget.computeSize = show ? origProps[widget.name].origComputeSize : () => [0, -4];
-
-	widget.linkedWidgets?.forEach(w => toggleWidget(node, w, ":" + widget.name, show));	
-
-	const height = show ? Math.max(node.computeSize()[1], origSize[1]) : node.size[1];
-	node.setSize([node.size[0], height]);
-}
-
-function toggleInput(node, name, show = false) {
-	if(!show){
-	}
-}
 
 function widgetLogic(node, widget) {
 	if (widget.name === 'lora_name') {
@@ -74,18 +51,18 @@ function widgetLogic(node, widget) {
 		updateNodeHeight(node)
 	}
 	if (widget.name === 'image_output') {
-	    if (widget.value === 'Sender' || widget.value === 'Sender/Save'){
+	    if (widget.value === 'Sender' || widget.value === 'Sender&Save'){
 	        toggleWidget(node, findWidgetByName(node, 'link_id'), true)
 	    }else {
 	        toggleWidget(node, findWidgetByName(node, 'link_id'))
 	    }
-		if (widget.value === 'Hide' || widget.value === 'Preview' || widget.value === 'Sender') {
+		if (widget.value === 'Hide' || widget.value === 'Preview' || widget.value == 'Preview&Choose' || widget.value === 'Sender') {
 			toggleWidget(node, findWidgetByName(node, 'save_prefix'))
 			toggleWidget(node, findWidgetByName(node, 'output_path'))
 			toggleWidget(node, findWidgetByName(node, 'embed_workflow'))
 			toggleWidget(node, findWidgetByName(node, 'number_padding'))
 			toggleWidget(node, findWidgetByName(node, 'overwrite_existing'))
-		} else if (widget.value === 'Save' || widget.value === 'Hide/Save' || widget.value === 'Sender/Save') {
+		} else if (widget.value === 'Save' || widget.value === 'Hide&Save' || widget.value === 'Sender&Save') {
 			toggleWidget(node, findWidgetByName(node, 'save_prefix'), true)
 			toggleWidget(node, findWidgetByName(node, 'output_path'), true)
 			toggleWidget(node, findWidgetByName(node, 'embed_workflow'), true)
@@ -93,7 +70,7 @@ function widgetLogic(node, widget) {
 			toggleWidget(node, findWidgetByName(node, 'overwrite_existing'), true)
 		}
 
-		if(widget.value === 'Hide' || widget.value === 'Hide/Save'){
+		if(widget.value === 'Hide' || widget.value === 'Hide&Save'){
 			toggleWidget(node, findWidgetByName(node, 'decode_vae_name'))
 		}else{
 			toggleWidget(node, findWidgetByName(node, 'decode_vae_name'), true)
@@ -270,17 +247,66 @@ function widgetLogic(node, widget) {
 	if (widget.name === 'num_embeds') {
 		let number_to_show = widget.value + 1
 		for (let i = 0; i < number_to_show; i++) {
-			toggleInput(node, 'image'+i, true)
-			toggleInput(node, 'mask'+i, true)
 			toggleWidget(node, findWidgetByName(node, 'weight'+i), true)
 		}
 		for (let i = number_to_show; i < 6; i++) {
-			toggleInput(node, 'image'+i)
-			toggleInput(node, 'mask'+i)
 			toggleWidget(node, findWidgetByName(node, 'weight'+i))
 		}
 		updateNodeHeight(node)
 	}
+
+	if (widget.name === 'guider'){
+		switch (widget.value){
+			case 'Basic':
+				toggleWidget(node, findWidgetByName(node, 'cfg'))
+				toggleWidget(node, findWidgetByName(node, 'cfg_negative'))
+				break
+			case 'CFG':
+				toggleWidget(node, findWidgetByName(node, 'cfg'),true)
+				toggleWidget(node, findWidgetByName(node, 'cfg_negative'))
+				break
+			case 'IP2P+DualCFG':
+			case 'DualCFG':
+				toggleWidget(node, findWidgetByName(node, 'cfg'),true)
+				toggleWidget(node, findWidgetByName(node, 'cfg_negative'), true)
+				break
+
+		}
+		updateNodeHeight(node)
+	}
+
+	if (widget.name === 'scheduler'){
+		if (['karrasADV','exponentialADV','polyExponential'].includes(widget.value)){
+			toggleWidget(node, findWidgetByName(node, 'sigma_max'), true)
+			toggleWidget(node, findWidgetByName(node, 'sigma_min'), true)
+			toggleWidget(node, findWidgetByName(node, 'denoise'))
+			toggleWidget(node, findWidgetByName(node, 'beta_d'))
+			toggleWidget(node, findWidgetByName(node, 'beta_min'))
+			toggleWidget(node, findWidgetByName(node, 'eps_s'))
+			if(widget.value != 'exponentialADV'){
+				toggleWidget(node, findWidgetByName(node, 'rho'), true)
+			}else{
+				toggleWidget(node, findWidgetByName(node, 'rho'))
+			}
+		}else if(widget.value == 'vp'){
+			toggleWidget(node, findWidgetByName(node, 'sigma_max'))
+			toggleWidget(node, findWidgetByName(node, 'sigma_min'))
+			toggleWidget(node, findWidgetByName(node, 'denoise'))
+			toggleWidget(node, findWidgetByName(node, 'rho'))
+			toggleWidget(node, findWidgetByName(node, 'beta_d'),true)
+			toggleWidget(node, findWidgetByName(node, 'beta_min'),true)
+			toggleWidget(node, findWidgetByName(node, 'eps_s'),true)
+		}else{
+			toggleWidget(node, findWidgetByName(node, 'denoise'),true)
+			toggleWidget(node, findWidgetByName(node, 'sigma_max'))
+			toggleWidget(node, findWidgetByName(node, 'sigma_min'))
+			toggleWidget(node, findWidgetByName(node, 'beta_d'))
+			toggleWidget(node, findWidgetByName(node, 'beta_min'))
+			toggleWidget(node, findWidgetByName(node, 'eps_s'))
+			toggleWidget(node, findWidgetByName(node, 'rho'))
+		}
+		updateNodeHeight(node)
+	}
 }
 
 function widgetLogic2(node, widget) {
@@ -533,8 +559,10 @@ app.registerExtension({
 			case "easy dynamiCrafterLoader":
 			case "easy loraStack":
 			case "easy latentNoisy":
+			case "easy preSampling":
 			case "easy preSamplingAdvanced":
 			case "easy preSamplingNoiseIn":
+			case "easy preSamplingCustom":
 			case "easy preSamplingSdTurbo":
 			case "easy preSamplingCascade":
 			case "easy preSamplingLayerDiffusion":
@@ -550,6 +578,8 @@ app.registerExtension({
 			case "easy hiresFix":
 			case "easy detailerFix":
 			case "easy imageRemBg":
+			case "easy imageColorMatch":
+			case "easy loadImageBase64":
 			case "easy XYInputs: Steps":
 			case "easy XYInputs: Sampler/Scheduler":
 			case 'easy XYInputs: Checkpoint':
@@ -777,6 +807,7 @@ app.registerExtension({
 					const pos = this.widgets.findIndex((w) => w.name === "spent_time");
 					if (pos !== -1 && this.widgets[pos]) {
 						const w = this.widgets[pos]
+						console.log(text)
 						w.value = text;
 					}
 				}
@@ -916,6 +947,7 @@ app.registerExtension({
 							seed_control.value = 'fixed'
 						}
 						seed_widget.value = Math.floor(Math.random() * 1125899906842624)
+						app.queuePrompt(0, 1)
 					})
 				}
 			}
@@ -924,9 +956,11 @@ app.registerExtension({
 				onAdded ? onAdded.apply(this, []) : undefined;
 				const seed_widget = this.widgets.find(w => ['seed_num','seed'].includes(w.name))
 				const seed_control = this.widgets.find(w=> ['control_before_generate','control_after_generate'].includes(w.name))
-				if(seed_control.name == 'control_before_generate' && seed_widget.value === 0){
-					seed_widget.value = Math.floor(Math.random() * 1125899906842624)
-				}
+				setTimeout(_=>{
+					if(seed_control.name == 'control_before_generate' && seed_widget.value === 0) {
+						seed_widget.value = Math.floor(Math.random() * 1125899906842624)
+					}
+				},1)
 			}
 		}
 
@@ -957,7 +991,7 @@ app.registerExtension({
 			}
 		}
 
-		if(['easy showAnything', 'easy showTensorShape'].includes(nodeData.name)){
+		if(['easy showAnything', 'easy showTensorShape', 'easy imageInterrogator'].includes(nodeData.name)){
 			function populate(text) {
 				if (this.widgets) {
 					const pos = this.widgets.findIndex((w) => w.name === "text");
@@ -996,13 +1030,15 @@ app.registerExtension({
 				populate.call(this, message.text);
 			};
 
-			const onConfigure = nodeType.prototype.onConfigure;
-			nodeType.prototype.onConfigure = function () {
-				onConfigure?.apply(this, arguments);
-				if (this.widgets_values?.length) {
-					populate.call(this, this.widgets_values);
-				}
-			};
+			if(!['easy imageInterrogator'].includes(nodeData.name)) {
+				const onConfigure = nodeType.prototype.onConfigure;
+				nodeType.prototype.onConfigure = function () {
+					onConfigure?.apply(this, arguments);
+					if (this.widgets_values?.length) {
+						populate.call(this, this.widgets_values);
+					}
+				};
+			}
 		}
 
 		if(nodeData.name == 'easy convertAnything'){
@@ -1028,6 +1064,36 @@ app.registerExtension({
 
 			}
 		}
+
+		if (nodeData.name == 'easy promptLine') {
+			const onAdded = nodeType.prototype.onAdded;
+			nodeType.prototype.onAdded = async function () {
+				onAdded ? onAdded.apply(this, []) : undefined;
+				let prompt_widget = this.widgets.find(w => w.name == "prompt")
+				const button = this.addWidget("button", "get values from COMBO link", '', () => {
+					const output_link = this.outputs[1]?.links?.length>0 ? this.outputs[1]['links'][0] : null
+					const all_nodes = app.graph._nodes
+					const node = all_nodes.find(cate=> cate.inputs?.find(input=> input.link == output_link))
+					if(!output_link || !node){
+						toast.error($t('No COMBO link'), 3000)
+						return
+					}
+					else{
+						const input = node.inputs.find(input=> input.link == output_link)
+						const widget_name = input.widget.name
+						const widgets = node.widgets
+						const widget = widgets.find(cate=> cate.name == widget_name)
+						let values = widget?.options.values || null
+						if(values){
+							values = values.join('\n')
+							prompt_widget.value = values
+						}
+					}
+				}, {
+					serialize: false
+				})
+			}
+		}
 	}
 });
 
@@ -1040,7 +1106,7 @@ const getSetWidgets = ['rescale_after_model', 'rescale',
 						'num_loras', 'mode', 'toggle', 'resolution', 'target_parameter',
 	'input_count', 'replace_count', 'downscale_mode', 'range_mode','text_combine_mode', 'input_mode',
 	'lora_count','ckpt_count', 'conditioning_mode', 'preset', 'use_tiled', 'use_batch', 'num_embeds',
-	"easing_mode"
+	"easing_mode", "guider", "scheduler"
 ]
 
 function getSetters(node) {

web/js/easy/easyExtraMenu.js

@@ -1,11 +1,14 @@
-import {app} from "/scripts/app.js";
+import {app} from "../../../../scripts/app.js";
+import {$t} from '../common/i18n.js'
+import {CheckpointInfoDialog, LoraInfoDialog} from "../common/model.js";
 
 const loaders = ['easy fullLoader', 'easy a1111Loader', 'easy comfyLoader']
-const preSampling = ['easy preSampling', 'easy preSamplingAdvanced', 'easy preSamplingDynamicCFG', 'easy preSamplingNoiseIn', 'easy preSamplingLayerDiffusion', 'easy fullkSampler']
+const preSampling = ['easy preSampling', 'easy preSamplingAdvanced', 'easy preSamplingDynamicCFG', 'easy preSamplingNoiseIn', 'easy preSamplingCustom', 'easy preSamplingLayerDiffusion', 'easy fullkSampler']
 const kSampler = ['easy kSampler', 'easy kSamplerTiled', 'easy kSamplerInpainting', 'easy kSamplerDownscaleUnet', 'easy kSamplerLayerDiffusion']
 const controlnet = ['easy controlnetLoader', 'easy controlnetLoaderADV', 'easy instantIDApply', 'easy instantIDApplyADV']
-const ipadapter = ['easy ipadapterApply', 'easy ipadapterApplyADV']
+const ipadapter = ['easy ipadapterApply', 'easy ipadapterApplyADV', 'easy ipadapterStyleComposition', 'easy ipadapterApplyFromParams']
 const positive_prompt = ['easy positive', 'easy wildcards']
+const imageNode = ['easy loadImageBase64', 'LoadImage', 'LoadImageMask']
 const widgetMapping = {
     "positive_prompt":{
         "text": "positive",
@@ -56,6 +59,11 @@ const widgetMapping = {
         "end_at": "end_at",
         "cache_mode": "cache_mode",
         "use_tiled": "use_tiled",
+    },
+    "load_image":{
+        "image":"image",
+        "base64_data":"base64_data",
+        "channel": "channel"
     }
 }
 const inputMapping = {
@@ -88,6 +96,7 @@ const inputMapping = {
     "ipadapter":{
         "model":"model",
         "image":"image",
+        "image_style": "image",
         "attn_mask":"attn_mask",
         "optional_ipadapter":"optional_ipadapter"
     }
@@ -128,7 +137,7 @@ const outputMapping = {
         "tiles":"tiles",
         "masks":"masks",
         "ipadapter":"ipadapter"
-    }
+    },
 };
 
 // 替换节点
@@ -264,6 +273,25 @@ const addMenu = (content, type, nodes_include, nodeType, has_submenu=true) => {
             has_submenu: has_submenu,
             callback: (value, options, e, menu, node) => showSwapMenu(value, options, e, menu, node, type, nodes_include)
         })
+        if(type == 'loaders') {
+            options.unshift({
+                content: $t("💎 View Lora Info..."),
+                callback: (value, options, e, menu, node) => {
+                    const widget = node.widgets.find(cate => cate.name == 'lora_name')
+                    let name = widget.value;
+                    if (!name || name == 'None') return
+                    new LoraInfoDialog(name).show('loras', name);
+                }
+            })
+            options.unshift({
+                content: $t("💎 View Checkpoint Info..."),
+                callback: (value, options, e, menu, node) => {
+                    let name = node.widgets[0].value;
+                    if (!name || name == 'None') return
+                    new CheckpointInfoDialog(name).show('checkpoints', name);
+                }
+            })
+        }
     })
 }
 const showSwapMenu = (value, options, e, menu, node, type, nodes_include) => {
@@ -482,7 +510,7 @@ app.registerExtension({
         // 刷新节点
         addMenuHandler(nodeType, function (_, options) {
             options.unshift({
-                content: "🔃 Reload Node",
+                content: $t("🔃 Reload Node"),
                 callback: (value, options, e, menu, node) => {
                     let graphcanvas = LGraphCanvas.active_canvas;
                     if (!graphcanvas.selected_nodes || Object.keys(graphcanvas.selected_nodes).length <= 1) {
@@ -494,7 +522,19 @@ app.registerExtension({
                     }
                 }
             })
+            // ckptNames
+            if(nodeData.name == 'easy ckptNames'){
+                options.unshift({
+                    content: $t("💎 View Checkpoint Info..."),
+                    callback: (value, options, e, menu, node) => {
+                        let name = node.widgets[0].value;
+                        if (!name || name == 'None') return
+                        new CheckpointInfoDialog(name).show('checkpoints', name);
+                    }
+                })
+            }
         })
+
         // Swap提示词
         if (positive_prompt.includes(nodeData.name)) {
             addMenu("↪️ Swap EasyPrompt", 'positive_prompt', positive_prompt, nodeType)
@@ -519,6 +559,10 @@ app.registerExtension({
         if (ipadapter.includes(nodeData.name)) {
             addMenu("↪️ Swap EasyIPAdapater", 'ipadapter', ipadapter, nodeType)
         }
+        // Swap Image
+        if (imageNode.includes(nodeData.name)) {
+            addMenu("↪️ Swap LoadImage", 'load_image', imageNode, nodeType)
+        }
     }
 });
 

web/js/easy/easyInterface.js

@@ -1,12 +1,12 @@
-import { app } from "/scripts/app.js";
-import { api } from "/scripts/api.js";
-import { $el } from "/scripts/ui.js";
+import { app } from "../../../../scripts/app.js";
+import { api } from "../../../../scripts/api.js";
+import { $el } from "../../../../scripts/ui.js";
 import {addPreconnect, addCss} from "../common/utils.js";
 
 const locale = localStorage['AGL.Locale'] || localStorage['Comfy.Settings.AGL.Locale'] || 'en-US'
 
 const customThemeColor = "#3f3eed"
-const customThemeColorLight = "#006691"
+const customThemeColorLight = "#008ecb"
 // 增加Slot颜色
 const customPipeLineLink = "#7737AA"
 const customPipeLineSDXLLink = "#7737AA"
@@ -131,7 +131,6 @@ try{
             settings["AE.highlight"] = false
         }
         // 主题设置
-        console.log(theme_name)
         if(!theme_name && _settings['Comfy.ColorPalette']) {
             theme_name = `"${_settings['Comfy.ColorPalette']}"`
             localStorage.setItem('Comfy.Settings.Comfy.ColorPalette', theme_name)
@@ -743,8 +742,12 @@ const NODE_COLORS = {
     "easy positive":"green",
     "easy negative":"red",
     "easy promptList":"cyan",
+    "easy promptLine":"cyan",
+    "easy promptConcat":"cyan",
+    "easy promptReplace":"cyan",
     "easy XYInputs: Seeds++ Batch": customXYLink,
     "easy XYInputs: ModelMergeBlocks": customXYLink,
+    'easy textSwitch': "pale_blue"
 }
 
 function setNodeColors(node, theme) {

web/js/easy/easyQuick.js

@@ -1,7 +1,7 @@
 // 1.0.2
-import { app } from "/scripts/app.js";
-import { GroupNodeConfig } from "/extensions/core/groupNode.js";
-import { api } from "/scripts/api.js";
+import { app } from "../../../../scripts/app.js";
+import { GroupNodeConfig } from "../../../../extensions/core/groupNode.js";
+import { api } from "../../../../scripts/api.js";
 import { $t } from "../common/i18n.js"
 
 const nodeTemplateShortcutId = "Comfy.EasyUse.NodeTemplateShortcut"

web/js/easy/easySaveImage.js

@@ -1,5 +1,5 @@
-import { app } from "/scripts/app.js";
-import { applyTextReplacements } from "/scripts/utils.js";
+import { app } from "../../../../scripts/app.js";
+import { applyTextReplacements } from "../../../../scripts/utils.js";
 
 const extraNodes = ["easy imageSave", "easy fullkSampler", "easy kSampler", "easy kSamplerTiled","easy kSamplerInpainting", "easy kSamplerDownscaleUnet", "easy kSamplerSDTurbo","easy detailerFix"]
 

web/js/easy/easySeg.js

@@ -0,0 +1,136 @@
+import {app} from "../../../../scripts/app.js";
+import {$el} from "../../../../scripts/ui.js";
+import {$t} from "../common/i18n.js";
+import {findWidgetByName, toggleWidget} from "../common/utils.js";
+
+
+const tags = {
+    "selfie_multiclass_256x256": ["Background", "Hair", "Body", "Face", "Clothes", "Others",],
+    "human_parsing_lip":["Background","Hat","Hair","Glove","Sunglasses","Upper-clothes","Dress","Coat","Socks","Pants","Jumpsuits","Scarf","Skirt","Face","Left-arm","Right-arm","Left-leg","Right-leg","Left-shoe","Right-shoe"],
+}
+function getTagList(tags) {
+    let rlist=[]
+    tags.forEach((k,i) => {
+        rlist.push($el(
+            "label.easyuse-prompt-styles-tag",
+            {
+                dataset: {
+                    tag: i,
+                    name: $t(k),
+                    index: i
+                },
+                $: (el) => {
+                    el.children[0].onclick = () => {
+                        el.classList.toggle("easyuse-prompt-styles-tag-selected");
+                    };
+                },
+            },
+            [
+                $el("input",{
+                    type: 'checkbox',
+                    name: i
+                }),
+                $el("span",{
+                    textContent: $t(k),
+                })
+            ]
+        ))
+    });
+    return rlist
+}
+
+
+app.registerExtension({
+    name: 'comfy.easyUse.seg',
+    async beforeRegisterNodeDef(nodeType, nodeData, app) {
+
+        if (nodeData.name == 'easy humanSegmentation') {
+            // 创建时
+            const onNodeCreated = nodeType.prototype.onNodeCreated;
+            nodeType.prototype.onNodeCreated = function () {
+                onNodeCreated ? onNodeCreated?.apply(this, arguments) : undefined;
+                const method = this.widgets.findIndex((w) => w.name == 'method');
+                const list = $el("ul.easyuse-prompt-styles-list.no-top", []);
+                let method_values = ''
+                this.setProperty("values", [])
+
+                let selector = this.addDOMWidget('mask_components',"btn",$el('div.easyuse-prompt-styles',[list]))
+
+                Object.defineProperty(this.widgets[method],'value',{
+                    set:(value)=>{
+                        method_values = value
+                        if(method_values){
+                            selector.element.children[0].innerHTML = ''
+                            if(method_values == 'selfie_multiclass_256x256'){
+                                toggleWidget(this, findWidgetByName(this, 'confidence'), true)
+                                this.setSize([300, 200]);
+                            }else{
+                                toggleWidget(this, findWidgetByName(this, 'confidence'))
+                                this.setSize([300, 400]);
+                            }
+                            let list = getTagList(tags[method_values]);
+                            selector.element.children[0].append(...list)
+                        }
+                    },
+                    get: () => {
+                        return method_values
+                    }
+                })
+
+                let mask_select_values = ''
+
+                Object.defineProperty(selector, "value", {
+                    set: (value) => {
+                        setTimeout(_=>{
+                            selector.element.children[0].querySelectorAll(".easyuse-prompt-styles-tag").forEach(el => {
+                                let arr = value.split(',')
+                                if (arr.includes(el.dataset.tag)) {
+                                    el.classList.add("easyuse-prompt-styles-tag-selected");
+                                    el.children[0].checked = true
+                                }
+                            })
+                        },100)
+                    },
+                    get: () => {
+                        selector.element.children[0].querySelectorAll(".easyuse-prompt-styles-tag").forEach(el => {
+                            if(el.classList.value.indexOf("easyuse-prompt-styles-tag-selected")>=0){
+                                if(!this.properties["values"].includes(el.dataset.tag)){
+                                    this.properties["values"].push(el.dataset.tag);
+                                }
+                            }else{
+                                if(this.properties["values"].includes(el.dataset.tag)){
+                                    this.properties["values"]= this.properties["values"].filter(v=>v!=el.dataset.tag);
+                                }
+                            }
+                        });
+                        mask_select_values = this.properties["values"].join(',');
+                        return mask_select_values;
+                    }
+                });
+
+                let old_values = ''
+                let mask_lists_dom = selector.element.children[0]
+
+                // 初始化
+                setTimeout(_=>{
+                    if(!method_values) {
+                        method_values = 'selfie_multiclass_256x256'
+                        selector.element.children[0].innerHTML = ''
+                        // 重新排序
+                        let list = getTagList(tags[method_values]);
+                        selector.element.children[0].append(...list)
+                    }
+                    if(method_values == 'selfie_multiclass_256x256'){
+                        toggleWidget(this, findWidgetByName(this, 'confidence'), true)
+                        this.setSize([300, 200]);
+                    }else{
+                        toggleWidget(this, findWidgetByName(this, 'confidence'))
+                        this.setSize([300, 420]);
+                    }
+                },1)
+
+                return onNodeCreated;
+            }
+        }
+    }
+})

web/js/easy/easySelector.js

@@ -1,7 +1,8 @@
 // 1.0.3
-import { app } from "/scripts/app.js";
-import { api } from "/scripts/api.js";
-import { $el } from "/scripts/ui.js";
+import { app } from "../../../../scripts/app.js";
+import { api } from "../../../../scripts/api.js";
+import { $el } from "../../../../scripts/ui.js";
+import { $t } from "../common/i18n.js";
 
 // 获取风格列表
 let styles_list_cache = {}
@@ -118,14 +119,13 @@ app.registerExtension({
                 onNodeCreated ? onNodeCreated?.apply(this, arguments) : undefined;
                 const styles_id =  this.widgets.findIndex((w) => w.name == 'styles');
                 const language = localStorage['AGL.Locale'] || localStorage['Comfy.Settings.AGL.Locale'] || 'en-US'
-
                 const list = $el("ul.easyuse-prompt-styles-list",[]);
                 let styles_values = ''
                 this.setProperty("values", [])
 
                 let selector = this.addDOMWidget('select_styles',"btn",$el('div.easyuse-prompt-styles',[$el('div.tools', [
                         $el('button.delete',{
-                            textContent: language == 'zh-CN' ? '清空所有' : 'Empty All',
+                            textContent: $t('Empty All'),
                             style:{},
                             onclick:()=>{
                                 selector.element.children[0].querySelectorAll(".search").forEach(el=>{
@@ -145,7 +145,7 @@ app.registerExtension({
                             dir:"ltr",
                             style:{"overflow-y": "scroll"},
                             rows:1,
-                            placeholder:language == 'zh-CN' ? "🔎 在此处输入以搜索样式 ..."  : "🔎 Type here to search styles ...",
+                            placeholder:$t("🔎 Type here to search styles ..."),
                             oninput:(e)=>{
                                 let value = e.target.value
                                 selector.element.children[1].querySelectorAll(".easyuse-prompt-styles-tag").forEach(el => {

web/js/easy/easySuggestion.js

@@ -1,14 +1,14 @@
-import {app} from "/scripts/app.js";
-import {api} from "/scripts/api.js";
-import {$el} from "/scripts/ui.js";
+import {app} from "../../../../scripts/app.js";
+import {api} from "../../../../scripts/api.js";
+import {$el} from "../../../../scripts/ui.js";
 
-const propmts = ["easy wildcards", "easy positive", "easy negative", "easy stylesSelector"]
+const propmts = ["easy wildcards", "easy positive", "easy negative", "easy stylesSelector", "easy promptConcat", "easy promptReplace"]
 const loaders = ["easy a1111Loader", "easy comfyLoader", "easy fullLoader", "easy svdLoader", "easy cascadeLoader", "easy sv3dLoader"]
-const preSamplingNodes = ["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingNoiseIn", "easy preSamplingDynamicCFG","easy preSamplingSdTurbo", "easy preSamplingLayerDiffusion"]
+const preSamplingNodes = ["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingNoiseIn", "easy preSamplingCustom", "easy preSamplingDynamicCFG","easy preSamplingSdTurbo", "easy preSamplingLayerDiffusion"]
 const kSampler = ["easy kSampler", "easy kSamplerTiled","easy kSamplerInpainting", "easy kSamplerDownscaleUnet", "easy kSamplerSDTurbo"]
 const controlNetNodes = ["easy controlnetLoader", "easy controlnetLoaderADV"]
 const instantIDNodes = ["easy instantIDApply", "easy instantIDApplyADV"]
-const ipadapterNodes = ["easy ipadapterApply", "easy ipadapterApplyADV"]
+const ipadapterNodes = ["easy ipadapterApply", "easy ipadapterApplyADV" , "easy ipadapterStyleComposition"]
 const pipeNodes = ['easy pipeIn','easy pipeOut', 'easy pipeEdit']
 const xyNodes = ['easy XYPlot', 'easy XYPlotAdvanced']
 const extraNodes = ['easy setNode']
@@ -20,6 +20,36 @@ const suggestions = {
             "INT": [...["Reroute"],...preSamplingNodes,...['easy fullkSampler']]
         }
     },
+    "easy positive":{
+       "from":{
+           "STRING": [...["Reroute"],...propmts]
+       }
+    },
+    "easy negative":{
+       "from":{
+           "STRING": [...["Reroute"],...propmts]
+       }
+    },
+    "easy wildcards":{
+        "from":{
+           "STRING": [...["Reroute","easy showAnything"],...propmts,]
+       }
+    },
+    "easy stylesSelector":{
+        "from":{
+           "STRING": [...["Reroute","easy showAnything"],...propmts,]
+       }
+    },
+    "easy promptConcat":{
+        "from":{
+           "STRING": [...["Reroute","easy showAnything"],...propmts,]
+       }
+    },
+    "easy promptReplace":{
+        "from":{
+           "STRING": [...["Reroute","easy showAnything"],...propmts,]
+       }
+    },
     // sd相关
     "easy fullLoader": {
         "from":{
@@ -89,6 +119,11 @@ const suggestions = {
          "from": {
              "PIPE_LINE": [...["Reroute"], ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
          }
+    },
+     "easy preSamplingCustom": {
+         "from": {
+             "PIPE_LINE": [...["Reroute"], ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
+         }
     },
     "easy preSamplingLayerDiffusion": {
          "from": {
@@ -127,12 +162,33 @@ const suggestions = {
         "from": {
             "PIPE_LINE": [...["Reroute"], ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
              "MODEL": modelNormalNodes
+        },
+        "to":{
+            "COMBO": [...["Reroute", "easy promptLine"]]
         }
     },
     "easy instantIDApplyADV":{
         "from": {
             "PIPE_LINE": [...["Reroute"], ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
             "MODEL": modelNormalNodes
+        },
+        "to":{
+            "COMBO": [...["Reroute", "easy promptLine"]]
+        }
+    },
+    "easy ipadapterApply":{
+        "to":{
+            "COMBO": [...["Reroute", "easy promptLine"]]
+        }
+    },
+    "easy ipadapterApplyADV":{
+        "to":{
+            "COMBO": [...["Reroute", "easy promptLine"]]
+        }
+    },
+    "easy ipadapterStyleComposition":{
+        "to":{
+            "COMBO": [...["Reroute", "easy promptLine"]]
         }
     },
     // fix

web/js/easy/easyWidgets.js

@@ -1,5 +1,5 @@
-import { app } from "/scripts/app.js";
-import { ComfyWidgets } from "/scripts/widgets.js";
+import { app } from "../../../../scripts/app.js";
+import { ComfyWidgets } from "../../../../scripts/widgets.js";
 
 const KEY_CODES = { ENTER: 13, ESC: 27, ARROW_DOWN: 40, ARROW_UP: 38 };
 const WIDGET_GAP = -4;
@@ -150,7 +150,6 @@ const cssCode = `
 	border-radius: 7px;
 	text-align: center;
 	text-wrap: balance;
-	text-transform: uppercase;
 }
 .hideInfo-dropdown {
 	position: absolute;

web/js/easy/easyXYPlot.js

@@ -1,4 +1,4 @@
-import { app } from "/scripts/app.js";
+import { app } from "../../../../scripts/app.js";
 import {removeDropdown, createDropdown} from "../common/dropdown.js";
 
 function generateNumList(dictionary) {

web/js/getset.js

@@ -1,5 +1,4 @@
 import { app } from "../../../scripts/app.js";
-import { ComfyWidgets } from '/scripts/widgets.js'
 
 // Node that allows you to tunnel connections for cleaner graphs
 
@@ -201,7 +200,7 @@ app.registerExtension({
 					},
 					{
 						values: () => {
-                            const setterNodes = graph._nodes.filter((otherNode) => otherNode.type == 'easy setNode');
+                            const setterNodes = node.graph._nodes.filter((otherNode) => otherNode.type == 'easy setNode');
                             return setterNodes.map((otherNode) => otherNode.widgets[0].value).sort();
                         }
 					}

web/js/image_chooser/chooser.js

@@ -0,0 +1,199 @@
+import { app } from "../../../../scripts/app.js";
+import { api } from "../../../../scripts/api.js";
+import { ComfyDialog, $el } from "../../../../scripts/ui.js";
+
+import { restart_from_here } from "./prompt.js";
+import { FlowState } from "./state.js";
+import { send_cancel, send_message, send_onstart, skip_next_restart_message } from "./messaging.js";
+import { display_preview_images, additionalDrawBackground, click_is_in_image } from "./preview.js";
+import {$t} from "../common/i18n.js";
+
+
+class chooserImageDialog extends ComfyDialog {
+
+    constructor() {
+		super();
+        this.node = null
+        this.select_index = []
+        this.dialog_div = null
+	}
+
+    show(image,node){
+        this.select_index = []
+        this.node = node
+
+        const images_div = image.map((img, index) => {
+            const imgEl = $el('img', {
+                src: img.src,
+                onclick: _ => {
+                    if(this.select_index.includes(index)){
+                        this.select_index = this.select_index.filter(i => i !== index)
+                        imgEl.classList.remove('selected')
+                    } else {
+                        this.select_index.push(index)
+                        imgEl.classList.add('selected')
+                    }
+                    if (node.selected.has(index)) node.selected.delete(index);
+                    else node.selected.add(index);
+                }
+            })
+            return imgEl
+        })
+        super.show($el('div.easyuse-chooser-dialog',[
+            $el('h5.easyuse-chooser-dialog-title', $t('Choose images to continue')),
+            $el('div.easyuse-chooser-dialog-images',images_div)
+        ]))
+    }
+    createButtons() {
+        const btns = super.createButtons();
+        btns[0].onclick = _ => {
+            cancelButtonPressed()
+            super.close()
+        }
+        btns.unshift($el('button', {
+            type: 'button',
+            textContent: $t('Choose Selected Images'),
+            onclick: _ => {
+                if (FlowState.paused()) {
+                    send_message(this.node.id, [...this.node.selected, -1, ...this.node.anti_selected]);
+                }
+                if (FlowState.idle()) {
+                    skip_next_restart_message();
+                    restart_from_here(this.node.id).then(() => { send_message(this.node.id, [...this.node.selected, -1, ...this.node.anti_selected]); });
+                }
+                super.close()
+            }
+        }))
+        return btns
+    }
+
+}
+
+function progressButtonPressed() {
+    const node = app.graph._nodes_by_id[this.node_id];
+    if (node) {
+        if (FlowState.paused()) {
+            send_message(node.id, [...node.selected, -1, ...node.anti_selected]);
+        }
+        if (FlowState.idle()) {
+            skip_next_restart_message();
+            restart_from_here(node.id).then(() => { send_message(node.id, [...node.selected, -1, ...node.anti_selected]); });
+        }
+    }
+}
+
+function cancelButtonPressed() { if (FlowState.running()) { send_cancel(); } }
+
+app.registerExtension({
+    name:'comfy.easyuse.imageChooser',
+    init() {
+        window.addEventListener("beforeunload", send_cancel, true);
+    },
+    setup(app) {
+        function easyuseImageChooser(event) {
+            const {node,image,isKSampler} = display_preview_images(event);
+            if(isKSampler) {
+                const dialog = new chooserImageDialog();
+                dialog.show(image,node)
+            }
+        }
+        api.addEventListener("easyuse-image-choose", easyuseImageChooser);
+
+        /*
+        If a run is interrupted, send a cancel message (unless we're doing the cancelling, to avoid infinite loop)
+        */
+        const original_api_interrupt = api.interrupt;
+        api.interrupt = function () {
+            if (FlowState.paused() && !FlowState.cancelling) send_cancel();
+            original_api_interrupt.apply(this, arguments);
+        }
+
+        /*
+        At the start of execution
+        */
+        function on_execution_start() {
+            if (send_onstart()) {
+                app.graph._nodes.forEach((node)=> {
+                    if (node.selected || node.anti_selected) {
+                        node.selected.clear();
+                        node.anti_selected.clear();
+                        node.update();
+                    }
+                })
+            }
+        }
+        api.addEventListener("execution_start", on_execution_start);
+    },
+
+    async nodeCreated(node, app) {
+        if(node.comfyClass == 'easy imageChooser'){
+            node.send_button_widget = node.addWidget("button", "", "", progressButtonPressed, {serialize: false});
+            node.cancel_button_widget = node.addWidget("button", "", "", cancelButtonPressed, {serialize: false});
+            /* Capture clicks */
+            const org_onMouseDown = node.onMouseDown;
+
+            /* A property defining the top of the image when there is just one */
+            if(node?.imageIndex === undefined){
+              Object.defineProperty(node, 'imageIndex', {
+                    get : function() { return null; },
+                    set: function (v) {node.overIndex= v},
+                })
+            }
+            if(node?.imagey === undefined){
+                Object.defineProperty(node, 'imagey', {
+                    get : function() { return null; },
+                    set: function (v) {return node.widgets[node.widgets.length-1].last_y+LiteGraph.NODE_WIDGET_HEIGHT;},
+                })
+            }
+
+            node.onMouseDown = function( e, pos, canvas ) {
+                if (e.isPrimary) {
+                    const i = click_is_in_image(node, pos);
+                    if (i>=0) { this.imageClicked(i); }
+                }
+                return (org_onMouseDown && org_onMouseDown.apply(this, arguments));
+            }
+
+        }
+    },
+
+    beforeRegisterNodeDef(nodeType, nodeData, app) {
+        if(nodeData?.name == 'easy imageChooser'){
+
+            const onDrawBackground = nodeType.prototype.onDrawBackground;
+            nodeType.prototype.onDrawBackground = function(ctx) {
+                onDrawBackground.apply(this, arguments);
+                additionalDrawBackground(this, ctx);
+            }
+
+            nodeType.prototype.imageClicked = function (imageIndex) {
+                if (this.selected.has(imageIndex)) this.selected.delete(imageIndex);
+                else this.selected.add(imageIndex);
+                this.update();
+            }
+
+            const update = nodeType.prototype.update;
+            nodeType.prototype.update = function() {
+                if (update) update.apply(this,arguments);
+                if (this.send_button_widget) {
+                    this.send_button_widget.node_id = this.id;
+                    const selection = ( this.selected ? this.selected.size : 0 ) + ( this.anti_selected ? this.anti_selected.size : 0 )
+                    const maxlength = this.imgs.length;
+                    if (FlowState.paused_here(this.id) && selection>0) {
+                        this.send_button_widget.name = (selection>1) ? "Progress selected (" + selection + '/' + maxlength  +")" : "Progress selected image";
+                    } else if (FlowState.idle() && selection>0) {
+                        this.send_button_widget.name = (selection>1) ? "Progress selected (" + selection + '/' + maxlength  +")" : "Progress selected image as restart";
+                    }
+                    else {
+                        this.send_button_widget.name = "";
+                    }
+                }
+                if (this.cancel_button_widget) {
+                    const isRunning = FlowState.running()
+                    this.cancel_button_widget.name = isRunning ? "Cancel current run" : "";
+                }
+                this.setDirtyCanvas(true,true);
+            }
+		}
+    }
+})

web/js/image_chooser/messaging.js

@@ -0,0 +1,34 @@
+import { api } from "../../../../scripts/api.js";
+import { FlowState } from "./state.js";
+
+function send_message_from_pausing_node(message) {
+    const id = app.runningNodeId;
+    send_message(id, message);
+}
+
+function send_message(id, message) {
+    const body = new FormData();
+    body.append('message',message);
+    body.append('id', id);
+    api.fetchApi("/easyuse/image_chooser_message", { method: "POST", body, });
+}
+
+function send_cancel() {
+    send_message(-1,'__cancel__');
+    //FlowState.cancelling = true;
+    //api.interrupt();
+    //FlowState.cancelling = false;
+}
+
+var skip_next = 0;
+function skip_next_restart_message() { skip_next += 1; }
+function send_onstart() {
+    if (skip_next>0) {
+        skip_next -= 1;
+        return false;
+    }
+    send_message(-1,'__start__');
+    return true;
+}
+
+export { send_message_from_pausing_node, send_cancel, send_message, send_onstart, skip_next_restart_message }

web/js/image_chooser/preview.js

@@ -0,0 +1,90 @@
+import { app } from "../../../../scripts/app.js";
+
+const kSampler = ['easy kSampler', 'easy kSamplerTiled', 'easy fullkSampler']
+
+function display_preview_images(event) {
+    const node = app.graph._nodes_by_id[event.detail.id];
+    if (node) {
+        node.selected = new Set();
+        node.anti_selected = new Set();
+        const image = showImages(node, event.detail.urls);
+        return {node,image,isKSampler:kSampler.includes(node.type)}
+    } else {
+        console.log(`Image Chooser Preview - failed to find ${event.detail.id}`)
+    }
+}
+
+function showImages(node, urls) {
+    node.imgs = [];
+    urls.forEach((u)=> {
+        const img = new Image();
+        node.imgs.push(img);
+        img.onload = () => { app.graph.setDirtyCanvas(true); };
+        img.src = `/view?filename=${encodeURIComponent(u.filename)}&type=temp&subfolder=${app.getPreviewFormatParam()}`
+    })
+    node.setSizeForImage?.();
+    return node.imgs
+}
+
+function drawRect(node, s, ctx) {
+    const padding = 1;
+    var rect;
+    if (node.imageRects) {
+        rect = node.imageRects[s];
+    } else {
+        const y = node.imagey;
+        rect = [padding,y+padding,node.size[0]-2*padding,node.size[1]-y-2*padding];
+    }
+    ctx.strokeRect(rect[0]+padding, rect[1]+padding, rect[2]-padding*2, rect[3]-padding*2);
+}
+
+function additionalDrawBackground(node, ctx) {
+    if (!node.imgs) return;
+    if (node.imageRects) {
+        for (let i = 0; i < node.imgs.length; i++) {
+            // delete underlying image
+            ctx.fillStyle = "#000";
+            ctx.fillRect(...node.imageRects[i])
+            // draw the new one
+            const img = node.imgs[i];
+            const cellWidth = node.imageRects[i][2];
+            const cellHeight = node.imageRects[i][3];
+
+            let wratio = cellWidth/img.width;
+            let hratio = cellHeight/img.height;
+            var ratio = Math.min(wratio, hratio);
+
+            let imgHeight = ratio * img.height;
+            let imgWidth = ratio * img.width;
+
+            const imgX = node.imageRects[i][0] + (cellWidth - imgWidth)/2;
+            const imgY = node.imageRects[i][1] + (cellHeight - imgHeight)/2;
+            const cell_padding = 2;
+            ctx.drawImage(img, imgX+cell_padding, imgY+cell_padding, imgWidth-cell_padding*2, imgHeight-cell_padding*2);
+
+        }
+    }
+    ctx.lineWidth = 2;
+    ctx.strokeStyle = "green";
+    node?.selected?.forEach((s) => { drawRect(node,s, ctx) })
+    ctx.strokeStyle = "#F88";
+    node?.anti_selected?.forEach((s) => { drawRect(node,s, ctx) })
+}
+
+function click_is_in_image(node, pos) {
+    if (node.imgs?.length>1) {
+        for (var i = 0; i<node.imageRects.length; i++) {
+            const dx = pos[0] - node.imageRects[i][0];
+            const dy = pos[1] - node.imageRects[i][1];
+            if ( dx > 0 && dx < node.imageRects[i][2] &&
+                dy > 0 && dy < node.imageRects[i][3] ) {
+                    return i;
+                }
+        }
+    } else if (node.imgs?.length==1) {
+        if (pos[1]>node.imagey) return 0;
+    }
+    return -1;
+}
+
+export { display_preview_images, additionalDrawBackground, click_is_in_image }

web/js/image_chooser/prompt.js

@@ -0,0 +1,114 @@
+import { app } from "../../../../scripts/app.js";
+
+function links_with(p, node_id, down, up) {
+    const links_with = [];
+    p.workflow.links.forEach((l) => {
+        if (down && l[1]===node_id && !links_with.includes(l[3])) links_with.push(l[3])
+        if (up && l[3]===node_id && !links_with.includes(l[1])) links_with.push(l[1])
+    });
+    return links_with;
+}
+
+function _all_v_nodes(p, here_id) {
+    /*
+    Make a list of all downstream nodes.
+    */
+    const downstream = [];
+    const to_process = [here_id]
+    while(to_process.length>0) {
+        const id = to_process.pop();
+        downstream.push(id);
+        to_process.push(
+            ...links_with(p,id,true,false).filter((nid)=>{
+                return !(downstream.includes(nid) || to_process.includes(nid))
+            })
+        )
+    }
+
+    /*
+    Now all upstream nodes from any of the downstream nodes (except us).
+    Put us on the result list so we don't flow up through us
+    */
+    to_process.push(...downstream.filter((n)=>{ return n!=here_id}));
+    const back_upstream = [here_id];
+    while(to_process.length>0) {
+        const id = to_process.pop();
+        back_upstream.push(id);
+        to_process.push(
+            ...links_with(p,id,false,true).filter((nid)=>{
+                return !(back_upstream.includes(nid) || to_process.includes(nid))
+            })
+        )
+    }
+
+    const keep = [];
+    keep.push(...downstream);
+    keep.push(...back_upstream.filter((n)=>{return !keep.includes(n)}));
+
+    console.log(`Nodes to keep: ${keep}`);
+    return keep;
+}
+
+async function all_v_nodes(here_id) {
+    const p = structuredClone(await app.graphToPrompt());
+    const all_nodes = [];
+    p.workflow.nodes.forEach((node)=>{all_nodes.push(node.id)})
+    p.workflow.links = p.workflow.links.filter((l)=>{ return (all_nodes.includes(l[1]) && all_nodes.includes(l[3]))} )
+    return _all_v_nodes(p,here_id);
+}
+
+async function restart_from_here(here_id, go_down_to_chooser=false) {
+    const p = structuredClone(await app.graphToPrompt());
+    /*
+    Make a list of all nodes, and filter out links that are no longer valid
+    */
+    const all_nodes = [];
+    p.workflow.nodes.forEach((node)=>{all_nodes.push(node.id)})
+    p.workflow.links = p.workflow.links.filter((l)=>{ return (all_nodes.includes(l[1]) && all_nodes.includes(l[3]))} )
+
+    /* Move downstream to a chooser */
+    if (go_down_to_chooser) {
+        while (!app.graph._nodes_by_id[here_id].isChooser) {
+            here_id = links_with(p, here_id, true, false)[0];
+        }
+    }
+
+    const keep = _all_v_nodes(p, here_id);
+
+    /*
+    Filter p.workflow.nodes and p.workflow.links
+    */
+    p.workflow.nodes = p.workflow.nodes.filter((node) => {
+        if (node.id===here_id) node.inputs.forEach((i)=>{i.link=null})  // remove our upstream links
+        return (keep.includes(node.id))                                 // only keep keepers
+    })
+    p.workflow.links = p.workflow.links.filter((l) => {return (keep.includes(l[1]) && keep.includes(l[3]))})
+
+    /*
+    Filter the p.output object to only include nodes we're keeping
+    */
+    const new_output = {}
+    for (const [key, value] of Object.entries(p.output)) {
+        if (keep.includes(parseInt(key))) new_output[key] = value;
+    }
+    /*
+    Filter the p.output entry for the start node to remove any list (ie link) inputs
+    */
+    const new_inputs = {};
+    for (const [key, value] of Object.entries(new_output[here_id.toString()].inputs)) {
+        if (!Array.isArray(value)) new_inputs[key] = value;
+    }
+    new_output[here_id.toString()].inputs = new_inputs;
+
+    p.output = new_output;
+
+    // temporarily hijack graph_to_prompt with a version that restores the old one but returns this prompt
+    const gtp_was = app.graphToPrompt;
+    app.graphToPrompt = () => {
+        app.graphToPrompt = gtp_was;
+        return p;
+    }
+    app.queuePrompt(0);
+}
+
+export { restart_from_here, all_v_nodes }

web/js/image_chooser/state.js

@@ -0,0 +1,26 @@
+import { app } from "../../../../scripts/app.js";
+
+export class FlowState {
+    constructor(){}
+    static idle() {
+        return (!app.runningNodeId);
+    }
+    static paused() {
+        return true;
+    }
+    static paused_here(node_id) {
+        return (FlowState.paused() && FlowState.here(node_id))
+    }
+    static running() {
+        return (!FlowState.idle());
+    }
+    static here(node_id) {
+        return (app.runningNodeId==node_id);
+    }
+    static state() {
+        if (FlowState.paused()) return "Paused";
+        if (FlowState.running()) return "Running";
+        return "Idle";
+    }
+    static cancelling = false;
+}