File size: 3,688 Bytes
0163a2c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
function gradioApp() {
const elems = document.getElementsByTagName('gradio-app')
const gradioShadowRoot = elems.length == 0 ? null : elems[0].shadowRoot
return !!gradioShadowRoot ? gradioShadowRoot : document;
}
let executed = false
/** @type {(() => void)[]} */
/**
* @param {string} tab
* @param {boolean} show
*/
function kohya_sd_webui__toggle_runner_button(tab, show) {
gradioApp().getElementById(`kohya_sd_webui__${tab}_run_button`).style.display = show ? 'block' : 'none'
gradioApp().getElementById(`kohya_sd_webui__${tab}_stop_button`).style.display = show ? 'none' : 'block'
}
window.addEventListener('DOMContentLoaded', () => {
const observer = new MutationObserver((m) => {
if (!executed && gradioApp().querySelector('#kohya_sd_webui__root')) {
executed = true;
/** @type {Record<string, string>} */
const helps = kohya_sd_webui__help_map
/** @type {string[]} */
const all_tabs = kohya_sd_webui__all_tabs
const initializeTerminalObserver = () => {
const container = gradioApp().querySelector("#kohya_sd_webui__terminal_outputs")
const parentContainer = container.parentElement
const clearBtn = document.createElement('button')
clearBtn.innerText = 'Clear The Terminal'
clearBtn.style.color = 'yellow';
parentContainer.insertBefore(clearBtn, container)
let clearTerminal = false;
clearBtn.addEventListener('click', () => {
container.innerHTML = ''
clearTerminal = true
})
setInterval(async () => {
const res = await fetch('./internal/extensions/kohya-sd-scripts-webui/terminal/outputs', {
method: "POST",
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
output_index: container.children.length,
clear_terminal: clearTerminal,
}),
})
clearTerminal = false
const obj = await res.json()
const isBottom = container.scrollHeight - container.scrollTop === container.clientHeight
for(const line of obj.outputs){
const el = document.createElement('div')
el.innerText = line
container.appendChild(el)
}
if(isBottom) container.scrollTop = container.scrollHeight
}, 1000)
}
const checkProcessIsAlive = () => {
setInterval(async () => {
const res = await fetch('./internal/extensions/kohya-sd-scripts-webui/process/alive')
const obj = await res.json()
for (const tab of all_tabs)
kohya_sd_webui__toggle_runner_button(tab, !obj.alive)
}, 1000)
}
initializeTerminalObserver()
checkProcessIsAlive()
for (const tab of all_tabs)
gradioApp().querySelector(`#kohya_sd_webui__${tab}_run_button`).addEventListener('click', () => kohya_sd_webui__toggle_runner_button(tab, false))
for (const [k, v] of Object.entries(helps)) {
el = gradioApp().getElementById(k)
if (!el) continue
el.title = v
}
}
})
observer.observe(gradioApp(), { childList: true, subtree: true })
}) |