const log = (...msg) => { const dt = new Date(); const ts = `${dt.getHours().toString().padStart(2, '0')}:${dt.getMinutes().toString().padStart(2, '0')}:${dt.getSeconds().toString().padStart(2, '0')}.${dt.getMilliseconds().toString().padStart(3, '0')}`; console.log(ts, ...msg); // eslint-disable-line no-console }; const debug = (...msg) => { const dt = new Date(); const ts = `${dt.getHours().toString().padStart(2, '0')}:${dt.getMinutes().toString().padStart(2, '0')}:${dt.getSeconds().toString().padStart(2, '0')}.${dt.getMilliseconds().toString().padStart(3, '0')}`; console.debug(ts, ...msg); // eslint-disable-line no-console }; function gradioApp() { const elems = document.getElementsByTagName('gradio-app'); const elem = elems.length === 0 ? document : elems[0]; if (elem !== document) elem.getElementById = (id) => document.getElementById(id); return elem.shadowRoot ? elem.shadowRoot : elem; } function getUICurrentTab() { return gradioApp().querySelector('#tabs button.selected'); } function getUICurrentTabContent() { return gradioApp().querySelector('.tabitem[id^=tab_]:not([style*="display: none"])'); } const get_uiCurrentTabContent = getUICurrentTabContent; const get_uiCurrentTab = getUICurrentTab; const uiAfterUpdateCallbacks = []; const uiUpdateCallbacks = []; const uiLoadedCallbacks = []; const uiTabChangeCallbacks = []; const optionsChangedCallbacks = []; let uiCurrentTab = null; let uiAfterUpdateTimeout = null; function onAfterUiUpdate(callback) { uiAfterUpdateCallbacks.push(callback); } function onUiUpdate(callback) { uiUpdateCallbacks.push(callback); } function onUiLoaded(callback) { uiLoadedCallbacks.push(callback); } function onUiTabChange(callback) { uiTabChangeCallbacks.push(callback); } function onOptionsChanged(callback) { optionsChangedCallbacks.push(callback); } function executeCallbacks(queue, arg) { // if (!uiLoaded) return for (const callback of queue) { try { callback(arg); } catch (e) { console.error('error running callback', callback, ':', e); } } } function scheduleAfterUiUpdateCallbacks() { clearTimeout(uiAfterUpdateTimeout); uiAfterUpdateTimeout = setTimeout(() => executeCallbacks(uiAfterUpdateCallbacks, 500)); } let executedOnLoaded = false; document.addEventListener('DOMContentLoaded', () => { const mutationObserver = new MutationObserver((m) => { if (!executedOnLoaded && gradioApp().getElementById('txt2img_prompt')) { executedOnLoaded = true; executeCallbacks(uiLoadedCallbacks); } if (executedOnLoaded) { executeCallbacks(uiUpdateCallbacks, m); scheduleAfterUiUpdateCallbacks(); } const newTab = getUICurrentTab(); 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', (e) => { let handled = false; if (e.key !== undefined) { if ((e.key === 'Enter' && (e.metaKey || e.ctrlKey || e.altKey))) handled = true; } else if (e.keyCode !== undefined) { if ((e.keyCode === 13 && (e.metaKey || e.ctrlKey || e.altKey))) handled = true; } if (handled) { const button = getUICurrentTabContent().querySelector('button[id$=_generate]'); if (button) button.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; }