Spaces:
Runtime error
Runtime error
import { app } from "../../../scripts/app.js"; | |
import { ComfyWidgets } from "../../../scripts/widgets.js"; | |
// Displays output caption text | |
app.registerExtension({ | |
name: "Img2TxtNode", | |
async beforeRegisterNodeDef(nodeType, nodeData, app) { | |
if (nodeData.name === "img2txt BLIP/Llava Multimodel Tagger") { | |
function populate(message) { | |
console.log("message", message); | |
console.log("message.text", message.text); | |
const insertIndex = this.widgets.findIndex((w) => w.name === "output_text"); | |
if (insertIndex !== -1) { | |
for (let i = insertIndex; i < this.widgets.length; i++) { | |
this.widgets[i].onRemove?.(); | |
} | |
this.widgets.length = insertIndex; | |
} | |
const outputWidget = ComfyWidgets["STRING"]( | |
this, | |
"output_text", | |
["STRING", { multiline: true }], | |
app | |
).widget; | |
outputWidget.inputEl.readOnly = true; | |
outputWidget.inputEl.style.opacity = 0.6; | |
outputWidget.value = message.text.join(""); | |
requestAnimationFrame(() => { | |
const size_ = this.computeSize(); | |
if (size_[0] < this.size[0]) { | |
size_[0] = this.size[0]; | |
} | |
if (size_[1] < this.size[1]) { | |
size_[1] = this.size[1]; | |
} | |
this.onResize?.(size_); | |
app.graph.setDirtyCanvas(true, false); | |
}); | |
} | |
const onExecuted = nodeType.prototype.onExecuted; | |
nodeType.prototype.onExecuted = function (message) { | |
onExecuted?.apply(this, arguments); | |
populate.call(this, message); | |
}; | |
} | |
}, | |
}); | |