Spaces:
Running
on
Zero
Running
on
Zero
| 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); | |
| }; | |
| } | |
| }, | |
| }); | |