Commit f00eaa4d authored by Khachatur Avanesian's avatar Khachatur Avanesian Committed by GitHub

Start / Restart generation by Ctrl (Alt) + Enter

Add ability to interrupt current generation and start generation again by Ctrl (Alt) + Enter
parent d4255506
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('#tabs > .tabitem[id^=tab_]:not([style*="display: none"])'); return gradioApp().querySelector('#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);
} }
/** /**
* Register callback to be called soon after UI updates. * Register callback to be called soon after UI updates.
* The callback receives no arguments. * The callback receives no arguments.
* *
* This is preferred over `onUiUpdate` if you don't need * This is preferred over `onUiUpdate` if you don't need
* access to the MutationRecords, as your function will * access to the MutationRecords, as your function will
* not be called quite as often. * not be called quite as often.
*/ */
function onAfterUiUpdate(callback) { function onAfterUiUpdate(callback) {
uiAfterUpdateCallbacks.push(callback); uiAfterUpdateCallbacks.push(callback);
} }
/** /**
* Register callback to be called when the UI is loaded. * Register callback to be called when the UI is loaded.
* The callback receives no arguments. * The callback receives no arguments.
*/ */
function onUiLoaded(callback) { function onUiLoaded(callback) {
uiLoadedCallbacks.push(callback); uiLoadedCallbacks.push(callback);
} }
/** /**
* Register callback to be called when the UI tab is changed. * Register callback to be called when the UI tab is changed.
* The callback receives no arguments. * The callback receives no arguments.
*/ */
function onUiTabChange(callback) { function onUiTabChange(callback) {
uiTabChangeCallbacks.push(callback); uiTabChangeCallbacks.push(callback);
} }
/** /**
* Register callback to be called when the options are changed. * Register callback to be called when the options are changed.
* The callback receives no arguments. * The callback receives no arguments.
* @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);
} }
} }
} }
/** /**
* Schedule the execution of the callbacks registered with onAfterUiUpdate. * Schedule the execution of the callbacks registered with onAfterUiUpdate.
* The callbacks are executed after a short while, unless another call to this function * 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 * is made before that time. IOW, the callbacks are executed only once, even
* 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 (Alt) + Enter as a shortcut to start / restart a generation
*/ */
document.addEventListener('keydown', function(e) { document.addEventListener('keydown', (e) => {
var handled = false; const isEnter = e.key === 'Enter' || e.keyCode === 13
if (e.key !== undefined) { const isModifierKey = e.metaKey || e.ctrlKey || e.altKey
if ((e.key == "Enter" && (e.metaKey || e.ctrlKey || e.altKey))) handled = true;
} else if (e.keyCode !== undefined) { const interruptButton = get_uiCurrentTabContent().querySelector('button[id$=_interrupt]')
if ((e.keyCode == 13 && (e.metaKey || e.ctrlKey || e.altKey))) handled = true; const generateButton = get_uiCurrentTabContent().querySelector('button[id$=_generate]')
}
if (handled) { if (isEnter && isModifierKey) {
var button = get_uiCurrentTabContent().querySelector('button[id$=_generate]'); if (interruptButton.style.display === 'block') {
if (button) { interruptButton.click()
button.click(); setTimeout(() => generateButton.click(), 500)
} } else {
e.preventDefault(); generateButton.click()
} }
}); e.preventDefault()
}
/** })
* checks that a UI element is not in another hidden element or tab content
*/ /**
function uiElementIsVisible(el) { * checks that a UI element is not in another hidden element or tab content
if (el === document) { */
return true; function uiElementIsVisible(el) {
} if (el === document) {
return true;
const computedStyle = getComputedStyle(el); }
const isVisible = computedStyle.display !== 'none';
const computedStyle = getComputedStyle(el);
if (!isVisible) return false; const isVisible = computedStyle.display !== 'none';
return uiElementIsVisible(el.parentNode);
} if (!isVisible) return false;
return uiElementIsVisible(el.parentNode);
function uiElementInSight(el) { }
const clRect = el.getBoundingClientRect();
const windowHeight = window.innerHeight; function uiElementInSight(el) {
const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight; const clRect = el.getBoundingClientRect();
const windowHeight = window.innerHeight;
return isOnScreen; const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight;
}
return isOnScreen;
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment