lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-21 06:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Gradio 4 + WebUI 1.10

Browse Source
This commit is contained in:
layerdiffusion
2024-07-26 08:51:34 -07:00
parent e95333c556
commit e26abf87ec
201 changed files with 7562 additions and 4834 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
extensions-builtin/sd_forge_controlnet/javascript/active_units.js
View File

@@ -95,7 +95,6 @@
this.attachImageUploadListener();
this.attachImageStateChangeObserver();
this.attachA1111SendInfoObserver();
this.attachPresetDropdownObserver();
this.attachAccordionStateObserver();
}
@@ -119,7 +118,7 @@
*/
getUnitHeaderTextElement() {
return this.tab.querySelector(
`:nth-child(${this.tabIndex + 1}) span.svelte-s1r2yt`
`button > span:nth-child(1)`
);
}
@@ -192,16 +191,17 @@
unitHeader.appendChild(span);
}
getInputImageSrc() {
const img = this.inputImageGroup.querySelector('.cnet-image img');
return img ? img.src : null;
const img = this.inputImageGroup.querySelector('.cnet-image .forge-image');
return (img && img.src.startsWith('data')) ? img.src : null;
}
getPreprocessorPreviewImageSrc() {
const img = this.generatedImageGroup.querySelector('.cnet-image img');
return img ? img.src : null;
const img = this.generatedImageGroup.querySelector('.cnet-image .forge-image');
return (img && img.src.startsWith('data')) ? img.src : null;
}
getMaskImageSrc() {
function isEmptyCanvas(canvas) {
if (!canvas) return true;
if (canvas.width == 0 || canvas.height ==0) return true;
const ctx = canvas.getContext('2d');
// Get the image data
const imageData = ctx.getImageData(0, 0, canvas.width, canvas.height);
@@ -216,14 +216,14 @@
}
return isPureBlack;
}
const maskImg = this.maskImageGroup.querySelector('.cnet-mask-image img');
const maskImg = this.maskImageGroup.querySelector('.cnet-mask-image .forge-image');
// Hand-drawn mask on mask upload.
const handDrawnMaskCanvas = this.maskImageGroup.querySelector('.cnet-mask-image canvas[key="mask"]');
const handDrawnMaskCanvas = this.maskImageGroup.querySelector('.cnet-mask-image .forge-drawing-canvas');
// Hand-drawn mask on input image upload.
const inputImageHandDrawnMaskCanvas = this.inputImageGroup.querySelector('.cnet-image canvas[key="mask"]');
const inputImageHandDrawnMaskCanvas = this.inputImageGroup.querySelector('.cnet-image .forge-drawing-canvas');
if (!isEmptyCanvas(handDrawnMaskCanvas)) {
return handDrawnMaskCanvas.toDataURL();
} else if (maskImg) {
} else if (maskImg && maskImg.src.startsWith('data')) {
return maskImg.src;
} else if (!isEmptyCanvas(inputImageHandDrawnMaskCanvas)) {
return inputImageHandDrawnMaskCanvas.toDataURL();
@@ -347,25 +347,6 @@
}
}
attachPresetDropdownObserver() {
const presetDropDown = this.tab.querySelector('.cnet-preset-dropdown');
new MutationObserver((mutationsList) => {
for (const mutation of mutationsList) {
if (mutation.removedNodes.length > 0) {
setTimeout(() => {
this.updateActiveState();
this.updateActiveUnitCount();
this.updateActiveControlType();
}, 1000);
return;
}
}
}).observe(presetDropDown, {
childList: true,
subtree: true,
});
}
/**
* Observer that triggers when the ControlNetUnit's accordion(tab) closes.
*/