Add persistent any text preview node

2026-06-26 09:32:11 +02:00
parent 49c6ee77a6
commit a3371afe0c
4 changed files with 159 additions and 0 deletions
@@ -0,0 +1,46 @@
+import { app } from "../../scripts/app.js";
+import { api } from "../../scripts/api.js";
+
+const EXTENSION = "ethanfel.prompt_builder.preview_any_text";
+const NODE_NAME = "SxCPPreviewAnyAsText";
+
+function isPreviewTextNode(node) {
+    return node?.comfyClass === NODE_NAME || node?.type === NODE_NAME;
+}
+
+function getNodeById(id) {
+    return app.graph?.getNodeById?.(Number(id)) || app.graph?._nodes_by_id?.[id] || app.graph?._nodes_by_id?.[Number(id)];
+}
+
+function widget(node, name) {
+    return node.widgets?.find((item) => item.name === name);
+}
+
+function outputPreviewText(output) {
+    const raw = output?.preview_text;
+    if (Array.isArray(raw)) return raw[0] ?? "";
+    return raw ?? "";
+}
+
+function setPreviewWidget(node, text) {
+    const preview = widget(node, "preview_text");
+    if (!preview) return;
+    preview.value = text;
+    if (preview.inputEl) preview.inputEl.value = text;
+    preview.callback?.(text, app.canvas, node);
+    node.setDirtyCanvas?.(true, true);
+    app.graph?.setDirtyCanvas?.(true, true);
+}
+
+app.registerExtension({
+    name: EXTENSION,
+
+    async setup() {
+        api.addEventListener("executed", ({detail}) => {
+            const node = getNodeById(detail?.display_node ?? detail?.node);
+            if (!isPreviewTextNode(node)) return;
+            const text = outputPreviewText(detail?.output || {});
+            setPreviewWidget(node, text);
+        });
+    },
+});