wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-13 23:12:35 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Added three button locations that are now selectable in the settings menu under 'Change Reboot Server Button Location' locations available are 'Menu Bar Top Icon' this places a small ⚡ in the top right corner next to settings. 'Small Button' This places the button as a small button next to View History. 'Main Menu' This places the button under Load Default on the main menu. I also added a toggle in the Settings Menu so you can choose whether it confirms you wanting to reboot the server

Browse Source
This commit is contained in:
ThisModernDay 2023-10-22 12:54:21 -04:00
parent 95f7199e14
commit 4aeca626bd
1 changed files with 50 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
web/scripts/ui.js
View File

@ -46,6 +46,7 @@ export function $el(tag, propsOrChildren, children) {
return element; return element;
} }
function dragElement(dragEl, settings) { function dragElement(dragEl, settings) {
var posDiffX = 0, var posDiffX = 0,
posDiffY = 0, posDiffY = 0,
@ -165,6 +166,7 @@ function dragElement(dragEl, settings) {
document.onmouseup = null; document.onmouseup = null;
document.onmousemove = null; document.onmousemove = null;
} }
} }
export class ComfyDialog { export class ComfyDialog {
@ -618,6 +620,11 @@ export class ComfyUI {
}, [ }, [
$el("span.drag-handle"), $el("span.drag-handle"),
$el("span", {$: (q) => (this.queueSize = q)}), $el("span", {$: (q) => (this.queueSize = q)}),
$el("button.comfy-reboot-server-button-icon", {id: "comfy-reboot-server-icon", textContent: "⚡", onclick: () => {
if (!confirmReboot.value || confirm("Are you sure you'd like to reboot the server?")) {
api.fetchApi("/reboot")
}
}}),
$el("button.comfy-settings-btn", {textContent: "⚙️", onclick: () => this.settings.show()}), $el("button.comfy-settings-btn", {textContent: "⚙️", onclick: () => this.settings.show()}),
]), ]),
$el("button.comfy-queue-btn", { $el("button.comfy-queue-btn", {
@ -704,6 +711,13 @@ export class ComfyUI {
this.history.toggle(); this.history.toggle();
}, },
}), }),
$el("button", {
id: "comfy-reboot-server-button-small", textContent: "Reboot Server", onclick: () => {
if (!confirmReboot.value || confirm("Are you sure you'd like to reboot the server?")) {
api.fetchApi("/reboot")
}
}
})
]), ]),
this.queue.element, this.queue.element,
this.history.element, this.history.element,
@ -789,14 +803,16 @@ export class ComfyUI {
} }
}), }),
$el("button", { $el("button", {
id: "comfy-reboot-server-button", textContent: "Reboot Server", onclick: () => { id: "comfy-reboot-server-button-large", textContent: "Reboot Server", onclick: () => {
if (!confirmClear.value || confirm("Are you sure you'd like to reboot the server?")) { if (!confirmReboot.value || confirm("Are you sure you'd like to reboot the server?")) {
api.fetchApi("/reboot") api.fetchApi("/reboot")
} }
} }
}) })
]); ]);
const devMode = this.settings.addSetting({ const devMode = this.settings.addSetting({
id: "Comfy.DevMode", id: "Comfy.DevMode",
name: "Enable Dev mode Options", name: "Enable Dev mode Options",
@ -805,6 +821,38 @@ export class ComfyUI {
onChange: function(value) { document.getElementById("comfy-dev-save-api-button").style.display = value ? "block" : "none"}, onChange: function(value) { document.getElementById("comfy-dev-save-api-button").style.display = value ? "block" : "none"},
}); });
const changeRebootLocation = this.settings.addSetting({
id: "Comfy.RebootLocation",
name: "Change Reboot Server Button Location",
type: "combo",
options: ['Menu Bar Top Icon', 'Small Button', 'Main Menu'],
defaultValue: 'Main Menu',
onChange: (value) => {
if( value == 'Menu Bar Top Icon') {
document.querySelector('#comfy-reboot-server-icon').style.display = ""
} else {
document.querySelector('#comfy-reboot-server-icon').style.display = "none"
}
if(value == 'Small Button') {
document.querySelector('#comfy-reboot-server-button-small').style.display = ""
} else {
document.querySelector('#comfy-reboot-server-button-small').style.display = "none"
}
if(value == 'Main Menu') {
document.querySelector('#comfy-reboot-server-button-large').style.display = ""
} else {
document.querySelector('#comfy-reboot-server-button-large').style.display = "none"
}
}
});
const confirmReboot = this.settings.addSetting({
id: "Comfy.ConfirmReboot",
name: "Require confirmation when rebooting server",
type: "boolean",
defaultValue: true,
});
dragElement(this.menuContainer, this.settings); dragElement(this.menuContainer, this.settings);
this.setStatus({exec_info: {queue_remaining: "X"}}); this.setStatus({exec_info: {queue_remaining: "X"}});