add buttons to the attention seekers

js/FluxAttentionSeeker.js

@@ -0,0 +1,51 @@
+import { app } from "../../scripts/app.js";
+
+app.registerExtension({
+    name: "essentials.FluxAttentionSeeker",
+    async beforeRegisterNodeDef(nodeType, nodeData, app) {
+        if (!nodeData?.category?.startsWith("essentials")) {
+            return;
+        }
+
+        if (nodeData.name === "FluxAttentionSeeker+") {
+            const onCreated = nodeType.prototype.onNodeCreated;
+
+            nodeType.prototype.onNodeCreated = function () {
+                console.log(LGraphCanvas);
+                this.addWidget("button", "RESET ALL", null, () => {
+                    this.widgets.forEach(w => {
+                        if (w.type === "slider") {
+                            w.value = 1.0;
+                        }
+                    });
+                });
+
+                this.addWidget("button", "ZERO ALL", null, () => {
+                    this.widgets.forEach(w => {
+                        if (w.type === "slider") {
+                            w.value = 0.0;
+                        }
+                    });
+                });
+
+                this.addWidget("button", "REPEAT FIRST", null, () => {
+                    var clip_value = undefined;
+                    var t5_value = undefined;
+                    this.widgets.forEach(w => {
+                        if (w.name.startsWith('clip_l')) {
+                            if (clip_value === undefined) {
+                                clip_value = w.value;
+                            }
+                            w.value = clip_value;
+                        } else if (w.name.startsWith('t5')) {
+                            if (t5_value === undefined) {
+                                t5_value = w.value;
+                            }
+                            w.value = t5_value;
+                        }
+                    });
+                });
+            };
+        }
+    },
+});

misc.py

@@ -3,6 +3,7 @@ import torch
 from .utils import AnyType
 import comfy.model_management
 from nodes import MAX_RESOLUTION
+import time
 
 any = AnyType("*")