Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update script.js

Browse Source
This commit is contained in:
Khachatur Avanesian 2023-10-15 10:48:50 +03:00 committed by GitHub
parent d295e97a0d
commit 3e223523ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 61 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
script.js
View File

@ -1,45 +1,43 @@
function gradioApp() { function gradioApp() {
const elems = document.getElementsByTagName('gradio-app') const elems = document.getElementsByTagName('gradio-app');
const elem = elems.length == 0 ? document : elems[0] const elem = elems.length == 0 ? document : elems[0];
if (elem !== document) { if (elem !== document) {
elem.getElementById = function(id) { elem.getElementById = function(id) {
return document.getElementById(id) return document.getElementById(id);
};
} }
} return elem.shadowRoot ? elem.shadowRoot : elem;
return elem.shadowRoot ? elem.shadowRoot : elem
} }
/** /**
* Get the currently selected top-level UI tab button (e.g. the button that says "Extras"). * Get the currently selected top-level UI tab button (e.g. the button that says "Extras").
*/ */
function get_uiCurrentTab() { function get_uiCurrentTab() {
return gradioApp().querySelector('#tabs > .tab-nav > button.selected') return gradioApp().querySelector('#tabs > .tab-nav > button.selected');
} }
/** /**
* Get the first currently visible top-level UI tab content (e.g. the div hosting the "txt2img" UI). * Get the first currently visible top-level UI tab content (e.g. the div hosting the "txt2img" UI).
*/ */
function get_uiCurrentTabContent() { function get_uiCurrentTabContent() {
return gradioApp().querySelector( return gradioApp().querySelector('#tabs > .tabitem[id^=tab_]:not([style*="display: none"])');
'#tabs > .tabitem[id^=tab_]:not([style*="display: none"])'
)
} }
var uiUpdateCallbacks = [] var uiUpdateCallbacks = [];
var uiAfterUpdateCallbacks = [] var uiAfterUpdateCallbacks = [];
var uiLoadedCallbacks = [] var uiLoadedCallbacks = [];
var uiTabChangeCallbacks = [] var uiTabChangeCallbacks = [];
var optionsChangedCallbacks = [] var optionsChangedCallbacks = [];
var uiAfterUpdateTimeout = null var uiAfterUpdateTimeout = null;
var uiCurrentTab = null var uiCurrentTab = null;
/** /**
* Register callback to be called at each UI update. * Register callback to be called at each UI update.
* The callback receives an array of MutationRecords as an argument. * The callback receives an array of MutationRecords as an argument.
*/ */
function onUiUpdate(callback) { function onUiUpdate(callback) {
uiUpdateCallbacks.push(callback) uiUpdateCallbacks.push(callback);
} }
/** /**
@ -51,7 +49,7 @@ function onUiUpdate(callback) {
* not be called quite as often. * not be called quite as often.
*/ */
function onAfterUiUpdate(callback) { function onAfterUiUpdate(callback) {
uiAfterUpdateCallbacks.push(callback) uiAfterUpdateCallbacks.push(callback);
} }
/** /**
@ -59,7 +57,7 @@ function onAfterUiUpdate(callback) {
* The callback receives no arguments. * The callback receives no arguments.
*/ */
function onUiLoaded(callback) { function onUiLoaded(callback) {
uiLoadedCallbacks.push(callback) uiLoadedCallbacks.push(callback);
} }
/** /**
@ -67,7 +65,7 @@ function onUiLoaded(callback) {
* The callback receives no arguments. * The callback receives no arguments.
*/ */
function onUiTabChange(callback) { function onUiTabChange(callback) {
uiTabChangeCallbacks.push(callback) uiTabChangeCallbacks.push(callback);
} }
/** /**
@ -76,15 +74,15 @@ function onUiTabChange(callback) {
* @param callback * @param callback
*/ */
function onOptionsChanged(callback) { function onOptionsChanged(callback) {
optionsChangedCallbacks.push(callback) optionsChangedCallbacks.push(callback);
} }
function executeCallbacks(queue, arg) { function executeCallbacks(queue, arg) {
for (const callback of queue) { for (const callback of queue) {
try { try {
callback(arg) callback(arg);
} catch (e) { } catch (e) {
console.error('error running callback', callback, ':', e) console.error("error running callback", callback, ":", e);
} }
} }
} }
@ -96,78 +94,74 @@ function executeCallbacks(queue, arg) {
* when there are multiple mutations observed. * when there are multiple mutations observed.
*/ */
function scheduleAfterUiUpdateCallbacks() { function scheduleAfterUiUpdateCallbacks() {
clearTimeout(uiAfterUpdateTimeout) clearTimeout(uiAfterUpdateTimeout);
uiAfterUpdateTimeout = setTimeout(function() { uiAfterUpdateTimeout = setTimeout(function() {
executeCallbacks(uiAfterUpdateCallbacks) executeCallbacks(uiAfterUpdateCallbacks);
}, 200) }, 200);
} }
var executedOnLoaded = false var executedOnLoaded = false;
document.addEventListener('DOMContentLoaded', function () { document.addEventListener("DOMContentLoaded", function() {
var mutationObserver = new MutationObserver(function(m) { var mutationObserver = new MutationObserver(function(m) {
if (!executedOnLoaded && gradioApp().querySelector('#txt2img_prompt')) { if (!executedOnLoaded && gradioApp().querySelector('#txt2img_prompt')) {
executedOnLoaded = true executedOnLoaded = true;
executeCallbacks(uiLoadedCallbacks) executeCallbacks(uiLoadedCallbacks);
} }
executeCallbacks(uiUpdateCallbacks, m) executeCallbacks(uiUpdateCallbacks, m);
scheduleAfterUiUpdateCallbacks() scheduleAfterUiUpdateCallbacks();
const newTab = get_uiCurrentTab() const newTab = get_uiCurrentTab();
if (newTab && newTab !== uiCurrentTab) { if (newTab && (newTab !== uiCurrentTab)) {
uiCurrentTab = newTab uiCurrentTab = newTab;
executeCallbacks(uiTabChangeCallbacks) executeCallbacks(uiTabChangeCallbacks);
} }
}) });
mutationObserver.observe(gradioApp(), {childList: true, subtree: true}) mutationObserver.observe(gradioApp(), {childList: true, subtree: true});
}) });
/** /**
* Add a ctrl+enter as a shortcut to start a generation * Add a ctrl+enter as a shortcut to start a generation
*/ */
document.addEventListener('keydown', function(e) { document.addEventListener('keydown', function(e) {
const isEnter = e.key === 'Enter' || e.keyCode === 13 const isEnter = e.key === 'Enter' || e.keyCode === 13;
const isModifierKey = e.metaKey || e.ctrlKey || e.altKey const isModifierKey = e.metaKey || e.ctrlKey || e.altKey;
const interruptButton = get_uiCurrentTabContent().querySelector( const interruptButton = get_uiCurrentTabContent().querySelector('button[id$=_interrupt]');
'button[id$=_interrupt]' const generateButton = get_uiCurrentTabContent().querySelector('button[id$=_generate]');
)
const generateButton = get_uiCurrentTabContent().querySelector(
'button[id$=_generate]'
)
if (isEnter && isModifierKey) { if (isEnter && isModifierKey) {
if (interruptButton.style.display === 'block') { if (interruptButton.style.display === 'block') {
interruptButton.click() interruptButton.click();
setTimeout(function() { setTimeout(function() {
generateButton.click() generateButton.click();
}, 500) }, 500);
} else { } else {
generateButton.click() generateButton.click();
} }
e.preventDefault() e.preventDefault();
} }
}) });
/** /**
* checks that a UI element is not in another hidden element or tab content * checks that a UI element is not in another hidden element or tab content
*/ */
function uiElementIsVisible(el) { function uiElementIsVisible(el) {
if (el === document) { if (el === document) {
return true return true;
} }
const computedStyle = getComputedStyle(el) const computedStyle = getComputedStyle(el);
const isVisible = computedStyle.display !== 'none' const isVisible = computedStyle.display !== 'none';
if (!isVisible) return false if (!isVisible) return false;
return uiElementIsVisible(el.parentNode) return uiElementIsVisible(el.parentNode);
} }
function uiElementInSight(el) { function uiElementInSight(el) {
const clRect = el.getBoundingClientRect() const clRect = el.getBoundingClientRect();
const windowHeight = window.innerHeight const windowHeight = window.innerHeight;
const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight;
return isOnScreen return isOnScreen;
} }