Allow custom words

py/autocomplete.py

@@ -0,0 +1,19 @@
+from server import PromptServer
+from aiohttp import web
+import os 
+
+dir = os.path.abspath(os.path.join(__file__, "../../user"))
+if not os.path.exists(dir):
+    os.mkdir(dir)
+file = os.path.join(dir, "autocomplete.txt")
+
+@PromptServer.instance.routes.get("/pysssss/autocomplete")
+async def get_autocomplete(request):
+    if os.path.isfile(file):
+        return web.FileResponse(file)
+    return web.Response(status=404)
+
+@PromptServer.instance.routes.post("/pysssss/autocomplete")
+async def update_autocomplete(request):
+    with open(file, "w") as f:
+        f.write(request.text())

web/js/autocompleter.js

@@ -41,7 +41,24 @@ app.registerExtension({
 			TextAreaAutoComplete.updateWords("pysssss.embeddings", words);
 		}
 
-		addEmbeddings();
+		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();
+							p[n] = { text: n };
+							return p;
+						}, {})
+					);
+				}
+			}
+		}
+
+		Promise.all([addEmbeddings(), addCustomWords()]);
 
 		const STRING = ComfyWidgets.STRING;
 		ComfyWidgets.STRING = function (node, inputName, inputData) {

web/js/common/autocomplete.js

@@ -341,23 +341,19 @@ export class TextAreaAutoComplete {
 			switch (e.key) {
 				case "ArrowUp":
 					e.preventDefault();
-					this.selected.el.classList.remove("pysssss-autocomplete-item--selected");
 					if (this.selected.index) {
 						this.#setSelected(this.currentWords[this.selected.index - 1]);
 					} else {
 						this.#setSelected(this.currentWords[this.currentWords.length - 1]);
 					}
-					this.selected.el.classList.add("pysssss-autocomplete-item--selected");
 					break;
 				case "ArrowDown":
 					e.preventDefault();
-					this.selected.el.classList.remove("pysssss-autocomplete-item--selected");
 					if (this.selected.index === this.currentWords.length - 1) {
 						this.#setSelected(this.currentWords[0]);
 					} else {
 						this.#setSelected(this.currentWords[this.selected.index + 1]);
 					}
-					this.selected.el.classList.add("pysssss-autocomplete-item--selected");
 					break;
 				case "Tab":
 					e.preventDefault();