简单优化 添加新的动物提示词

nodejs/prompt_combine.js

@@ -0,0 +1,59 @@
+import { app } from "../../../scripts/app.js";
+import { ComfyWidgets } from "../../../scripts/widgets.js";
+
+// Displays input text on a node
+
+app.registerExtension({
+	name: "PromptWrapper.nodes.CombinePrompt",
+	async beforeRegisterNodeDef(nodeType, nodeData, app) {
+		if (nodeData.name === "CombinePrompt") {
+			
+			const onExecuted = nodeType.prototype.onExecuted;
+			nodeType.prototype.onExecuted = function (message) {
+				onExecuted?.apply(this, arguments);
+
+				if (this.widgets) {
+					const pos = this.widgets.findIndex((w) => w.name === "text");
+					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, "result", ["STRING", { multiline: true }], app).widget;
+				w.inputEl.readOnly = true;
+				w.inputEl.style.opacity = 0.6;
+				w.value = message.text;
+
+				this.onResize?.(this.size);
+			};
+		}
+
+		if (nodeData.name === "MultiCombinePrompt") {
+			
+			const onExecuted = nodeType.prototype.onExecuted;
+			nodeType.prototype.onExecuted = function (message) {
+				onExecuted?.apply(this, arguments);
+
+				if (this.widgets) {
+					const pos = this.widgets.findIndex((w) => w.name === "text");
+					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, "result", ["STRING", { multiline: true }], app).widget;
+				w.inputEl.readOnly = true;
+				w.inputEl.style.opacity = 0.6;
+				w.value = message.text;
+
+				this.onResize?.(this.size);
+			};
+		}
+	},
+});

nodejs/prompt_preview.js

@@ -0,0 +1,76 @@
+import { app } from "../../../scripts/app.js";
+import { ComfyWidgets } from "../../../scripts/widgets.js";
+
+// Displays input text on a node
+
+app.registerExtension({
+	name: "PromptWrapper.nodes.PreviewPrompt",
+	async beforeRegisterNodeDef(nodeType, nodeData, app) {
+		if (nodeData.name === "PreviewPrompt") {
+			function update(text) {
+				if (this.widgets) {
+					// On older frontend versions there is a hidden converted-widget
+					const isConvertedWidget = +!!this.inputs?.[0].widget;
+					for (let i = isConvertedWidget; i < this.widgets.length; i++) {
+						this.widgets[i].onRemove?.();
+					}
+					this.widgets.length = isConvertedWidget;
+				}
+
+				const v = [...text];
+				if (!v[0]) {
+					v.shift();
+				}
+				for (let list of v) {
+					// Force list to be an array, not sure why sometimes it is/isn't
+					if (!(list instanceof Array)) list = [list];
+					for (const l of list) {
+						const w = ComfyWidgets["STRING"](this, "text_" + this.widgets?.length ?? 0, ["STRING", { multiline: true }], app).widget;
+						w.inputEl.readOnly = true;
+						w.inputEl.style.opacity = 0.6;
+						w.value = l;
+					}
+				}
+
+				requestAnimationFrame(() => {
+					const sz = this.computeSize();
+					if (sz[0] < this.size[0]) {
+						sz[0] = this.size[0];
+					}
+					if (sz[1] < this.size[1]) {
+						sz[1] = this.size[1];
+					}
+					this.onResize?.(sz);
+					app.graph.setDirtyCanvas(true, false);
+				});
+			}
+
+			// When the node is executed we will be sent the input text, display this in the widget
+			const onExecuted = nodeType.prototype.onExecuted;
+			nodeType.prototype.onExecuted = function (message) {
+				onExecuted?.apply(this, arguments);
+				update.call(this, message.text);
+			};
+
+			const VALUES = Symbol();
+			const configure = nodeType.prototype.configure;
+			nodeType.prototype.configure = function () {
+				// Store unmodified widget values as they get removed on configure by new frontend
+				this[VALUES] = arguments[0]?.widgets_values;
+				return configure?.apply(this, arguments);
+			};
+
+			const onConfigure = nodeType.prototype.onConfigure;
+			nodeType.prototype.onConfigure = function () {
+				onConfigure?.apply(this, arguments);
+				const widgets_values = this[VALUES];
+				if (widgets_values?.length) {
+					// In newer frontend there seems to be a delay in creating the initial widget
+					requestAnimationFrame(() => {
+						update.call(this, widgets_values.slice(+(widgets_values.length > 1 && this.inputs?.[0].widget)));
+					});
+				}
+			};
+		}
+	},
+});