47 lines
1.4 KiB
JavaScript
47 lines
1.4 KiB
JavaScript
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);
|
|
});
|
|
},
|
|
});
|