|
@@ -1,167 +1,167 @@
|
|
-function gradioApp() {
|
|
|
|
- const elems = document.getElementsByTagName('gradio-app');
|
|
|
|
- const elem = elems.length == 0 ? document : elems[0];
|
|
|
|
-
|
|
|
|
- if (elem !== document) {
|
|
|
|
- elem.getElementById = function(id) {
|
|
|
|
- return document.getElementById(id);
|
|
|
|
- };
|
|
|
|
- }
|
|
|
|
- return elem.shadowRoot ? elem.shadowRoot : elem;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Get the currently selected top-level UI tab button (e.g. the button that says "Extras").
|
|
|
|
- */
|
|
|
|
-function get_uiCurrentTab() {
|
|
|
|
- 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).
|
|
|
|
- */
|
|
|
|
-function get_uiCurrentTabContent() {
|
|
|
|
- return gradioApp().querySelector('#tabs > .tabitem[id^=tab_]:not([style*="display: none"])');
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-var uiUpdateCallbacks = [];
|
|
|
|
-var uiAfterUpdateCallbacks = [];
|
|
|
|
-var uiLoadedCallbacks = [];
|
|
|
|
-var uiTabChangeCallbacks = [];
|
|
|
|
-var optionsChangedCallbacks = [];
|
|
|
|
-var uiAfterUpdateTimeout = null;
|
|
|
|
-var uiCurrentTab = null;
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Register callback to be called at each UI update.
|
|
|
|
- * The callback receives an array of MutationRecords as an argument.
|
|
|
|
- */
|
|
|
|
-function onUiUpdate(callback) {
|
|
|
|
- uiUpdateCallbacks.push(callback);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Register callback to be called soon after UI updates.
|
|
|
|
- * The callback receives no arguments.
|
|
|
|
- *
|
|
|
|
- * This is preferred over `onUiUpdate` if you don't need
|
|
|
|
- * access to the MutationRecords, as your function will
|
|
|
|
- * not be called quite as often.
|
|
|
|
- */
|
|
|
|
-function onAfterUiUpdate(callback) {
|
|
|
|
- uiAfterUpdateCallbacks.push(callback);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Register callback to be called when the UI is loaded.
|
|
|
|
- * The callback receives no arguments.
|
|
|
|
- */
|
|
|
|
-function onUiLoaded(callback) {
|
|
|
|
- uiLoadedCallbacks.push(callback);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Register callback to be called when the UI tab is changed.
|
|
|
|
- * The callback receives no arguments.
|
|
|
|
- */
|
|
|
|
-function onUiTabChange(callback) {
|
|
|
|
- uiTabChangeCallbacks.push(callback);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Register callback to be called when the options are changed.
|
|
|
|
- * The callback receives no arguments.
|
|
|
|
- * @param callback
|
|
|
|
- */
|
|
|
|
-function onOptionsChanged(callback) {
|
|
|
|
- optionsChangedCallbacks.push(callback);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-function executeCallbacks(queue, arg) {
|
|
|
|
- for (const callback of queue) {
|
|
|
|
- try {
|
|
|
|
- callback(arg);
|
|
|
|
- } catch (e) {
|
|
|
|
- console.error("error running callback", callback, ":", e);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Schedule the execution of the callbacks registered with onAfterUiUpdate.
|
|
|
|
- * The callbacks are executed after a short while, unless another call to this function
|
|
|
|
- * is made before that time. IOW, the callbacks are executed only once, even
|
|
|
|
- * when there are multiple mutations observed.
|
|
|
|
- */
|
|
|
|
-function scheduleAfterUiUpdateCallbacks() {
|
|
|
|
- clearTimeout(uiAfterUpdateTimeout);
|
|
|
|
- uiAfterUpdateTimeout = setTimeout(function() {
|
|
|
|
- executeCallbacks(uiAfterUpdateCallbacks);
|
|
|
|
- }, 200);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-var executedOnLoaded = false;
|
|
|
|
-
|
|
|
|
-document.addEventListener("DOMContentLoaded", function() {
|
|
|
|
- var mutationObserver = new MutationObserver(function(m) {
|
|
|
|
- if (!executedOnLoaded && gradioApp().querySelector('#txt2img_prompt')) {
|
|
|
|
- executedOnLoaded = true;
|
|
|
|
- executeCallbacks(uiLoadedCallbacks);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- executeCallbacks(uiUpdateCallbacks, m);
|
|
|
|
- scheduleAfterUiUpdateCallbacks();
|
|
|
|
- const newTab = get_uiCurrentTab();
|
|
|
|
- if (newTab && (newTab !== uiCurrentTab)) {
|
|
|
|
- uiCurrentTab = newTab;
|
|
|
|
- executeCallbacks(uiTabChangeCallbacks);
|
|
|
|
- }
|
|
|
|
- });
|
|
|
|
- mutationObserver.observe(gradioApp(), {childList: true, subtree: true});
|
|
|
|
-});
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * Add a ctrl+enter as a shortcut to start a generation
|
|
|
|
- */
|
|
|
|
-document.addEventListener('keydown', function(e) {
|
|
|
|
- const isEnter = e.key === 'Enter' || e.keyCode === 13;
|
|
|
|
- const isModifierKey = e.metaKey || e.ctrlKey || e.altKey;
|
|
|
|
-
|
|
|
|
- const interruptButton = get_uiCurrentTabContent().querySelector('button[id$=_interrupt]');
|
|
|
|
- const generateButton = get_uiCurrentTabContent().querySelector('button[id$=_generate]');
|
|
|
|
-
|
|
|
|
- if (isEnter && isModifierKey) {
|
|
|
|
- if (interruptButton.style.display === 'block') {
|
|
|
|
- interruptButton.click();
|
|
|
|
- setTimeout(function() {
|
|
|
|
- generateButton.click();
|
|
|
|
- }, 500);
|
|
|
|
- } else {
|
|
|
|
- generateButton.click();
|
|
|
|
- }
|
|
|
|
- e.preventDefault();
|
|
|
|
- }
|
|
|
|
-});
|
|
|
|
-
|
|
|
|
-/**
|
|
|
|
- * checks that a UI element is not in another hidden element or tab content
|
|
|
|
- */
|
|
|
|
-function uiElementIsVisible(el) {
|
|
|
|
- if (el === document) {
|
|
|
|
- return true;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- const computedStyle = getComputedStyle(el);
|
|
|
|
- const isVisible = computedStyle.display !== 'none';
|
|
|
|
-
|
|
|
|
- if (!isVisible) return false;
|
|
|
|
- return uiElementIsVisible(el.parentNode);
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-function uiElementInSight(el) {
|
|
|
|
- const clRect = el.getBoundingClientRect();
|
|
|
|
- const windowHeight = window.innerHeight;
|
|
|
|
- const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight;
|
|
|
|
-
|
|
|
|
- return isOnScreen;
|
|
|
|
-}
|
|
|
|
|
|
+function gradioApp() {
|
|
|
|
+ const elems = document.getElementsByTagName('gradio-app');
|
|
|
|
+ const elem = elems.length == 0 ? document : elems[0];
|
|
|
|
+
|
|
|
|
+ if (elem !== document) {
|
|
|
|
+ elem.getElementById = function(id) {
|
|
|
|
+ return document.getElementById(id);
|
|
|
|
+ };
|
|
|
|
+ }
|
|
|
|
+ return elem.shadowRoot ? elem.shadowRoot : elem;
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Get the currently selected top-level UI tab button (e.g. the button that says "Extras").
|
|
|
|
+ */
|
|
|
|
+function get_uiCurrentTab() {
|
|
|
|
+ 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).
|
|
|
|
+ */
|
|
|
|
+function get_uiCurrentTabContent() {
|
|
|
|
+ return gradioApp().querySelector('#tabs > .tabitem[id^=tab_]:not([style*="display: none"])');
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+var uiUpdateCallbacks = [];
|
|
|
|
+var uiAfterUpdateCallbacks = [];
|
|
|
|
+var uiLoadedCallbacks = [];
|
|
|
|
+var uiTabChangeCallbacks = [];
|
|
|
|
+var optionsChangedCallbacks = [];
|
|
|
|
+var uiAfterUpdateTimeout = null;
|
|
|
|
+var uiCurrentTab = null;
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Register callback to be called at each UI update.
|
|
|
|
+ * The callback receives an array of MutationRecords as an argument.
|
|
|
|
+ */
|
|
|
|
+function onUiUpdate(callback) {
|
|
|
|
+ uiUpdateCallbacks.push(callback);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Register callback to be called soon after UI updates.
|
|
|
|
+ * The callback receives no arguments.
|
|
|
|
+ *
|
|
|
|
+ * This is preferred over `onUiUpdate` if you don't need
|
|
|
|
+ * access to the MutationRecords, as your function will
|
|
|
|
+ * not be called quite as often.
|
|
|
|
+ */
|
|
|
|
+function onAfterUiUpdate(callback) {
|
|
|
|
+ uiAfterUpdateCallbacks.push(callback);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Register callback to be called when the UI is loaded.
|
|
|
|
+ * The callback receives no arguments.
|
|
|
|
+ */
|
|
|
|
+function onUiLoaded(callback) {
|
|
|
|
+ uiLoadedCallbacks.push(callback);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Register callback to be called when the UI tab is changed.
|
|
|
|
+ * The callback receives no arguments.
|
|
|
|
+ */
|
|
|
|
+function onUiTabChange(callback) {
|
|
|
|
+ uiTabChangeCallbacks.push(callback);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Register callback to be called when the options are changed.
|
|
|
|
+ * The callback receives no arguments.
|
|
|
|
+ * @param callback
|
|
|
|
+ */
|
|
|
|
+function onOptionsChanged(callback) {
|
|
|
|
+ optionsChangedCallbacks.push(callback);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+function executeCallbacks(queue, arg) {
|
|
|
|
+ for (const callback of queue) {
|
|
|
|
+ try {
|
|
|
|
+ callback(arg);
|
|
|
|
+ } catch (e) {
|
|
|
|
+ console.error("error running callback", callback, ":", e);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Schedule the execution of the callbacks registered with onAfterUiUpdate.
|
|
|
|
+ * The callbacks are executed after a short while, unless another call to this function
|
|
|
|
+ * is made before that time. IOW, the callbacks are executed only once, even
|
|
|
|
+ * when there are multiple mutations observed.
|
|
|
|
+ */
|
|
|
|
+function scheduleAfterUiUpdateCallbacks() {
|
|
|
|
+ clearTimeout(uiAfterUpdateTimeout);
|
|
|
|
+ uiAfterUpdateTimeout = setTimeout(function() {
|
|
|
|
+ executeCallbacks(uiAfterUpdateCallbacks);
|
|
|
|
+ }, 200);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+var executedOnLoaded = false;
|
|
|
|
+
|
|
|
|
+document.addEventListener("DOMContentLoaded", function() {
|
|
|
|
+ var mutationObserver = new MutationObserver(function(m) {
|
|
|
|
+ if (!executedOnLoaded && gradioApp().querySelector('#txt2img_prompt')) {
|
|
|
|
+ executedOnLoaded = true;
|
|
|
|
+ executeCallbacks(uiLoadedCallbacks);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ executeCallbacks(uiUpdateCallbacks, m);
|
|
|
|
+ scheduleAfterUiUpdateCallbacks();
|
|
|
|
+ const newTab = get_uiCurrentTab();
|
|
|
|
+ if (newTab && (newTab !== uiCurrentTab)) {
|
|
|
|
+ uiCurrentTab = newTab;
|
|
|
|
+ executeCallbacks(uiTabChangeCallbacks);
|
|
|
|
+ }
|
|
|
|
+ });
|
|
|
|
+ mutationObserver.observe(gradioApp(), {childList: true, subtree: true});
|
|
|
|
+});
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * Add a ctrl+enter as a shortcut to start a generation
|
|
|
|
+ */
|
|
|
|
+document.addEventListener('keydown', function(e) {
|
|
|
|
+ const isEnter = e.key === 'Enter' || e.keyCode === 13;
|
|
|
|
+ const isModifierKey = e.metaKey || e.ctrlKey || e.altKey;
|
|
|
|
+
|
|
|
|
+ const interruptButton = get_uiCurrentTabContent().querySelector('button[id$=_interrupt]');
|
|
|
|
+ const generateButton = get_uiCurrentTabContent().querySelector('button[id$=_generate]');
|
|
|
|
+
|
|
|
|
+ if (isEnter && isModifierKey) {
|
|
|
|
+ if (interruptButton.style.display === 'block') {
|
|
|
|
+ interruptButton.click();
|
|
|
|
+ setTimeout(function() {
|
|
|
|
+ generateButton.click();
|
|
|
|
+ }, 500);
|
|
|
|
+ } else {
|
|
|
|
+ generateButton.click();
|
|
|
|
+ }
|
|
|
|
+ e.preventDefault();
|
|
|
|
+ }
|
|
|
|
+});
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * checks that a UI element is not in another hidden element or tab content
|
|
|
|
+ */
|
|
|
|
+function uiElementIsVisible(el) {
|
|
|
|
+ if (el === document) {
|
|
|
|
+ return true;
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ const computedStyle = getComputedStyle(el);
|
|
|
|
+ const isVisible = computedStyle.display !== 'none';
|
|
|
|
+
|
|
|
|
+ if (!isVisible) return false;
|
|
|
|
+ return uiElementIsVisible(el.parentNode);
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+function uiElementInSight(el) {
|
|
|
|
+ const clRect = el.getBoundingClientRect();
|
|
|
|
+ const windowHeight = window.innerHeight;
|
|
|
|
+ const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight;
|
|
|
|
+
|
|
|
|
+ return isOnScreen;
|
|
|
|
+}
|