chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add container to the buttons injected into the right menu

Browse Source
This commit is contained in:
EmmanuelMr18 2025-01-10 01:08:42 -06:00
parent 73853a60ff
commit 230cee40d2
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
web-plugin/index.js
View File

@ -2120,7 +2120,7 @@ function addQueueButtons(buttonConfigs = DEFAULT_BUTTONS) {
// Function to add butons to the menu right // Function to add butons to the menu right
function addMenuRightButtons(buttonConfigs) { function addMenuRightButtons(buttonConfigs) {
const menuRightButtons = document.querySelector('.comfyui-menu-right') const menuRightButtons = document.querySelector('.comfyui-menu-right .flex')
if (!menuRightButtons) return if (!menuRightButtons) return
@ -2129,19 +2129,22 @@ function addMenuRightButtons(buttonConfigs) {
for (const button of existingButtons) { for (const button of existingButtons) {
button.remove() button.remove()
} }
if (!menuRightButtons) return const container = document.createElement('div')
container.className = 'comfyui-button-group mx-2'
for (const config of buttonConfigs) { for (const config of buttonConfigs) {
const button = createMenuRightButton(config) const button = createMenuRightButton(config)
menuRightButtons.appendChild(button) container.appendChild(button)
} }
menuRightButtons.appendChild(container)
} }
function createMenuRightButton(config) { function createMenuRightButton(config) {
const button = document.createElement('button') const button = document.createElement('button')
button.id = `cd-button-${config.id}` button.id = `cd-button-${config.id}`
button.className = 'p-button p-component p-button-secondary p-button-md' button.className = `p-button p-component p-button-secondary p-button-md ${config.btnClasses}`
button.innerHTML = ` button.innerHTML = `
<span class="p-button-icon pi ${config.icon}"></span> <span class="p-button-icon pi ${config.icon}"></span>
<span class="p-button-label">${config.label}</span> <span class="p-button-label">${config.label}</span>