1580158
/
stable-diffusion-webui-images-browser
mirror of https://github.com/AlUlkesh/stable-diffusion-webui-images-browser.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

startup-procedure with MutationObserver, #135

pull/150/head
AlUlkesh 2023-04-10 23:15:56 +02:00
parent 2cba040fb4
commit 79c028e47d
1 changed files with 33 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
javascript/image_browser.js
View File

@ -10,12 +10,40 @@ onUiLoaded(image_browser_start_it_up)
async function image_browser_wait_for_webui() {
if (image_browser_debug) console.log("image_browser_wait_for_webui:start")
await image_browser_delay(100)
while (gradioApp().getElementById("setting_sd_model_checkpoint").querySelector(".eta-bar")) {
await image_browser_delay(200)
}
await image_browser_delay(500)
sd_model = gradioApp().getElementById("setting_sd_model_checkpoint")
if (!sd_model.querySelector(".eta-bar")) {
image_browser_webui_ready = true
image_browser_start()
} else {
// Set timeout for MutationObserver
const startTime = Date.now()
// 40 seconds in milliseconds
const timeout = 40000
const webuiObserver = new MutationObserver(function(mutationsList) {
if (image_browser_debug) console.log("webuiObserver:start")
let found = false
outerLoop: for (let i = 0; i < mutationsList.length; i++) {
let mutation = mutationsList[i];
if (mutation.type === "childList") {
for (let j = 0; j < mutation.removedNodes.length; j++) {
let node = mutation.removedNodes[j];
if (node.nodeType === Node.ELEMENT_NODE && node.classList.contains("eta-bar")) {
found = true
break outerLoop;
}
}
}
}
if (found || (Date.now() - startTime > timeout)) {
image_browser_webui_ready = true
webuiObserver.disconnect()
if (image_browser_debug) console.log("webuiObserver:end")
image_browser_start()
}
})
webuiObserver.observe(gradioApp(), { childList:true, subtree:true })
}
if (image_browser_debug) console.log("image_browser_wait_for_webui:end")
}