script.js 2.3 KB
Newer Older
1
function gradioApp(){
2
    return document.getElementsByTagName('gradio-app')[0].shadowRoot;
3 4
}

5 6 7 8
function get_uiCurrentTab() {
    return gradioApp().querySelector('.tabs button:not(.border-transparent)')
}

9 10 11 12
function get_uiCurrentTabContent() {
    return gradioApp().querySelector('.tabitem[id^=tab_]:not([style*="display: none"])')
}

A
AUTOMATIC 已提交
13
uiUpdateCallbacks = []
14 15 16
uiTabChangeCallbacks = []
let uiCurrentTab = null

A
AUTOMATIC 已提交
17 18
function onUiUpdate(callback){
    uiUpdateCallbacks.push(callback)
D
DepFA 已提交
19
}
20 21 22
function onUiTabChange(callback){
    uiTabChangeCallbacks.push(callback)
}
D
DepFA 已提交
23

A
AUTOMATIC 已提交
24
function runCallback(x, m){
25
    try {
A
AUTOMATIC 已提交
26
        x(m)
27 28 29 30
    } catch (e) {
        (console.error || console.log).call(console, e.message, e);
    }
}
A
AUTOMATIC 已提交
31 32
function executeCallbacks(queue, m) {
    queue.forEach(function(x){runCallback(x, m)})
33 34 35 36
}

document.addEventListener("DOMContentLoaded", function() {
    var mutationObserver = new MutationObserver(function(m){
A
AUTOMATIC 已提交
37
        executeCallbacks(uiUpdateCallbacks, m);
38 39 40 41 42
        const newTab = get_uiCurrentTab();
        if ( newTab && ( newTab !== uiCurrentTab ) ) {
            uiCurrentTab = newTab;
            executeCallbacks(uiTabChangeCallbacks);
        }
43
    });
44
    mutationObserver.observe( gradioApp(), { childList:true, subtree:true })
45
});
46

47 48 49 50 51 52
/**
 * Add a ctrl+enter as a shortcut to start a generation
 */
 document.addEventListener('keydown', function(e) {
    var handled = false;
    if (e.key !== undefined) {
A
aoirusann 已提交
53
        if((e.key == "Enter" && (e.metaKey || e.ctrlKey || e.altKey))) handled = true;
54
    } else if (e.keyCode !== undefined) {
A
aoirusann 已提交
55
        if((e.keyCode == 13 && (e.metaKey || e.ctrlKey || e.altKey))) handled = true;
56
    }
57 58 59 60 61
    if (handled) {
        button = get_uiCurrentTabContent().querySelector('button[id$=_generate]');
        if (button) {
            button.click();
        }
62 63 64 65
        e.preventDefault();
    }
})

66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
/**
 * checks that a UI element is not in another hidden element or tab content
 */
function uiElementIsVisible(el) {
    let isVisible = !el.closest('.\\!hidden');
    if ( ! isVisible ) {
        return false;
    }

    while( isVisible = el.closest('.tabitem')?.style.display !== 'none' ) {
        if ( ! isVisible ) {
            return false;
        } else if ( el.parentElement ) {
            el = el.parentElement
        } else {
            break;
        }
    }
    return isVisible;
}