Hugging Face's logo

saik0s
/
comfy_backup

GGUF
conversational
Model card Files
xet
 Community
comfy_backup
File size: 1,320 Bytes
33c25ce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import { app } from "../../../scripts/app.js";
import { ComfyWidgets } from "../../../scripts/widgets.js";

app.registerExtension({
	name: "pysssss.MathExpression",
	init() {
		const STRING = ComfyWidgets.STRING;
		ComfyWidgets.STRING = function (node, inputName, inputData) {
			const r = STRING.apply(this, arguments);
			r.widget.dynamicPrompts = inputData?.[1].dynamicPrompts;
			return r;
		};
	},
	beforeRegisterNodeDef(nodeType) {
		if (nodeType.comfyClass === "MathExpression|pysssss") {
			const onDrawForeground = nodeType.prototype.onDrawForeground;

			nodeType.prototype.onNodeCreated = function() {
				// These are typed as any to bypass backend validation
				// update frontend to restrict types
				for(const input of this.inputs) {
					input.type = "INT,FLOAT,IMAGE,LATENT";
				}
			}

			nodeType.prototype.onDrawForeground = function (ctx) {
				const r = onDrawForeground?.apply?.(this, arguments);

				const v = app.nodeOutputs?.[this.id + ""];
				if (!this.flags.collapsed && v) {
					const text = v.value[0] + "";
					ctx.save();
					ctx.font = "bold 12px sans-serif";
					ctx.fillStyle = "dodgerblue";
					const sz = ctx.measureText(text);
					ctx.fillText(text, this.size[0] - sz.width - 5, LiteGraph.NODE_SLOT_HEIGHT * 3);
					ctx.restore();
				}

				return r;
			};
		}
	},
});