diff --git a/.github/workflows/package_wheels.yml b/.github/workflows/package_wheels.yml
index d2a9cd1..4f7eba5 100644
--- a/.github/workflows/package_wheels.yml
+++ b/.github/workflows/package_wheels.yml
@@ -35,9 +35,7 @@ jobs:
- name: 📦 Building and Bundling wheels
shell: bash
run: |
- python -m pip wheel --no-cache-dir -r requirements-wheels.txt -w ./wheels > build.log
-
- cat build.log
+ python -m pip wheel --no-cache-dir --no-deps -r requirements-wheels.txt -w ./wheels 2>&1 | tee build.log
# find source wheels
packages=$(cat build.log | awk -F 'Building wheels for collected packages: ' '{print $2}')
diff --git a/.gitignore b/.gitignore
index 80053fe..d5bec60 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
__pycache__
*.py[cod]
*.onnx
+wheels/
node_modules/
\ No newline at end of file
diff --git a/INSTALL-CN.md b/INSTALL-CN.md
new file mode 100644
index 0000000..934e540
--- /dev/null
+++ b/INSTALL-CN.md
@@ -0,0 +1,93 @@
+# 安装
+- [安装](#安装)
+ - [自动安装（推荐）](#自动安装推荐)
+ - [ComfyUI 管理器](#comfyui-管理器)
+ - [虚拟环境](#虚拟环境)
+ - [模型下载](#模型下载)
+ - [网络扩展](#网络扩展)
+ - [旧的安装方法 (MANUAL)](#旧的安装方法-manual)
+ - [依赖关系](#依赖关系)
+### 自动安装（推荐）
+
+### ComfyUI 管理器
+
+从 0.1.0 版开始，该扩展将使用 [ComfyUI-Manager](https://github.com/ltdrdata/ComfyUI-Manager) 进行安装，这对处理各种环境下的各种安装问题大有帮助。
+
+### 虚拟环境
+还有一种试验性的单行安装方法，即在 ComfyUI 根目录下使用以下命令进行安装。它将下载代码、安装依赖项并运行安装脚本：
+
+
+```bash
+curl -sSL "https://raw.githubusercontent.com/username/repo/main/install.py" | python3 -
+```
+
+## 模型下载
+某些节点需要下载额外的模型，您可以使用与上述相同的 python 环境以交互方式完成下载：
+
+```bash
+python scripts/download_models.py
+```
+
+然后根据提示或直接按回车键下载每个模型。
+
+> **Note**
+> 您可以使用以下方法下载所有型号，无需提示：
+ ```bash
+ python scripts/download_models.py -y
+ ```
+
+#### 网络扩展
+
+首次运行时，脚本会尝试将 [网络扩展](https://github.com/melMass/comfy_mtb/tree/main/web)链接到你的 "web/extensions "文件夹，[请参阅](https://github.com/melMass/comfy_mtb/blob/d982b69a58c05ccead9c49370764beaa4549992a/__init__.py#L45-L61)。
+
+Custom Virtualenv（我主要用这个） Comfy 便携式/单机版（来自 ComfyUI 版本） Google Colab カスタム Virtualenv (私は主にこれを使っています) Comfy-portable / standalone (ComfyUI リリースより) Google Colab Custom Virtualenv (I use this mainly) Comfy-portable / standalone (from ComfyUI releases) Google Colab MTB Debug Status: {'Enabled' if enabled else 'Disabled'}
+ """
+ return web.Response(
+ text=endpoint.render_base_template("Debug", html_response),
+ content_type="text/html",
+ )
+
+ # Return JSON for other requests
+ return web.json_response({"enabled": enabled})
+
+
+# - WAS Dictionary
+MANIFEST = {
+ "name": "MTB Nodes", # The title that will be displayed on Node Class menu,. and Node Class view
+ "version": (0, 1, 0), # Version of the custom_node or sub module
+ "author": "Mel Massadian", # Author or organization of the custom_node or sub module
+ "project": "https://github.com/melMass/comfy_mtb", # The address that the `name` value will link to on Node Class Views
+ "description": "Set of nodes that enhance your animation workflow and provide a range of useful tools including features such as manipulating bounding boxes, perform color corrections, swap faces in images, interpolate frames for smooth animation, export to ProRes format, apply various image operations, work with latent spaces, generate QR codes, and create normal and height maps for textures.",
+}
diff --git a/endpoint.py b/endpoint.py
new file mode 100644
index 0000000..1da1315
--- /dev/null
+++ b/endpoint.py
@@ -0,0 +1,39 @@
+from .utils import here
+
+
+def render_base_template(title, content):
+ css_content = ""
+ css_path = here / "html" / "style.css"
+ if css_path:
+ with open(css_path, "r") as css_file:
+ css_content = css_file.read()
+
+ github_icon_svg = """{title}
+
+
+
+
+
+
+ {content}
+
+
+
+
+
+ """
diff --git a/examples/01-faceswap.json b/examples/01-faceswap.json
new file mode 100644
index 0000000..605697a
--- /dev/null
+++ b/examples/01-faceswap.json
@@ -0,0 +1,1012 @@
+{
+ "last_node_id": 76,
+ "last_link_id": 161,
+ "nodes": [
+ {
+ "id": 59,
+ "type": "Reroute",
+ "pos": [
+ -150.35178124999982,
+ 644.4360633544919
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 12,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 124
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "VAE",
+ "links": [
+ 139
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 56,
+ "type": "Reroute",
+ "pos": [
+ -1580.8297949218763,
+ 644.7740239257807
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 8,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 117
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "VAE",
+ "links": [
+ 124
+ ]
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 57,
+ "type": "Reroute",
+ "pos": [
+ -673.8297949218747,
+ -185.22597607421872
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 11,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 135
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "MODEL",
+ "links": [
+ 120
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 65,
+ "type": "Reroute",
+ "pos": [
+ -1512.8297949218763,
+ -181.22597607421872
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 6,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 134
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "MODEL",
+ "links": [
+ 135
+ ]
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 7,
+ "type": "CLIPTextEncode",
+ "pos": [
+ -834.8297949218747,
+ 206.77402392578134
+ ],
+ "size": [
+ 210,
+ 54
+ ],
+ "flags": {},
+ "order": 10,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 142
+ },
+ {
+ "name": "text",
+ "type": "STRING",
+ "link": 87,
+ "widget": {
+ "name": "text",
+ "config": [
+ "STRING",
+ {
+ "multiline": true
+ }
+ ]
+ },
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CONDITIONING",
+ "type": "CONDITIONING",
+ "links": [
+ 6
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPTextEncode"
+ },
+ "widgets_values": [
+ "worst quality, hands, embedding:EasyNegative,"
+ ]
+ },
+ {
+ "id": 5,
+ "type": "EmptyLatentImage",
+ "pos": [
+ -913.8297949218747,
+ 326.77402392578125
+ ],
+ "size": [
+ 315,
+ 106
+ ],
+ "flags": {},
+ "order": 0,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "LATENT",
+ "type": "LATENT",
+ "links": [
+ 2
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "EmptyLatentImage"
+ },
+ "widgets_values": [
+ 768,
+ 512,
+ 1
+ ]
+ },
+ {
+ "id": 6,
+ "type": "CLIPTextEncode",
+ "pos": [
+ -1040,
+ -2
+ ],
+ "size": [
+ 422.84503173828125,
+ 164.31304931640625
+ ],
+ "flags": {},
+ "order": 7,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 141
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CONDITIONING",
+ "type": "CONDITIONING",
+ "links": [
+ 4
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPTextEncode"
+ },
+ "widgets_values": [
+ "Closeup portrait of an old bearded Caucasian man smiling, (NYC 1995), trench coat, golden ring, brown eyes"
+ ]
+ },
+ {
+ "id": 16,
+ "type": "CheckpointLoaderSimple",
+ "pos": [
+ -2001,
+ 193
+ ],
+ "size": [
+ 315,
+ 98
+ ],
+ "flags": {},
+ "order": 1,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "MODEL",
+ "type": "MODEL",
+ "links": [
+ 134
+ ],
+ "slot_index": 0
+ },
+ {
+ "name": "CLIP",
+ "type": "CLIP",
+ "links": [
+ 141,
+ 142
+ ],
+ "slot_index": 1
+ },
+ {
+ "name": "VAE",
+ "type": "VAE",
+ "links": [
+ 117
+ ],
+ "slot_index": 2
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CheckpointLoaderSimple"
+ },
+ "widgets_values": [
+ "revAnimated_v122.safetensors"
+ ]
+ },
+ {
+ "id": 3,
+ "type": "KSampler",
+ "pos": [
+ -483,
+ -21
+ ],
+ "size": [
+ 315,
+ 474
+ ],
+ "flags": {},
+ "order": 13,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "model",
+ "type": "MODEL",
+ "link": 120
+ },
+ {
+ "name": "positive",
+ "type": "CONDITIONING",
+ "link": 4
+ },
+ {
+ "name": "negative",
+ "type": "CONDITIONING",
+ "link": 6
+ },
+ {
+ "name": "latent_image",
+ "type": "LATENT",
+ "link": 2
+ }
+ ],
+ "outputs": [
+ {
+ "name": "LATENT",
+ "type": "LATENT",
+ "links": [
+ 138
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "KSampler"
+ },
+ "widgets_values": [
+ 542821171532982,
+ "fixed",
+ 45,
+ 8,
+ "dpmpp_sde",
+ "simple",
+ 1
+ ]
+ },
+ {
+ "id": 66,
+ "type": "VAEDecodeTiled",
+ "pos": [
+ 205,
+ -28
+ ],
+ "size": [
+ 210,
+ 46
+ ],
+ "flags": {
+ "collapsed": false
+ },
+ "order": 14,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "samples",
+ "type": "LATENT",
+ "link": 138
+ },
+ {
+ "name": "vae",
+ "type": "VAE",
+ "link": 139,
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 153,
+ 161
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "VAEDecodeTiled"
+ }
+ },
+ {
+ "id": 25,
+ "type": "PreviewImage",
+ "pos": [
+ 2338.9229387812507,
+ -632.8275743593749
+ ],
+ "size": [
+ 726.28564453125,
+ 475.3432312011719
+ ],
+ "flags": {},
+ "order": 19,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 150
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "PreviewImage"
+ }
+ },
+ {
+ "id": 75,
+ "type": "PreviewImage",
+ "pos": [
+ 1573.5946044921875,
+ -629.80322265625
+ ],
+ "size": [
+ 691.7459716796875,
+ 479.6098327636719
+ ],
+ "flags": {},
+ "order": 18,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 156
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "PreviewImage"
+ }
+ },
+ {
+ "id": 70,
+ "type": "Restore Face (mtb)",
+ "pos": [
+ 1899,
+ -51
+ ],
+ "size": [
+ 315,
+ 150
+ ],
+ "flags": {},
+ "order": 17,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "image",
+ "type": "IMAGE",
+ "link": 149
+ },
+ {
+ "name": "model",
+ "type": "FACEENHANCE_MODEL",
+ "link": 151,
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 150
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Restore Face (mtb)"
+ },
+ "widgets_values": [
+ "false",
+ "false",
+ 0.5,
+ "true"
+ ]
+ },
+ {
+ "id": 67,
+ "type": "Face Swap (mtb)",
+ "pos": [
+ 1127,
+ -26
+ ],
+ "size": [
+ 315,
+ 122
+ ],
+ "flags": {},
+ "order": 16,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "image",
+ "type": "IMAGE",
+ "link": 161
+ },
+ {
+ "name": "reference",
+ "type": "IMAGE",
+ "link": 160,
+ "slot_index": 1
+ },
+ {
+ "name": "faceswap_model",
+ "type": "FACESWAP_MODEL",
+ "link": 147,
+ "slot_index": 2
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 149,
+ 156
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Face Swap (mtb)"
+ },
+ "widgets_values": [
+ "0",
+ "false"
+ ]
+ },
+ {
+ "id": 71,
+ "type": "Load Face Enhance Model (mtb)",
+ "pos": [
+ 1476,
+ 240
+ ],
+ "size": [
+ 315,
+ 82
+ ],
+ "flags": {},
+ "order": 9,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "bg_upsampler",
+ "type": "UPSCALE_MODEL",
+ "link": 152,
+ "slot_index": 0
+ }
+ ],
+ "outputs": [
+ {
+ "name": "model",
+ "type": "FACEENHANCE_MODEL",
+ "links": [
+ 151
+ ],
+ "shape": 3
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Load Face Enhance Model (mtb)"
+ },
+ "widgets_values": [
+ "GFPGANv1.4.pth",
+ 2
+ ]
+ },
+ {
+ "id": 72,
+ "type": "UpscaleModelLoader",
+ "pos": [
+ 1110,
+ 241
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 2,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "UPSCALE_MODEL",
+ "type": "UPSCALE_MODEL",
+ "links": [
+ 152
+ ],
+ "shape": 3
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "UpscaleModelLoader"
+ },
+ "widgets_values": [
+ "4x-UltraSharp.pth"
+ ]
+ },
+ {
+ "id": 73,
+ "type": "PreviewImage",
+ "pos": [
+ 760,
+ -630
+ ],
+ "size": [
+ 671.1859741210938,
+ 483.1548156738281
+ ],
+ "flags": {},
+ "order": 15,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 153
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "PreviewImage"
+ }
+ },
+ {
+ "id": 76,
+ "type": "Load Image From Url (mtb)",
+ "pos": [
+ 161,
+ 231
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 3,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 160
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Load Image From Url (mtb)"
+ },
+ "widgets_values": [
+ "https://lucasmuseum.org/assets/general/Lucas_Headshot_Color_web.jpg"
+ ]
+ },
+ {
+ "id": 69,
+ "type": "Load Face Swap Model (mtb)",
+ "pos": [
+ 172,
+ 503
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 4,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "FACESWAP_MODEL",
+ "type": "FACESWAP_MODEL",
+ "links": [
+ 147
+ ],
+ "shape": 3
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Load Face Swap Model (mtb)"
+ },
+ "widgets_values": [
+ "inswapper_128.onnx"
+ ]
+ },
+ {
+ "id": 20,
+ "type": "Styles Loader (mtb)",
+ "pos": [
+ -1341,
+ 276
+ ],
+ "size": [
+ 315,
+ 78
+ ],
+ "flags": {
+ "collapsed": false
+ },
+ "order": 5,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "positive",
+ "type": "STRING",
+ "links": [],
+ "shape": 3
+ },
+ {
+ "name": "negative",
+ "type": "STRING",
+ "links": [
+ 87
+ ],
+ "shape": 3,
+ "slot_index": 1
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Styles Loader (mtb)"
+ },
+ "widgets_values": [
+ "❌Low Token"
+ ]
+ }
+ ],
+ "links": [
+ [
+ 2,
+ 5,
+ 0,
+ 3,
+ 3,
+ "LATENT"
+ ],
+ [
+ 4,
+ 6,
+ 0,
+ 3,
+ 1,
+ "CONDITIONING"
+ ],
+ [
+ 6,
+ 7,
+ 0,
+ 3,
+ 2,
+ "CONDITIONING"
+ ],
+ [
+ 87,
+ 20,
+ 1,
+ 7,
+ 1,
+ "STRING"
+ ],
+ [
+ 117,
+ 16,
+ 2,
+ 56,
+ 0,
+ "*"
+ ],
+ [
+ 120,
+ 57,
+ 0,
+ 3,
+ 0,
+ "MODEL"
+ ],
+ [
+ 124,
+ 56,
+ 0,
+ 59,
+ 0,
+ "*"
+ ],
+ [
+ 134,
+ 16,
+ 0,
+ 65,
+ 0,
+ "*"
+ ],
+ [
+ 135,
+ 65,
+ 0,
+ 57,
+ 0,
+ "*"
+ ],
+ [
+ 138,
+ 3,
+ 0,
+ 66,
+ 0,
+ "LATENT"
+ ],
+ [
+ 139,
+ 59,
+ 0,
+ 66,
+ 1,
+ "VAE"
+ ],
+ [
+ 141,
+ 16,
+ 1,
+ 6,
+ 0,
+ "CLIP"
+ ],
+ [
+ 142,
+ 16,
+ 1,
+ 7,
+ 0,
+ "CLIP"
+ ],
+ [
+ 147,
+ 69,
+ 0,
+ 67,
+ 2,
+ "FACESWAP_MODEL"
+ ],
+ [
+ 149,
+ 67,
+ 0,
+ 70,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 150,
+ 70,
+ 0,
+ 25,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 151,
+ 71,
+ 0,
+ 70,
+ 1,
+ "FACEENHANCE_MODEL"
+ ],
+ [
+ 152,
+ 72,
+ 0,
+ 71,
+ 0,
+ "UPSCALE_MODEL"
+ ],
+ [
+ 153,
+ 66,
+ 0,
+ 73,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 156,
+ 67,
+ 0,
+ 75,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 160,
+ 76,
+ 0,
+ 67,
+ 1,
+ "IMAGE"
+ ],
+ [
+ 161,
+ 66,
+ 0,
+ 67,
+ 0,
+ "IMAGE"
+ ]
+ ],
+ "groups": [
+ {
+ "title": "Txt2Img",
+ "bounding": [
+ -2061,
+ -234,
+ 1932,
+ 973
+ ],
+ "color": "#a1309b",
+ "locked": false
+ },
+ {
+ "title": "Save Intermediate Image",
+ "bounding": [
+ 147,
+ -152,
+ 303,
+ 213
+ ],
+ "color": "#3f789e",
+ "locked": false
+ },
+ {
+ "title": "SWAP & RESTORED",
+ "bounding": [
+ 2305,
+ -741,
+ 789,
+ 638
+ ],
+ "color": "#3f789e",
+ "locked": false
+ },
+ {
+ "title": "SWAP",
+ "bounding": [
+ 1520,
+ -743,
+ 774,
+ 642
+ ],
+ "color": "#3f789e",
+ "locked": false
+ },
+ {
+ "title": "SD OUTPUT",
+ "bounding": [
+ 655,
+ -745,
+ 854,
+ 648
+ ],
+ "color": "#3f789e",
+ "locked": false
+ }
+ ],
+ "config": {},
+ "extra": {},
+ "version": 0.4
+}
\ No newline at end of file
diff --git a/examples/02-film_interpolation.json b/examples/02-film_interpolation.json
new file mode 100644
index 0000000..218de4f
--- /dev/null
+++ b/examples/02-film_interpolation.json
@@ -0,0 +1,1349 @@
+{
+ "last_node_id": 82,
+ "last_link_id": 165,
+ "nodes": [
+ {
+ "id": 59,
+ "type": "Reroute",
+ "pos": [
+ -670,
+ 980
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 15,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 124
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "VAE",
+ "links": [
+ 139,
+ 154
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 56,
+ "type": "Reroute",
+ "pos": [
+ -2050,
+ 980
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 10,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 117
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "VAE",
+ "links": [
+ 124
+ ]
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 65,
+ "type": "Reroute",
+ "pos": [
+ -1990,
+ 160
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 8,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 134
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "MODEL",
+ "links": [
+ 135
+ ]
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 57,
+ "type": "Reroute",
+ "pos": [
+ -1200,
+ 150
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 12,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 135
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "MODEL",
+ "links": [
+ 120,
+ 143
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 5,
+ "type": "EmptyLatentImage",
+ "pos": [
+ -1410,
+ 660
+ ],
+ "size": [
+ 315,
+ 106
+ ],
+ "flags": {},
+ "order": 0,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "LATENT",
+ "type": "LATENT",
+ "links": [
+ 2,
+ 153
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "EmptyLatentImage"
+ },
+ "widgets_values": [
+ 768,
+ 512,
+ 1
+ ]
+ },
+ {
+ "id": 67,
+ "type": "Text",
+ "pos": [
+ -2280,
+ 250
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 1,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "STRING",
+ "type": "STRING",
+ "links": [
+ 142,
+ 164
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Text"
+ },
+ "widgets_values": [
+ "Photo of a Face of a man (looking down), rim lighting, tokyo 1987"
+ ]
+ },
+ {
+ "id": 3,
+ "type": "KSampler",
+ "pos": [
+ -1010,
+ 320
+ ],
+ "size": [
+ 315,
+ 474
+ ],
+ "flags": {},
+ "order": 17,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "model",
+ "type": "MODEL",
+ "link": 120
+ },
+ {
+ "name": "positive",
+ "type": "CONDITIONING",
+ "link": 4
+ },
+ {
+ "name": "negative",
+ "type": "CONDITIONING",
+ "link": 6
+ },
+ {
+ "name": "latent_image",
+ "type": "LATENT",
+ "link": 2
+ },
+ {
+ "name": "seed",
+ "type": "INT",
+ "link": 148,
+ "widget": {
+ "name": "seed",
+ "config": [
+ "INT",
+ {
+ "default": 0,
+ "min": 0,
+ "max": 18446744073709552000
+ }
+ ]
+ }
+ }
+ ],
+ "outputs": [
+ {
+ "name": "LATENT",
+ "type": "LATENT",
+ "links": [
+ 138
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "KSampler"
+ },
+ "widgets_values": [
+ 542821171532978,
+ "fixed",
+ 45,
+ 8,
+ "euler_ancestral",
+ "simple",
+ 1
+ ]
+ },
+ {
+ "id": 72,
+ "type": "PrimitiveNode",
+ "pos": [
+ -1290,
+ 880
+ ],
+ "size": [
+ 210,
+ 82
+ ],
+ "flags": {},
+ "order": 2,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "INT",
+ "type": "INT",
+ "links": [
+ 148,
+ 150
+ ],
+ "widget": {
+ "name": "seed",
+ "config": [
+ "INT",
+ {
+ "default": 0,
+ "min": 0,
+ "max": 18446744073709552000
+ }
+ ]
+ },
+ "slot_index": 0
+ }
+ ],
+ "title": "seed",
+ "properties": {},
+ "widgets_values": [
+ 542821171532978,
+ "fixed"
+ ]
+ },
+ {
+ "id": 79,
+ "type": "Load Film Model (mtb)",
+ "pos": [
+ 720,
+ 590
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 3,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "FILM_MODEL",
+ "type": "FILM_MODEL",
+ "links": [
+ 161
+ ],
+ "shape": 3
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Load Film Model (mtb)"
+ },
+ "widgets_values": [
+ "Style"
+ ]
+ },
+ {
+ "id": 80,
+ "type": "PreviewImage",
+ "pos": [
+ 1120,
+ 190
+ ],
+ "size": [
+ 210,
+ 246
+ ],
+ "flags": {},
+ "order": 23,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 163,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "PreviewImage"
+ }
+ },
+ {
+ "id": 71,
+ "type": "CLIPTextEncode",
+ "pos": [
+ -610,
+ 400
+ ],
+ "size": [
+ 210,
+ 75.28300476074219
+ ],
+ "flags": {},
+ "order": 16,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 152,
+ "slot_index": 0
+ },
+ {
+ "name": "text",
+ "type": "STRING",
+ "link": 151,
+ "widget": {
+ "name": "text",
+ "config": [
+ "STRING",
+ {
+ "multiline": true
+ }
+ ]
+ }
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CONDITIONING",
+ "type": "CONDITIONING",
+ "links": [
+ 144
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPTextEncode"
+ },
+ "widgets_values": [
+ "Face of a man (looking down), rim lighting, tokyo 1987"
+ ]
+ },
+ {
+ "id": 19,
+ "type": "CLIPSetLastLayer",
+ "pos": [
+ -1980,
+ 400
+ ],
+ "size": [
+ 315,
+ 58
+ ],
+ "flags": {},
+ "order": 9,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 116
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CLIP",
+ "type": "CLIP",
+ "links": [
+ 28,
+ 29,
+ 152
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPSetLastLayer"
+ },
+ "widgets_values": [
+ -2
+ ]
+ },
+ {
+ "id": 6,
+ "type": "CLIPTextEncode",
+ "pos": [
+ -1320,
+ 330
+ ],
+ "size": [
+ 210,
+ 54
+ ],
+ "flags": {},
+ "order": 13,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 28
+ },
+ {
+ "name": "text",
+ "type": "STRING",
+ "link": 142,
+ "widget": {
+ "name": "text",
+ "config": [
+ "STRING",
+ {
+ "multiline": true
+ }
+ ]
+ }
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CONDITIONING",
+ "type": "CONDITIONING",
+ "links": [
+ 4
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPTextEncode"
+ },
+ "widgets_values": [
+ "Face of a man (looking down), rim lighting, tokyo 1987"
+ ]
+ },
+ {
+ "id": 7,
+ "type": "CLIPTextEncode",
+ "pos": [
+ -1320,
+ 430
+ ],
+ "size": [
+ 210,
+ 54
+ ],
+ "flags": {},
+ "order": 14,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "clip",
+ "type": "CLIP",
+ "link": 29
+ },
+ {
+ "name": "text",
+ "type": "STRING",
+ "link": 87,
+ "widget": {
+ "name": "text",
+ "config": [
+ "STRING",
+ {
+ "multiline": true
+ }
+ ]
+ },
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "CONDITIONING",
+ "type": "CONDITIONING",
+ "links": [
+ 6,
+ 145
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CLIPTextEncode"
+ },
+ "widgets_values": [
+ "worst quality, hands, embedding:EasyNegative,"
+ ]
+ },
+ {
+ "id": 20,
+ "type": "Styles Loader (mtb)",
+ "pos": [
+ -1650,
+ 500
+ ],
+ "size": [
+ 285.32098388671875,
+ 78
+ ],
+ "flags": {
+ "collapsed": false
+ },
+ "order": 4,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "positive",
+ "type": "STRING",
+ "links": [],
+ "shape": 3
+ },
+ {
+ "name": "negative",
+ "type": "STRING",
+ "links": [
+ 87
+ ],
+ "shape": 3,
+ "slot_index": 1
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Styles Loader (mtb)"
+ },
+ "widgets_values": [
+ "❌Low Token"
+ ]
+ },
+ {
+ "id": 81,
+ "type": "Reroute",
+ "pos": [
+ -1740,
+ -10
+ ],
+ "size": [
+ 75,
+ 26
+ ],
+ "flags": {},
+ "order": 7,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "",
+ "type": "*",
+ "link": 164
+ }
+ ],
+ "outputs": [
+ {
+ "name": "",
+ "type": "STRING",
+ "links": [
+ 165
+ ]
+ }
+ ],
+ "properties": {
+ "showOutputText": false,
+ "horizontal": false
+ }
+ },
+ {
+ "id": 69,
+ "type": "String Replace (mtb)",
+ "pos": [
+ -1000,
+ 0
+ ],
+ "size": [
+ 315,
+ 82
+ ],
+ "flags": {},
+ "order": 11,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "string",
+ "type": "STRING",
+ "link": 165
+ }
+ ],
+ "outputs": [
+ {
+ "name": "STRING",
+ "type": "STRING",
+ "links": [
+ 151
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "String Replace (mtb)"
+ },
+ "widgets_values": [
+ "(looking down)",
+ "(looking up)"
+ ]
+ },
+ {
+ "id": 75,
+ "type": "VAEDecodeTiled",
+ "pos": [
+ 220,
+ 620
+ ],
+ "size": [
+ 210,
+ 46
+ ],
+ "flags": {
+ "collapsed": false
+ },
+ "order": 20,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "samples",
+ "type": "LATENT",
+ "link": 155,
+ "slot_index": 0
+ },
+ {
+ "name": "vae",
+ "type": "VAE",
+ "link": 154,
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 158
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "VAEDecodeTiled"
+ }
+ },
+ {
+ "id": 78,
+ "type": "Film Interpolation (mtb)",
+ "pos": [
+ 1140,
+ 520
+ ],
+ "size": [
+ 315,
+ 78
+ ],
+ "flags": {},
+ "order": 22,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 160,
+ "slot_index": 0
+ },
+ {
+ "name": "film_model",
+ "type": "FILM_MODEL",
+ "link": 161,
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 162
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Film Interpolation (mtb)"
+ },
+ "widgets_values": [
+ 4
+ ]
+ },
+ {
+ "id": 70,
+ "type": "KSampler",
+ "pos": [
+ -300,
+ 380
+ ],
+ "size": [
+ 315,
+ 474
+ ],
+ "flags": {},
+ "order": 18,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "model",
+ "type": "MODEL",
+ "link": 143
+ },
+ {
+ "name": "positive",
+ "type": "CONDITIONING",
+ "link": 144
+ },
+ {
+ "name": "negative",
+ "type": "CONDITIONING",
+ "link": 145
+ },
+ {
+ "name": "latent_image",
+ "type": "LATENT",
+ "link": 153,
+ "slot_index": 3
+ },
+ {
+ "name": "seed",
+ "type": "INT",
+ "link": 150,
+ "widget": {
+ "name": "seed",
+ "config": [
+ "INT",
+ {
+ "default": 0,
+ "min": 0,
+ "max": 18446744073709552000
+ }
+ ]
+ },
+ "slot_index": 4
+ }
+ ],
+ "outputs": [
+ {
+ "name": "LATENT",
+ "type": "LATENT",
+ "links": [
+ 155
+ ],
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "KSampler"
+ },
+ "widgets_values": [
+ 542821171532978,
+ "fixed",
+ 45,
+ 8,
+ "euler_ancestral",
+ "simple",
+ 1
+ ]
+ },
+ {
+ "id": 16,
+ "type": "CheckpointLoaderSimple",
+ "pos": [
+ -2470,
+ 530
+ ],
+ "size": [
+ 315,
+ 98
+ ],
+ "flags": {},
+ "order": 5,
+ "mode": 0,
+ "outputs": [
+ {
+ "name": "MODEL",
+ "type": "MODEL",
+ "links": [
+ 134
+ ],
+ "slot_index": 0
+ },
+ {
+ "name": "CLIP",
+ "type": "CLIP",
+ "links": [
+ 116
+ ],
+ "slot_index": 1
+ },
+ {
+ "name": "VAE",
+ "type": "VAE",
+ "links": [
+ 117
+ ],
+ "slot_index": 2
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "CheckpointLoaderSimple"
+ },
+ "widgets_values": [
+ "revAnimated_v122.safetensors"
+ ]
+ },
+ {
+ "id": 82,
+ "type": "Note",
+ "pos": [
+ -398,
+ 25
+ ],
+ "size": [
+ 500.61550625887776,
+ 270.54209761186075
+ ],
+ "flags": {},
+ "order": 6,
+ "mode": 0,
+ "properties": {
+ "text": "THIS LOOKS A BIT MESSY BUT WE BASICALLY USE THE SAME INPUTS (MODEL, SEED etc), ONLY STRING REPLACING THE PROMPT, LOOKING DOWN -> LOOKING UP"
+ },
+ "widgets_values": [
+ "THIS LOOKS A BIT MESSY BUT WE BASICALLY USE THE SAME INPUTS (MODEL, SEED etc), ONLY STRING REPLACING THE PROMPT, LOOKING DOWN -> LOOKING UP\n"
+ ],
+ "color": "#222",
+ "bgcolor": "#000"
+ },
+ {
+ "id": 66,
+ "type": "VAEDecodeTiled",
+ "pos": [
+ 231,
+ 403
+ ],
+ "size": [
+ 210,
+ 46
+ ],
+ "flags": {
+ "collapsed": false
+ },
+ "order": 19,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "samples",
+ "type": "LATENT",
+ "link": 138
+ },
+ {
+ "name": "vae",
+ "type": "VAE",
+ "link": 139,
+ "slot_index": 1
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 157
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "VAEDecodeTiled"
+ }
+ },
+ {
+ "id": 77,
+ "type": "Concat Images (mtb)",
+ "pos": [
+ 661,
+ 402
+ ],
+ "size": [
+ 210,
+ 46
+ ],
+ "flags": {},
+ "order": 21,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "imageA",
+ "type": "IMAGE",
+ "link": 157
+ },
+ {
+ "name": "imageB",
+ "type": "IMAGE",
+ "link": 158
+ }
+ ],
+ "outputs": [
+ {
+ "name": "IMAGE",
+ "type": "IMAGE",
+ "links": [
+ 160,
+ 163
+ ],
+ "shape": 3,
+ "slot_index": 0
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "Concat Images (mtb)"
+ }
+ },
+ {
+ "id": 76,
+ "type": "PreviewImage",
+ "pos": [
+ 1517,
+ 207
+ ],
+ "size": [
+ 511.33429483925056,
+ 675.8856963258129
+ ],
+ "flags": {},
+ "order": 24,
+ "mode": 0,
+ "inputs": [
+ {
+ "name": "images",
+ "type": "IMAGE",
+ "link": 162
+ }
+ ],
+ "properties": {
+ "Node name for S&R": "PreviewImage"
+ }
+ }
+ ],
+ "links": [
+ [
+ 2,
+ 5,
+ 0,
+ 3,
+ 3,
+ "LATENT"
+ ],
+ [
+ 4,
+ 6,
+ 0,
+ 3,
+ 1,
+ "CONDITIONING"
+ ],
+ [
+ 6,
+ 7,
+ 0,
+ 3,
+ 2,
+ "CONDITIONING"
+ ],
+ [
+ 28,
+ 19,
+ 0,
+ 6,
+ 0,
+ "CLIP"
+ ],
+ [
+ 29,
+ 19,
+ 0,
+ 7,
+ 0,
+ "CLIP"
+ ],
+ [
+ 87,
+ 20,
+ 1,
+ 7,
+ 1,
+ "STRING"
+ ],
+ [
+ 116,
+ 16,
+ 1,
+ 19,
+ 0,
+ "CLIP"
+ ],
+ [
+ 117,
+ 16,
+ 2,
+ 56,
+ 0,
+ "*"
+ ],
+ [
+ 120,
+ 57,
+ 0,
+ 3,
+ 0,
+ "MODEL"
+ ],
+ [
+ 124,
+ 56,
+ 0,
+ 59,
+ 0,
+ "*"
+ ],
+ [
+ 134,
+ 16,
+ 0,
+ 65,
+ 0,
+ "*"
+ ],
+ [
+ 135,
+ 65,
+ 0,
+ 57,
+ 0,
+ "*"
+ ],
+ [
+ 138,
+ 3,
+ 0,
+ 66,
+ 0,
+ "LATENT"
+ ],
+ [
+ 139,
+ 59,
+ 0,
+ 66,
+ 1,
+ "VAE"
+ ],
+ [
+ 142,
+ 67,
+ 0,
+ 6,
+ 1,
+ "STRING"
+ ],
+ [
+ 143,
+ 57,
+ 0,
+ 70,
+ 0,
+ "MODEL"
+ ],
+ [
+ 144,
+ 71,
+ 0,
+ 70,
+ 1,
+ "CONDITIONING"
+ ],
+ [
+ 145,
+ 7,
+ 0,
+ 70,
+ 2,
+ "CONDITIONING"
+ ],
+ [
+ 148,
+ 72,
+ 0,
+ 3,
+ 4,
+ "INT"
+ ],
+ [
+ 150,
+ 72,
+ 0,
+ 70,
+ 4,
+ "INT"
+ ],
+ [
+ 151,
+ 69,
+ 0,
+ 71,
+ 1,
+ "STRING"
+ ],
+ [
+ 152,
+ 19,
+ 0,
+ 71,
+ 0,
+ "CLIP"
+ ],
+ [
+ 153,
+ 5,
+ 0,
+ 70,
+ 3,
+ "LATENT"
+ ],
+ [
+ 154,
+ 59,
+ 0,
+ 75,
+ 1,
+ "VAE"
+ ],
+ [
+ 155,
+ 70,
+ 0,
+ 75,
+ 0,
+ "LATENT"
+ ],
+ [
+ 157,
+ 66,
+ 0,
+ 77,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 158,
+ 75,
+ 0,
+ 77,
+ 1,
+ "IMAGE"
+ ],
+ [
+ 160,
+ 77,
+ 0,
+ 78,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 161,
+ 79,
+ 0,
+ 78,
+ 1,
+ "FILM_MODEL"
+ ],
+ [
+ 162,
+ 78,
+ 0,
+ 76,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 163,
+ 77,
+ 0,
+ 80,
+ 0,
+ "IMAGE"
+ ],
+ [
+ 164,
+ 67,
+ 0,
+ 81,
+ 0,
+ "*"
+ ],
+ [
+ 165,
+ 81,
+ 0,
+ 69,
+ 0,
+ "STRING"
+ ]
+ ],
+ "groups": [
+ {
+ "title": "GENERATE IMAGES",
+ "bounding": [
+ -2477,
+ -219,
+ 3018,
+ 1350
+ ],
+ "color": "#a1309b",
+ "locked": false
+ },
+ {
+ "title": "FILM INTERPOLATION",
+ "bounding": [
+ 559,
+ 66,
+ 1264,
+ 832
+ ],
+ "color": "#3f789e",
+ "locked": false
+ }
+ ],
+ "config": {},
+ "extra": {},
+ "version": 0.4
+}
\ No newline at end of file
diff --git a/examples/README.md b/examples/README.md
new file mode 100644
index 0000000..23bf2e8
--- /dev/null
+++ b/examples/README.md
@@ -0,0 +1,12 @@
+# Examples
+All the examples use the [RevAnimated model 1.22](https://civitai.com/models/7371?modelVersionId=46846)
+## 01 Faceswap
+
+This example showcase the `Face Swap` & `Restore Face` nodes to replace the character with Georges Lucas's face.
+The face reference image is using the `Load Image From Url` node to avoid bundling input images.
+
+
+
+
+
+ `;
+ // set the class document wide if not present
+
+ shared.defineClass("bbox-input", `background-color: var(--comfy-input-bg);
+ color: var(--input-text);
+ overflow: hidden;
+ width:100%;
+ overflow-y: auto;
+ padding: 2px;
+ resize: none;
+ border: none;
+ box-sizing: border-box;
+ font-size: 10px;`)
+
+
+ const bboxInputs = widget.inputEl.querySelectorAll(".bbox-input");
+ bboxInputs.forEach((input) => {
+ input.addEventListener("change", widget.onInputChange.bind(this));
+ });
+
+ widget.desc = "Represents a Bounding Box with x, y, width, and height.";
+
+ document.body.appendChild(widget.inputEl);
+
+
+ console.log("Bounding Box Widget DOM", widget.inputEl)
+ return widget
+
+}
+/**
+ * @returns {import("./types/litegraph").IWidget} widget
+ */
+
+
+/**
+ * @returns {import("./types/litegraph").IWidget} widget
+ */
+
+
+
+// VIDEO: (node, inputName, inputData, app) => {
+// console.log("video")
+// const videoWidget = {
+// name: "VideoWidget",
+// description: "Video Player Widget",
+// value: inputData,
+// properties: {},
+// widget: null,
+
+// init: function () {
+// this.widget = document.createElement("video");
+// this.widget.width = 200;
+// this.widget.height = 120;
+// this.widget.controls = true;
+// this.widget.style.width = "100%";
+// this.widget.style.height = "100%";
+// this.widget.style.objectFit = "contain";
+// this.widget.style.backgroundColor = "black";
+// this.widget.style.pointerEvents = "none";
+// node.addWidget(inputName, videoWidget.widget, inputData);
+// },
+
+// setValue: function (value, options) {
+// if (value instanceof HTMLVideoElement) {
+// this.widget.src = value.src;
+// } else if (typeof value === "string") {
+// this.widget.src = value;
+// }
+// },
+
+// getValue: function () {
+// return this.widget.src;
+// },
+
+// append: function (parent) {
+// parent.appendChild(this.widget);
+// },
+
+// remove: function () {
+// this.widget.parentNode.removeChild(this.widget);
+// }
+// };
+// return {
+// widget: videoWidget,
+// }
+// }
+
+
+
+/**
+ * @returns {import("./types/comfy").ComfyExtension} extension
+ */
+const mtb_widgets = {
+ name: "mtb.widgets",
+
+ init: async () => {
+ console.log("Registering mtb.widgets")
+ try {
+
+ const res = await api.fetchApi('/mtb/debug')
+ const msg = await res.json()
+ window.MTB_DEBUG = msg.enabled;
+ }
+ catch (e) {
+ console.error('Error:', error);
+ }
+ },
+
+ setup: () => {
+ app.ui.settings.addSetting({
+ id: "mtb.Debug.enabled",
+ name: "[mtb] Enable Debug (py and js)",
+ type: "boolean",
+ defaultValue: false,
+
+ tooltip:
+ "This will enable debug messages in the console and in the python console respectively",
+ attrs: {
+ style: {
+ fontFamily: "monospace",
+ },
+ },
+ async onChange(value) {
+ if (value) {
+ console.log("Enabled DEBUG mode")
+ }
+ window.MTB_DEBUG = value;
+ await api.fetchApi('/mtb/debug', {
+ method: 'POST',
+ body: JSON.stringify({
+ enabled: value
+
+ })
+ }).then(response => { }).catch(error => {
+ console.error('Error:', error);
+ });
+
+ },
+ });
+ },
+
+
+ getCustomWidgets: function () {
+ return {
+ BOOL: (node, inputName, inputData, app) => {
+ console.debug("Registering bool")
+
+ return {
+ widget: node.addCustomWidget(MtbWidgets.BOOL(inputName, inputData[1]?.default || false)),
+ minWidth: 150,
+ minHeight: 30,
+ };
+ },
+
+ COLOR: (node, inputName, inputData, app) => {
+ console.debug("Registering color")
+ return {
+ widget: node.addCustomWidget(MtbWidgets.COLOR(inputName, inputData[1]?.default || "#ff0000")),
+ minWidth: 150,
+ minHeight: 30,
+ }
+ },
+ // BBOX: (node, inputName, inputData, app) => {
+ // console.debug("Registering bbox")
+ // return {
+ // widget: node.addCustomWidget(MtbWidgets.BBOX(inputName, inputData[1]?.default || [0, 0, 0, 0])),
+ // minWidth: 150,
+ // minHeight: 30,
+ // }
+
+ // }
+ }
+ },
+ /**
+ * @param {import("./types/comfy").NodeType} nodeType
+ * @param {import("./types/comfy").NodeDef} nodeData
+ * @param {import("./types/comfy").App} app
+ */
+ async beforeRegisterNodeDef(nodeType, nodeData, app) {
+
+ const rinputs = nodeData.input?.required;
+
+ let has_custom = false
+ if (nodeData.input && nodeData.input.required) {
+ for (const i of Object.keys(nodeData.input.required)) {
+ const input_type = nodeData.input.required[i][0]
+
+ if (newTypes.includes(input_type)) {
+ has_custom = true
+ break;
+ }
+ }
+ }
+ if (has_custom) {
+
+ //- Add widgets on node creation
+ const onNodeCreated = nodeType.prototype.onNodeCreated;
+ nodeType.prototype.onNodeCreated = function () {
+ const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
+ this.serialize_widgets = true;
+ for (const [key, input] of Object.entries(rinputs)) {
+ switch (input[0]) {
+ case "COLOR":
+ // const colW = colorWidget(key, input[1])
+ // this.addCustomWidget(colW)
+ // const associated_input = this.inputs.findIndex((i) => i.widget?.name === key);
+ // if (associated_input !== -1) {
+ // this.inputs[associated_input].widget = colW
+ // }
+
+
+
+ break;
+ case "BOOL":
+ // const widg = boolWidget(key, input[1])
+ // this.addCustomWidget(widg)
+ // this.addWidget("toggle", key, false, function (value, widget, node) {
+ // console.log(value)
+
+ // })
+ //this.removeInput(this.inputs.findIndex((i) => i.widget?.name === key));
+
+ break;
+ case "BBOX":
+ // const bboxW = bboxWidget(key, input[1])
+ // this.addCustomWidget(bboxW)
+ break;
+ default:
+ break
+ }
+
+
+ // }
+ }
+
+ this.setSize?.(this.computeSize())
+
+ this.onRemoved = function () {
+ // When removing this node we need to remove the input from the DOM
+ for (let y in this.widgets) {
+ if (this.widgets[y].canvas) {
+ this.widgets[y].canvas.remove();
+ }
+ }
+ };
+ }
+
+ //- Extra menus
+ const origGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
+ nodeType.prototype.getExtraMenuOptions = function (_, options) {
+ const r = origGetExtraMenuOptions ? origGetExtraMenuOptions.apply(this, arguments) : undefined;
+ if (this.widgets) {
+ let toInput = [];
+ let toWidget = [];
+ for (const w of this.widgets) {
+ if (w.type === shared.CONVERTED_TYPE) {
+ //- This is already handled by widgetinputs.js
+ // toWidget.push({
+ // content: `Convert ${w.name} to widget`,
+ // callback: () => shared.convertToWidget(this, w),
+ // });
+ } else if (newTypes.includes(w.type)) {
+ const config = nodeData?.input?.required[w.name] || nodeData?.input?.optional?.[w.name] || [w.type, w.options || {}];
+
+ toInput.push({
+ content: `Convert ${w.name} to input`,
+ callback: () => shared.convertToInput(this, w, config),
+ });
+ }
+ }
+ if (toInput.length) {
+ options.push(...toInput, null);
+ }
+
+ if (toWidget.length) {
+ options.push(...toWidget, null);
+ }
+ }
+
+ return r;
+ };
+
+ }
+
+ //- Extending Python Nodes
+ switch (nodeData.name) {
+ case "Psd Save (mtb)": {
+ // const onConnectionsChange = nodeType.prototype.onConnectionsChange;
+ nodeType.prototype.onConnectionsChange = function (type, index, connected, link_info) {
+ // const r = onConnectionsChange ? onConnectionsChange.apply(this, arguments) : undefined;
+ shared.dynamic_connection(this, index, connected)
+ }
+ break
+ }
+ case "Save Gif (mtb)": {
+ const onExecuted = nodeType.prototype.onExecuted;
+ nodeType.prototype.onExecuted = function (message) {
+ const r = onExecuted ? onExecuted.apply(this, message) : undefined;
+ console.log(message)
+ if (this.widgets) {
+ const pos = this.widgets.findIndex((w) => w.name === "anything_0");
+ if (pos !== -1) {
+ for (let i = pos; i < this.widgets.length; i++) {
+ console.log(this.widgets[i])
+ console.log(this)
+ this.widgets[i].onRemove?.();
+
+ }
+ this.widgets.length = pos
+
+
+ }
+
+ let imgURLs = []
+ if (message && message.gif) {
+ imgURLs = imgURLs.concat(message.gif.map(params => {
+ return api.apiURL("/view?" + new URLSearchParams(params).toString());
+ }))
+ console.log(imgURLs)
+ for (const img of imgURLs) {
+ const w = this.addCustomWidget(MtbWidgets.DEBUG_IMG(img, 0))
+ w.parent = this;
+ }
+ }
+ this.setSize?.(this.computeSize())
+ return r
+
+ }
+
+ const onRemoved = nodeType.prototype.onRemoved;
+ nodeType.prototype.onRemoved = function (message) {
+ const r = onRemoved ? onRemoved.apply(this, message) : undefined;
+ if (!this.widgets) return r
+ for (const w of this.widgets) {
+ if (w.canvas) {
+ w.canvas.remove();
+ }
+ w.onRemove?.()
+ w.onRemoved?.()
+ }
+ return r
+
+ }
+ }
+
+ break
+ }
+ case "Animation Builder (mtb)": {
+ // console.log(nodeType.prototype)
+
+
+ const onNodeCreated = nodeType.prototype.onNodeCreated;
+ nodeType.prototype.onNodeCreated = function () {
+ const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
+
+ this.changeMode(LiteGraph.ALWAYS)
+ // api.addEventListener("executed", ({ detail }) => {
+
+ // console.log("executed", detail)
+ // console.log(this)
+
+ // })
+ const raw_iteration = this.widgets.find((w) => w.name === "raw_iteration");
+ const raw_loop = this.widgets.find((w) => w.name === "raw_loop");
+
+
+ const total_frames = this.widgets.find((w) => w.name === "total_frames");
+ const loop_count = this.widgets.find((w) => w.name === "loop_count");
+
+ shared.hideWidgetForGood(this, raw_iteration);
+ shared.hideWidgetForGood(this, raw_loop);
+
+ raw_iteration._value = 0
+ // Object.defineProperty(raw_iteration, "value", {
+ // get() {
+ // return this._value
+ // },
+ // set(value) {
+ // this._value = value;
+ // },
+ // });
+
+ const value_preview = ComfyWidgets["STRING"](this, "PREVIEW_raw_iteration", ["STRING", { multiline: true }], app).widget;
+ value_preview.inputEl.readOnly = true;
+ value_preview.inputEl.disabled = true;
+
+
+ // value_preview.inputEl.style.opacity = 0.6;
+ value_preview.inputEl.style.textAlign = "center";
+ value_preview.inputEl.style.fontSize = "2.5em";
+ value_preview.inputEl.style.backgroundColor = "black";
+
+ value_preview.inputEl.style.setProperty("--comfy-input-bg", "transparent");
+ value_preview.inputEl.style.setProperty("background", "red", "important");
+ // remove the comfy-multiline-input class
+
+ // disallow user selection
+ value_preview.inputEl.style.userSelect = "none";
+
+ const loop_preview = ComfyWidgets["STRING"](this, "PREVIEW_raw_iteration", ["STRING", { multiline: true }], app).widget;
+ loop_preview.inputEl.readOnly = true;
+ loop_preview.inputEl.disabled = true;
+
+
+ // loop_preview.inputEl.style.opacity = 0.6;
+ loop_preview.inputEl.style.textAlign = "center";
+ loop_preview.inputEl.style.fontSize = "1.5em";
+ loop_preview.inputEl.style.backgroundColor = "black";
+
+ loop_preview.inputEl.style.setProperty("--comfy-input-bg", "transparent");
+ loop_preview.inputEl.style.setProperty("background", "red", "important");
+ // remove the comfy-multiline-input class
+
+ // disallow user selection
+ loop_preview.inputEl.style.userSelect = "none";
+
+ const onReset = () => {
+ raw_iteration.value = 0;
+ raw_loop.value = 0;
+
+ value_preview.value = 0;
+ loop_preview.value = 0;
+
+ app.canvas.setDirty(true);
+ }
+
+ const reset_button = this.addWidget("button", `Reset`, "reset", onReset);
+
+ const run_button = this.addWidget("button", `Queue`, "queue", () => {
+ onReset() // this could maybe be a setting or checkbox
+ app.queuePrompt(0, total_frames.value * loop_count.value)
+
+ });
+
+
+
+ raw_iteration.afterQueued = function () {
+ this.value++;
+ raw_loop.value = Math.floor(this.value / total_frames.value);
+ value_preview.value = `raw: ${this.value}
+frame: ${this.value % total_frames.value}`;
+ if (raw_loop.value + 1 > loop_count.value) {
+ loop_preview.value = `Done 😎!`
+ return
+ }
+
+ loop_preview.value = `current loop: ${raw_loop.value + 1}/${loop_count.value}`
+
+ }
+
+ return r
+
+ }
+ const onExecuted = nodeType.prototype.onExecuted;
+ nodeType.prototype.onExecuted = function (data) {
+ onExecuted?.apply(this, data)
+ if (this.widgets) {
+ const pos = this.widgets.findIndex((w) => w.name === "preview");
+ if (pos !== -1) {
+ for (let i = pos; i < this.widgets.length; i++) {
+ this.widgets[i].onRemove?.();
+ }
+ this.widgets.length = pos;
+ }
+ }
+
+ const w = ComfyWidgets["STRING"](this, "preview", ["STRING", { multiline: true }], app).widget;
+ w.inputEl.readOnly = true;
+ w.inputEl.style.opacity = 0.6;
+ w.value = data.total_frames;
+
+ // this.onResize?.(this.size);
+ this.setSize?.(this.computeSize())
+
+ }
+ // const onAfterExecuteNode = nodeType.prototype.onAfterExecuteNode;
+ // nodeType.prototype.onAfterExecuteNode = function () {
+ // onAfterExecuteNode?.apply(this)
+ // console.log("after", this)
+
+ // }
+ console.debug(`Registered ${nodeType.name} node extra events`)
+ break
+
+ }
+ default: {
+ break
+
+ }
+
+ }
+ // const onNodeCreated = nodeType.prototype.onNodeCreated;
+
+ // nodeType.prototype.onNodeCreated = function () {
+ // const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
+
+ // }
+
+
+
+
+ // console.log(nodeData.output)
+ // if (nodeData.output.includes("VIDEO") && nodeData.output_node) {
+ // console.log(`Found video output for ${nodeType}`)
+ // console.log(nodeData)
+
+ // }
+
+ // if (nodeData.name === "Psd Save (mtb)") {
+ // console.log(`Found psd node`)
+ // console.log(nodeData)
+
+ // }
+
+
+
+ }
+};
+
+
+app.registerExtension(mtb_widgets);
\ No newline at end of file
+
+欢迎使用 MTB Nodes 项目！这个代码库是开放的，您可以自由地探索和利用。它的主要目的是构建用于 [MLOPs](https://github.com/Bismuth-Consultancy-BV/MLOPs) 中的概念验证（POCs）。该项目中的许多节点都是受到现有社区贡献或内置功能的启发而创建的。
+
+在继续之前，请注意与此项目中使用的某些库相关的许可证。例如，`deepbump` 库采用 [GPLv3](https://github.com/HugoTini/DeepBump/blob/master/LICENSE) 许可证。
+
+- [MTB Nodes](#mtb-nodes)
+- [安装](#安装)
+- [节点列表](#节点列表)
+ - [bbox](#bbox)
+ - [colors](#colors)
+ - [人脸检测/交换](#人脸检测交换)
+ - [图像插值（动画）](#图像插值动画)
+ - [图像操作](#图像操作)
+ - [潜在变量工具](#潜在变量工具)
+ - [其他工具](#其他工具)
+ - [纹理](#纹理)
+- [Comfy 资源](#comfy-资源)
+
+
+## 安装
+
+- 移至 [INSTALL-CN.md](./INSTALL-CN.md)
+
+
+## 节点列表
+
+### bbox
+- `Bounding Box`: BBox 构造函数（自定义类型）
+- `BBox From Mask`: 从遮罩中提取边界框
+- `Crop`: 根据边界框裁剪图像
+- `Uncrop`: 根据边界框还原图像
+
+### colors
+- `Colored Image`: 给定尺寸的纯色图像
+- `RGB to HSV`: -
+- `HSV to RGB`: -
+- `Color Correct`: 基本颜色校正工具
+