Alex Furer
4cb03b45dc
- Added "Show Text" node documentation - Added a video about the "Edit Generated Prompt" node - Updated the workflow example to use my own "Show Text" node
146 lines
6.2 KiB
JavaScript
146 lines
6.2 KiB
JavaScript
import { app } from "/scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "AF.PromptNodes",
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "AF_Edit_Generated_Prompt") {
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
const r = onExecuted ? onExecuted.apply(this, arguments) : undefined;
|
|
|
|
// Update generated prompt display
|
|
let displayText = "";
|
|
if (message.ui && message.ui.generated_prompt) {
|
|
displayText = message.ui.generated_prompt[0] || "";
|
|
} else if (message.generated_prompt) {
|
|
displayText = message.generated_prompt[0] || "";
|
|
}
|
|
|
|
if (displayText) {
|
|
let displayWidget = this.widgets?.find(w => w.name === "generated_prompt");
|
|
if (displayWidget) {
|
|
displayWidget.value = displayText;
|
|
this.setDirtyCanvas(true, true);
|
|
}
|
|
}
|
|
|
|
return r;
|
|
};
|
|
}
|
|
|
|
// Update the onExecuted handler for Show Text node
|
|
if (nodeData.name === "AF_Show_Text") {
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
const r = onExecuted ? onExecuted.apply(this, arguments) : undefined;
|
|
|
|
// Update text display
|
|
let displayText = "";
|
|
if (message.ui && message.ui.text) {
|
|
displayText = message.ui.text[0] || "";
|
|
} else if (message.text) {
|
|
displayText = message.text[0] || "";
|
|
}
|
|
|
|
if (displayText) {
|
|
let displayWidget = this.widgets?.find(w => w.name === "text");
|
|
if (displayWidget) {
|
|
displayWidget.value = displayText;
|
|
this.setDirtyCanvas(true, true);
|
|
}
|
|
}
|
|
|
|
return r;
|
|
};
|
|
}
|
|
},
|
|
|
|
async nodeCreated(node, app) {
|
|
if (node.comfyClass === "AF_Edit_Generated_Prompt") {
|
|
// Wait for widgets to be properly initialized
|
|
setTimeout(() => {
|
|
// Make generated_prompt readonly
|
|
const displayWidget = node.widgets?.find(w => w.name === "generated_prompt");
|
|
if (displayWidget) {
|
|
displayWidget.options = displayWidget.options || {};
|
|
displayWidget.options.readonly = true;
|
|
|
|
if (displayWidget.inputEl) {
|
|
displayWidget.inputEl.readOnly = true;
|
|
displayWidget.inputEl.style.cursor = "text";
|
|
displayWidget.inputEl.style.color = "#888888"; // Gray text
|
|
displayWidget.inputEl.style.opacity = "0.7"; // Slightly transparent
|
|
|
|
displayWidget.inputEl.addEventListener('keydown', function(e) {
|
|
const allowedKeys = [
|
|
'ArrowUp', 'ArrowDown', 'ArrowLeft', 'ArrowRight',
|
|
'Home', 'End', 'PageUp', 'PageDown'
|
|
];
|
|
|
|
const isCtrlKey = e.ctrlKey || e.metaKey;
|
|
const allowedCtrlKeys = ['a', 'c', 'x'];
|
|
|
|
if (allowedKeys.includes(e.key) ||
|
|
(isCtrlKey && allowedCtrlKeys.includes(e.key.toLowerCase()))) {
|
|
return;
|
|
}
|
|
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
});
|
|
}
|
|
}
|
|
|
|
// Add some spacer widgets for padding - FIX: Add empty callback to prevent console warning
|
|
const spacerTop = node.addWidget("text", "spacer_top", "", () => {});
|
|
spacerTop.computeSize = () => [0, 10]; // Empty widget with 10px height
|
|
spacerTop.draw = () => {}; // Don't draw anything
|
|
|
|
// Add copy button and immediately set its properties
|
|
const copyButton = node.addWidget("button", "copy_button", "", () => {
|
|
const sourceWidget = node.widgets?.find(w => w.name === "generated_prompt");
|
|
const targetWidget = node.widgets?.find(w => w.name === "manual_or_paste_generated");
|
|
|
|
if (sourceWidget && targetWidget && sourceWidget.value) {
|
|
targetWidget.value = sourceWidget.value;
|
|
if (targetWidget.inputEl) {
|
|
targetWidget.inputEl.value = sourceWidget.value;
|
|
}
|
|
if (targetWidget.callback) {
|
|
targetWidget.callback(targetWidget.value, targetWidget, node);
|
|
}
|
|
node.setDirtyCanvas(true, true);
|
|
}
|
|
});
|
|
|
|
// Set the button label after creation
|
|
copyButton.label = " Copy Generated Prompt for Editing "; // Add spaces for padding
|
|
copyButton.name = "copy_button";
|
|
|
|
// Add bottom spacer - FIX: Add empty callback to prevent console warning
|
|
const spacerBottom = node.addWidget("text", "spacer_bottom", "", () => {});
|
|
spacerBottom.computeSize = () => [0, 10]; // Empty widget with 10px height
|
|
spacerBottom.draw = () => {}; // Don't draw anything
|
|
|
|
}, 50);
|
|
}
|
|
|
|
// Handle Show Text node creation
|
|
if (node.comfyClass === "AF_Show_Text") {
|
|
setTimeout(() => {
|
|
const displayWidget = node.widgets?.find(w => w.name === "text");
|
|
if (displayWidget?.inputEl) {
|
|
displayWidget.inputEl.readOnly = true;
|
|
displayWidget.inputEl.style.cursor = "text";
|
|
}
|
|
|
|
// ComfyUI sets new nodes with size around [140, 70]
|
|
// If it's that small, it's probably a brand new node
|
|
if (node.size && node.size[0] <= 150 && node.size[1] <= 100) {
|
|
node.size = [500, 300]; // Default size for new nodes only
|
|
}
|
|
}, 50);
|
|
}
|
|
}
|
|
}); |