add PromptImage

__init__.py

@@ -503,7 +503,7 @@ PromptServer.add_routes=new_add_routes
 
 
 # 导入节点
-from .nodes.PromptNode import RandomPrompt,PromptSlide,PromptSimplification
+from .nodes.PromptNode import RandomPrompt,PromptSlide,PromptSimplification,PromptImage
 from .nodes.ImageNode import NoiseImage,TransparentImage,GradientImage,LoadImagesFromPath,LoadImagesFromURL,ResizeImage,TextImage,SvgImage,Image3D,ShowLayer,NewLayer,MergeLayers,AreaToMask,SmoothMask,FeatheredMask,SplitLongMask,ImageCropByAlpha,EnhanceImage,FaceToMask
 from .nodes.Vae import VAELoader,VAEDecode
 from .nodes.ScreenShareNode import ScreenShareNode,FloatingVideo
@@ -521,6 +521,7 @@ NODE_CLASS_MAPPINGS = {
     "RandomPrompt":RandomPrompt,
     "PromptSlide":PromptSlide,
     "PromptSimplification":PromptSimplification,
+    "PromptImage":PromptImage,
     "ClipInterrogator":ClipInterrogator,
     "NoiseImage":NoiseImage,
     "GradientImage":GradientImage,

nodes/PromptNode.py

@@ -1,9 +1,11 @@
 import random
 import comfy.utils
-import json
+import os
+import numpy as np
 from urllib import request, parse
-
-
+import folder_paths
+from PIL import Image, ImageOps,ImageFilter,ImageEnhance,ImageDraw,ImageSequence, ImageFont
+from PIL.PngImagePlugin import PngInfo
 # def queue_prompt(prompt_workflow):
 #     p = {"prompt": prompt_workflow}
 #     data = json.dumps(p).encode('utf-8')
@@ -45,6 +47,9 @@ default_prompt1='''Swing
 default_prompt1="\n".join([p.strip() for p in default_prompt1.split('\n') if p.strip()!=''])
 
 
+def tensor2pil(image):
+    return Image.fromarray(np.clip(255. * image.cpu().numpy().squeeze(), 0, 255).astype(np.uint8))
+
 def addWeight(text, weight=1):
     if weight == 1:
         return text
@@ -78,6 +83,73 @@ def prompt_delete_words(sentence, new_words_length):
 # result = prompt_delete_words(sentence, new_words_length)
 # print(result)
 
+class PromptImage:
+    def __init__(self):
+        self.temp_dir = folder_paths.get_temp_directory()
+        self.type = "temp"
+        self.prefix_append = "PromptImage"
+        self.compress_level = 4
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompts": ("STRING", 
+                         {
+                            "multiline": True, 
+                            "default": '',
+                            "dynamicPrompts": False
+                          }),
+
+                "images": ("IMAGE",{"default": None}), 
+                 "save_to_image": (["enable", "disable"],),
+                }
+            }
+    
+    RETURN_TYPES = ()
+   
+    OUTPUT_NODE = True
+
+    INPUT_IS_LIST = True
+
+    FUNCTION = "run"
+
+    CATEGORY = "♾️Mixlab/prompt"
+
+    # 运行的函数
+    def run(self,prompts,images,save_to_image):
+        filename_prefix="mixlab_"
+        filename_prefix += self.prefix_append
+        full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
+            filename_prefix, self.temp_dir, images[0].shape[1], images[0].shape[0])
+        results = list()
+
+        save_to_image=save_to_image[0]=='enable'
+
+        for index in range(len(images)):
+            image=images[index]
+            img=tensor2pil(image)
+           
+            metadata = None
+            if save_to_image:
+                metadata = PngInfo()
+                prompt_text=prompts[index]
+                if prompt_text is not None:
+                    metadata.add_text("prompt_text", prompt_text)
+                
+            file = f"{filename}_{index}_{counter:05}_.png"
+            img.save(os.path.join(full_output_folder, file), pnginfo=metadata, compress_level=self.compress_level)
+            results.append({
+                "filename": file,
+                "subfolder": subfolder,
+                "type": self.type
+            })
+            counter += 1
+
+        return { "ui": { "_images": results,"prompts":prompts } }
+
+
+
 
 class PromptSimplification:
     @classmethod

web/javascript/prompt_mixlab.js

@@ -132,8 +132,8 @@ app.registerExtension({
           inp.click()
           inp.addEventListener('change', event => {
             // 获取选择的文件
-            const file = event.target.files[0];
-            this.title=file.name.split('.')[0];
+            const file = event.target.files[0]
+            this.title = file.name.split('.')[0]
 
             // console.log(file.name.split('.')[0])
             // 创建文件读取器
@@ -202,3 +202,74 @@ app.registerExtension({
     }
   }
 })
+
+app.registerExtension({
+  name: 'Mixlab.prompt.PromptImage',
+  async beforeRegisterNodeDef (nodeType, nodeData, app) {
+    if (nodeType.comfyClass == 'PromptImage') {
+      const orig_nodeCreated = nodeType.prototype.onNodeCreated
+      nodeType.prototype.onNodeCreated = function () {
+        orig_nodeCreated?.apply(this, arguments)
+        console.log('#orig_nodeCreated', this)
+        const widget = {
+          type: 'div',
+          name: 'result',
+          draw (ctx, node, widget_width, y, widget_height) {
+            Object.assign(this.div.style, {
+              ...get_position_style(ctx, widget_width, y, node.size[1]),
+              flexWrap: 'wrap',
+              justifyContent: 'flex-start'
+            })
+          }
+        }
+
+        widget.div = $el('div', {})
+
+        document.body.appendChild(widget.div)
+
+        this.addCustomWidget(widget)
+
+        const onRemoved = this.onRemoved
+        this.onRemoved = () => {
+          widget.div.remove()
+          return onRemoved?.()
+        }
+
+        // this.serialize_widgets = true //需要保存参数
+      }
+
+      const onExecuted = nodeType.prototype.onExecuted
+      nodeType.prototype.onExecuted = function (message) {
+        onExecuted?.apply(this, arguments)
+        console.log('PromptImage', message.prompts, message._images)
+        // window._mixlab_app_json = message.json
+        try {
+          let widget = this.widgets.filter(w => w.name === 'result')[0]
+
+          widget.div.innerHTML = ``
+
+          for (let index = 0; index < message._images.length; index++) {
+            const img = message._images[index]
+            let url = api.apiURL(
+              `/view?filename=${encodeURIComponent(img.filename)}&type=${
+                img.type
+              }&subfolder=${
+                img.subfolder
+              }${app.getPreviewFormatParam()}${app.getRandParam()}`
+            )
+
+            // 创建card
+            let div = document.createElement('div')
+            div.style = `width: 150px;`
+            div.innerHTML = `<img src="${url}" style='width: 100%'/><p style="margin: 0;
+            font-size: 12px;
+            position: relative;
+            margin-top: -35px;
+            background: #8080808f;">${message.prompts[index]}</p>`
+            widget.div.appendChild(div)
+          }
+        } catch (error) {}
+      }
+    }
+  }
+})