Update artworks.yaml

artworks.yaml

@@ -1,9 +1,9 @@
 _id: artworks
 description: Image generation, mask extraction, etc.
-readme: ""
+readme: Added PuLID
 title: en=ArtWorks AI;ru=Узлы ArtWorks AI
 url: https://huggingface.co/PiperMy/Node-Packages/resolve/main/artworks.yaml
-version: 4
+version: 6
 nodes:
   image_to_image_artworks:
     _id: image_to_image_artworks
@@ -350,12 +350,15 @@ nodes:
     groups:
       inputs:
         SDXL:
-          order: 2
+          order: 3
           title: SDXL checkpoints
           description: Only for SDXL checkpoints
         base:
           order: 1
           title: en=Base parameters;ru=Базовые параметры
+        special:
+          order: 2
+          title: en=Special;ru=Специальные возможности
     inputs:
       checkpoint:
         order: 2
@@ -488,6 +491,13 @@ nodes:
         group: base
         type: integer
         placeholder: "-1"
+      controlNets:
+        order: 1
+        title: en=Control nets;ru=Контрольные сети
+        group: special
+        type: json
+        schema:
+          id: control-nets
     outputs:
       images:
         title: en=Images;ru=Изображения
@@ -560,9 +570,7 @@ nodes:
                       checkpoint,
                       cfgScale,
                       size,
-                      ...(performance !== "speed" ? {
-                          performance
-                      } : {}),
+                      performance: performance || "express",
                       ...(seed > 0 ? {
                           seed
                       } : {}),
@@ -1714,13 +1722,23 @@ nodes:
         type: image
         required: true
       labels:
-        order: 2
+        order: 3
         title: en=Labels;ru=Теги
         type: json
         required: true
         default: '{ "girl": "girl on photo", "male": "male on photo" }'
         schema:
           id: image-labels
+      model:
+        order: 2
+        title: en=Model;ru=Модель
+        type: string
+        required: true
+        default: clip
+        enum:
+          - clip
+          - siglip-1
+          - siglip-2
     outputs:
       labels:
         title: en=Labels;ru=Теги
@@ -1755,7 +1773,7 @@ nodes:
           });
 
           if (!state) {
-              const { image, labels } = inputs;
+              const { image, model, labels } = inputs;
 
               const payload = {
                   type: "classify-image",
@@ -1763,6 +1781,7 @@ nodes:
                   payload: {
                       base64: false,
                       image,
+                      model,
                       labels: Object.values(labels),
                   },
               };
@@ -2382,15 +2401,15 @@ nodes:
       OPEN_PAAS_USER:
         title: PaaS user
         type: string
-        scope: global
+        scope: user
       OPEN_PAAS_PASSWORD:
         title: PaaS password
         type: string
-        scope: global
+        scope: user
       PAAS_BASE_URL:
         title: PaaS base url
         type: string
-        scope: global
+        scope: user
     execution: regular
     groups:
       inputs:
@@ -2434,8 +2453,6 @@ nodes:
           - mixGEMAdam8witQegoow.NeWz.safetensors|MyMIX-G/Jem <SDXL> <Realistic>
           - mfcgPDXL_v10.safetensors|MFCG PDXL <SDXL>
           - aniku_0.2.fp16.safetensors|Anime Desire <SDXL> <Fantasy>
-          - flux1DevHyperNF4Flux1DevBNB_flux1DevHyperNF4.safetensors|Hyper NF4 Flux 1 <Flux>
-          - STOIQONewrealityFLUXSD_F1DPreAlpha.safetensors|New Reality Flux <Flux>
           - juggernautXL_v9Rundiffusionphoto2.safetensors|Juggernaut XL <SDXL>
           - animaPencilXL_v260.safetensors|Anima Pencil XL <SDXL> <Fantasy>
           - bluePencilXL_v401.safetensors|Blue Pencil XL <SDXL> <Fantasy>
@@ -2443,9 +2460,7 @@ nodes:
           - albedobaseXL_v20.safetensors|AlbedoBase XL <SDXL>
           - leosamsHelloworldXL_helloworldXL50GPT4V.safetensors|Helloworld XL <SDXL>
           - dynavisionXLAllInOneStylized_releaseV0610Bakedvae.safetensors|DynaVision XL All In One <SDXL>
-          - photon_v1.safetensors|Photon <SD 1.5>
           - afroditexlNudePeople_20Bkdvae.safetensors|Nude People <NSFW>
-          - nosft-float8-e4m3fn.safetensors|Flux Nude People <NSFW>
           - asdf_0.4a_lorapov_0.2_lust_0.4.fp16.safetensors|Real Porn <NSFW>
           - Sexy_Aesthetic_SDXL_0.4x0.2.fp16.safetensors|Sexy & Beauty <NSFW>
           - aniku_0.2.fp16.safetensors|Anime Desire <NSFW>
@@ -3330,3 +3345,494 @@ nodes:
     source: catalog
     title: en=Score aesthetics;ru=Оценить эстестику
     version: 1
+  dress_on_image:
+    _id: dress_on_image
+    arrange:
+      x: 210
+      y: 1100
+    category:
+      _id: deep_swap
+      title: en=Deep swap;ru=Глубокая замена
+    environment:
+      OPEN_PAAS_USER:
+        title: PaaS user
+        type: string
+        scope: global
+      OPEN_PAAS_PASSWORD:
+        title: PaaS password
+        type: string
+        scope: global
+      PAAS_BASE_URL:
+        title: PaaS base url
+        type: string
+        scope: global
+    execution: regular
+    inputs:
+      image:
+        order: 1
+        title: en=Image;ru=Изображение
+        type: image
+        required: true
+      prompt:
+        order: 3
+        title: en=Prompt;ru=Подсказка
+        type: string
+        required: true
+        multiline: true
+        default: RAW photo, red swimsuit
+      gender:
+        order: 2
+        title: en=Gender;ru=Пол
+        type: string
+        default: auto
+        enum:
+          - auto
+          - male
+          - female
+    outputs:
+      image:
+        title: en=Image;ru=Изображение
+        type: image
+    package: artworks
+    script: |-
+      const CHECK_TASK_INTERVAL = 3000;
+      const MAX_ATTEMPTS = 100;
+
+      export async function run({ inputs, state }) {
+
+          const { FatalError, TimeoutError, RepeatNode, NextNode } = DEFINITIONS;
+          const { ArtWorks, FatalError: ArtWorksError } = require('artworks');
+
+          const PAAS_BASE_URL = env.variables.get('PAAS_BASE_URL');
+          if (!PAAS_BASE_URL) {
+              throw new FatalError('Please, set PAAS_BASE_URL in environment');
+          }
+          const OPEN_PAAS_USER = env.variables.get('OPEN_PAAS_USER');
+          if (!OPEN_PAAS_USER) {
+              throw new FatalError('Please, set OPEN_PAAS_USER in environment');
+          }
+          const OPEN_PAAS_PASSWORD = env.variables.get('OPEN_PAAS_PASSWORD');
+          if (!OPEN_PAAS_PASSWORD) {
+              throw new FatalError('Please, set OPEN_PAAS_PASSWORD in environment');
+          }
+
+          const artworks = new ArtWorks({
+              baseUrl: PAAS_BASE_URL,
+              username: OPEN_PAAS_USER,
+              password: OPEN_PAAS_PASSWORD
+          });
+
+          if (!state) {
+              const { image, gender, prompt } = inputs;
+
+              const payload = {
+                  type: "dress-on-image",
+                  isFast: true,
+                  payload: {
+                      base64: false,
+                      image,
+                      gender,
+                      prompt
+                  },
+              };
+              try {
+                  const task = await artworks.createTask(payload);
+                  console.log(`Task created ${task}`);
+                  return RepeatNode.from({
+                      state: {
+                          payload,
+                          task,
+                          attempt: 0,
+                          startedAt: new Date()
+                      },
+                      progress: {
+                          total: MAX_ATTEMPTS,
+                          processed: 0
+                      },
+                      delay: 2000
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          } else {
+              const {
+                  payload,
+                  task,
+                  attempt,
+                  startedAt
+              } = state;
+
+              if (attempt > MAX_ATTEMPTS) {
+                  try {
+                      await artworks.cancelTask(task);
+                  } catch (e) { }
+
+                  const now = new Date();
+                  const time = (now - new Date(startedAt)) / 1000;
+                  throw new TimeoutError(`PaaS task for text to image ${task} timeout in ${time} sec`);
+              }
+
+              console.log(`Check task ${attempt} ${task}`);
+
+              try {
+                  const results = await artworks.checkState(task);
+                  if (!results) {
+                      return RepeatNode.from({
+                          delay: CHECK_TASK_INTERVAL,
+                          state: {
+                              payload,
+                              task,
+                              attempt: attempt + 1,
+                              startedAt,
+                          },
+                          progress: {
+                              total: MAX_ATTEMPTS,
+                              processed: attempt
+                          },
+                      });
+                  }
+                  let images = results.images.map((i) => i.url);
+                  const [image] = images;
+                  return NextNode.from({
+                      outputs: {
+                          image
+                      }
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          }
+      }
+    source: catalog
+    title: en=Dress on image;ru=Переодеть
+    version: 1
+  text_to_video:
+    _id: text_to_video
+    arrange:
+      x: 540
+      y: 1100
+    category:
+      _id: generate_videos
+      title: en=Generate videos;ru=Генерация видео
+    environment:
+      OPEN_PAAS_USER:
+        title: PaaS user
+        type: string
+        scope: global
+      OPEN_PAAS_PASSWORD:
+        title: PaaS password
+        type: string
+        scope: global
+      PAAS_BASE_URL:
+        title: PaaS base url
+        type: string
+        scope: global
+    execution: regular
+    inputs:
+      prompt:
+        order: 1
+        title: en=Prompt;ru=Подсказка
+        type: string
+        required: true
+        multiline: true
+        default: a sleek rocket launching from Earth
+      totalFrames:
+        order: 2
+        title: en=Total frames;ru=Кол-во кадров
+        type: integer
+        min: 32
+        max: 72
+        step: 4
+        default: 36
+      fps:
+        order: 3
+        title: FPS
+        type: integer
+        min: 12
+        max: 24
+        step: 2
+        default: 18
+    outputs:
+      video:
+        title: en=Video;ru=Видео
+        type: video
+    package: artworks
+    script: |-
+      const CHECK_TASK_INTERVAL = 5000;
+      const MAX_ATTEMPTS = 150;
+
+      export async function run({ inputs, state }) {
+
+          const { FatalError, TimeoutError, RepeatNode, NextNode } = DEFINITIONS;
+          const { ArtWorks, FatalError: ArtWorksError } = require('artworks');
+
+          const PAAS_BASE_URL = env.variables.get('PAAS_BASE_URL');
+          if (!PAAS_BASE_URL) {
+              throw new FatalError('Please, set PAAS_BASE_URL in environment');
+          }
+          const OPEN_PAAS_USER = env.variables.get('OPEN_PAAS_USER');
+          if (!OPEN_PAAS_USER) {
+              throw new FatalError('Please, set OPEN_PAAS_USER in environment');
+          }
+          const OPEN_PAAS_PASSWORD = env.variables.get('OPEN_PAAS_PASSWORD');
+          if (!OPEN_PAAS_PASSWORD) {
+              throw new FatalError('Please, set OPEN_PAAS_PASSWORD in environment');
+          }
+
+          const artworks = new ArtWorks({
+              baseUrl: PAAS_BASE_URL,
+              username: OPEN_PAAS_USER,
+              password: OPEN_PAAS_PASSWORD
+          });
+
+          if (!state) {
+              const { prompt, totalFrames, fps } = inputs;
+
+              const payload = {
+                  type: "text-to-video",
+                  isFast: true,
+                  payload: {
+                      base64: false,
+                      prompt,
+                      numFrames: totalFrames || 36,
+                      fps: fps || 18
+                  },
+              };
+              try {
+                  const task = await artworks.createTask(payload);
+                  console.log(`Task created ${task}`);
+                  return RepeatNode.from({
+                      state: {
+                          payload,
+                          task,
+                          attempt: 0,
+                          startedAt: new Date()
+                      },
+                      progress: {
+                          total: MAX_ATTEMPTS,
+                          processed: 0
+                      },
+                      delay: 2000
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          } else {
+              const {
+                  payload,
+                  task,
+                  attempt,
+                  startedAt
+              } = state;
+
+              if (attempt > MAX_ATTEMPTS) {
+                  try {
+                      await artworks.cancelTask(task);
+                  } catch (e) { }
+
+                  const now = new Date();
+                  const time = (now - new Date(startedAt)) / 1000;
+                  throw new TimeoutError(`PaaS task for text to image ${task} timeout in ${time} sec`);
+              }
+
+              console.log(`Check task ${attempt} ${task}`);
+
+              try {
+                  const results = await artworks.checkState(task);
+                  if (!results) {
+                      return RepeatNode.from({
+                          delay: CHECK_TASK_INTERVAL,
+                          state: {
+                              payload,
+                              task,
+                              attempt: attempt + 1,
+                              startedAt,
+                          },
+                          progress: {
+                              total: MAX_ATTEMPTS,
+                              processed: attempt
+                          },
+                      });
+                  }
+                  let { video: { url: video } } = results;
+                  return NextNode.from({
+                      outputs: {
+                          video
+                      }
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          }
+      }
+    source: catalog
+    title: en=Text to video;ru=Сгенерировать видео
+    version: 1
+  pulid:
+    _id: pulid
+    arrange:
+      x: 910
+      y: 1100
+    category:
+      _id: deep_swap
+      title: en=Deep swap;ru=Глубокая замена
+    environment:
+      OPEN_PAAS_USER:
+        title: PaaS user
+        type: string
+        scope: global
+      OPEN_PAAS_PASSWORD:
+        title: PaaS password
+        type: string
+        scope: global
+      PAAS_BASE_URL:
+        title: PaaS base url
+        type: string
+        scope: global
+    execution: regular
+    inputs:
+      person:
+        order: 1
+        title: en=Person;ru=Персона
+        type: image
+        required: true
+      prompt:
+        order: 2
+        title: en=Prompt;ru=Подсказка
+        type: string
+        required: true
+        multiline: true
+    outputs:
+      image:
+        title: ""
+        type: image
+    package: artworks
+    script: |-
+      const CHECK_TASK_INTERVAL = 5000;
+      const MAX_ATTEMPTS = 150;
+
+      export async function run({ inputs, state }) {
+
+          const { FatalError, TimeoutError, RepeatNode, NextNode } = DEFINITIONS;
+          const { ArtWorks, FatalError: ArtWorksError } = require('artworks');
+
+          const PAAS_BASE_URL = env.variables.get('PAAS_BASE_URL');
+          if (!PAAS_BASE_URL) {
+              throw new FatalError('Please, set PAAS_BASE_URL in environment');
+          }
+          const OPEN_PAAS_USER = env.variables.get('OPEN_PAAS_USER');
+          if (!OPEN_PAAS_USER) {
+              throw new FatalError('Please, set OPEN_PAAS_USER in environment');
+          }
+          const OPEN_PAAS_PASSWORD = env.variables.get('OPEN_PAAS_PASSWORD');
+          if (!OPEN_PAAS_PASSWORD) {
+              throw new FatalError('Please, set OPEN_PAAS_PASSWORD in environment');
+          }
+
+          const artworks = new ArtWorks({
+              baseUrl: PAAS_BASE_URL,
+              username: OPEN_PAAS_USER,
+              password: OPEN_PAAS_PASSWORD
+          });
+
+          if (!state) {
+              const { person, prompt } = inputs;
+
+              const payload = {
+                  type: 'run-comfy',
+                  isFast: true,
+                  payload: {
+                      pipeline: 'pullid',
+                      args: {
+                          image: person,
+                          prompt,
+                          aspect_ratio: '9:16',
+                      }
+                  },
+              };
+              try {
+                  const task = await artworks.createTask(payload);
+                  console.log(`Task created ${task}`);
+                  return RepeatNode.from({
+                      state: {
+                          payload,
+                          task,
+                          attempt: 0,
+                          startedAt: new Date()
+                      },
+                      progress: {
+                          total: MAX_ATTEMPTS,
+                          processed: 0
+                      },
+                      delay: 2000
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          } else {
+              const {
+                  payload,
+                  task,
+                  attempt,
+                  startedAt
+              } = state;
+
+              if (attempt > MAX_ATTEMPTS) {
+                  try {
+                      await artworks.cancelTask(task);
+                  } catch (e) { }
+
+                  const now = new Date();
+                  const time = (now - new Date(startedAt)) / 1000;
+                  throw new TimeoutError(`PaaS task for text to image ${task} timeout in ${time} sec`);
+              }
+
+              console.log(`Check task ${attempt} ${task}`);
+
+              try {
+                  const results = await artworks.checkState(task);
+                  if (!results) {
+                      return RepeatNode.from({
+                          delay: CHECK_TASK_INTERVAL,
+                          state: {
+                              payload,
+                              task,
+                              attempt: attempt + 1,
+                              startedAt,
+                          },
+                          progress: {
+                              total: MAX_ATTEMPTS,
+                              processed: attempt
+                          },
+                      });
+                  }
+                  const { images: [{ url: image }] } = results;
+                  return NextNode.from({
+                      outputs: { image }
+                  });
+              } catch (e) {
+                  if (e instanceof ArtWorksError) {
+                      throw new FatalError(e.message);
+                  }
+                  throw e;
+              }
+          }
+      }
+    source: catalog
+    title: PuLID
+    version: 1