38 lines
1.6 KiB
JavaScript
38 lines
1.6 KiB
JavaScript
import {app} from "../../scripts/app.js";
|
|
import {ComfyWidgets} from "../../scripts/widgets.js";
|
|
|
|
// Create a read-only string widget
|
|
function createWidget(app, node, widgetName, type) {
|
|
const widget = ComfyWidgets[type](node, widgetName, ["STRING", {multiline: true}], app).widget;
|
|
widget.inputEl.readOnly = true;
|
|
widget.inputEl.style.textAlign = "center";
|
|
widget.inputEl.style.fontSize = "0.75rem";
|
|
return widget;
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "sd_prompt_reader.extractorDisplay",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "SDParameterExtractor") {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const result = onNodeCreated?.apply(this, arguments);
|
|
|
|
// Create widgets
|
|
const value_display = createWidget(app, this, "value_display", "STRING");
|
|
};
|
|
|
|
// Update widgets
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
onExecuted?.apply(this, arguments);
|
|
this.widgets.find(obj => obj.name === "value_display").value = message.text[1]
|
|
this.widgets.find(obj => obj.name === "parameter").options.values = message.text[0]
|
|
if (this.widgets.find(obj => obj.name === "parameter").value === "parameters not loaded") {
|
|
this.widgets.find(obj => obj.name === "parameter").value = message.text[0][0]
|
|
}
|
|
};
|
|
}
|
|
},
|
|
}); |