34 lines
1.0 KiB
JavaScript
34 lines
1.0 KiB
JavaScript
// based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/v1.6.0/script.js
|
|
|
|
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;
|
|
}
|
|
|
|
function playNotification() {
|
|
gradioApp().querySelector('#audio_notification audio')?.play();
|
|
}
|
|
|
|
document.addEventListener('keydown', function(e) {
|
|
var 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) {
|
|
var button = gradioApp().querySelector('button[id=generate_button]');
|
|
if (button) {
|
|
button.click();
|
|
}
|
|
e.preventDefault();
|
|
}
|
|
});
|