Implement WebUI generator buttons in webui_generator.js

2026-02-24 16:12:05 +00:00
parent 68dfe1f897
commit 76303ae39d
1 changed files with 47 additions and 0 deletions
@@ -0,0 +1,47 @@
+import { app } from "../../scripts/app.js";
+import { api } from "../../scripts/api.js";
+
+function addButtons(node) {
+  if (node.__webuiGenButtonsAdded) return;
+  node.__webuiGenButtonsAdded = true;
+
+  node.addWidget("button", "Open Generator UI", null, async () => {
+    try {
+      const info = await api.fetchApi("/webui-generator/info").then(r => r.json());
+      const host = info.host || "127.0.0.1";
+      const port = info.port || 7860;
+
+      // Open generator (Gradio) in a new tab
+      const url = `http://${host}:${port}`;
+      window.open(url, "_blank", "noopener,noreferrer");
+    } catch (e) {
+      console.error(e);
+      alert("Failed to get generator URL. Check ComfyUI console logs.");
+    }
+  });
+
+  node.addWidget("button", "Open ComfyUI Output Folder", null, async () => {
+    try {
+      await api.fetchApi("/webui-generator/open-output", { method: "POST" });
+    } catch (e) {
+      console.error(e);
+      alert("Failed to open output folder.");
+    }
+  });
+
+  node.addWidget("text", "Note", "Upload the *API workflow JSON* in the generator UI.", () => {});
+}
+
+app.registerExtension({
+  name: "ai.joe.webui_generator_node",
+  async beforeRegisterNodeDef(nodeType, nodeData) {
+    if (nodeData.name === "WebUI_Generator_Launcher") {
+      const onNodeCreated = nodeType.prototype.onNodeCreated;
+      nodeType.prototype.onNodeCreated = function () {
+        const r = onNodeCreated?.apply(this, arguments);
+        addButtons(this);
+        return r;
+      };
+    }
+  }
+});