84 lines
2.1 KiB
JavaScript
84 lines
2.1 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { ComfyWidgets } from "../../../scripts/widgets.js";
|
|
import { api } from "../../../../scripts/api.js";
|
|
import { $el } from "../../../../scripts/ui.js";
|
|
import { TextAreaAutoComplete } from "./common/autocomplete.js";
|
|
import { ModelInfoDialog } from "./common/modelInfoDialog.js";
|
|
|
|
class EmbeddingInfoDialog extends ModelInfoDialog {
|
|
async addInfo() {
|
|
super.addInfo();
|
|
const info = await this.addCivitaiInfo();
|
|
if (info) {
|
|
$el("div", {
|
|
parent: this.content,
|
|
innerHTML: info.description,
|
|
style: {
|
|
maxHeight: "250px",
|
|
overflow: "auto",
|
|
},
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "pysssss.AutoCompleter",
|
|
init() {
|
|
async function addEmbeddings() {
|
|
const embeddings = await api.getEmbeddings();
|
|
const words = {};
|
|
words["embedding:"] = { text: "embedding:" };
|
|
|
|
for (const emb of embeddings) {
|
|
const v = `embedding:${emb}`;
|
|
words[v] = {
|
|
text: v,
|
|
info: () => new EmbeddingInfoDialog(emb).show("embeddings", emb),
|
|
};
|
|
}
|
|
|
|
TextAreaAutoComplete.updateWords("pysssss.embeddings", words);
|
|
}
|
|
|
|
async function addCustomWords() {
|
|
const resp = await api.fetchApi("/pysssss/autocomplete", { cache: "no-store" });
|
|
if (resp.status === 200) {
|
|
const text = await resp.text();
|
|
if (text) {
|
|
TextAreaAutoComplete.updateWords(
|
|
"pysssss.customwords",
|
|
text.split("\n").reduce((p, n) => {
|
|
n = n.trim();
|
|
const pos = n.lastIndexOf(",");
|
|
let priority = undefined;
|
|
if (pos > -1) {
|
|
const v = parseInt(n.substr(pos + 1).trim());
|
|
if (!isNaN(v)) {
|
|
priority = v;
|
|
n = n.substr(0, pos).trim();
|
|
}
|
|
}
|
|
p[n] = { text: n, priority };
|
|
return p;
|
|
}, {})
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
Promise.all([addEmbeddings(), addCustomWords()]);
|
|
|
|
const STRING = ComfyWidgets.STRING;
|
|
ComfyWidgets.STRING = function (node, inputName, inputData) {
|
|
const r = STRING.apply(this, arguments);
|
|
|
|
if (inputData[1]?.multiline) {
|
|
new TextAreaAutoComplete(r.widget.inputEl);
|
|
}
|
|
|
|
return r;
|
|
};
|
|
},
|
|
});
|