diff --git a/javascript/ui.js b/javascript/ui.js new file mode 100644 index 00000000..5d864cf3 --- /dev/null +++ b/javascript/ui.js @@ -0,0 +1,87 @@ +function gradioApp(){ + return document.getElementsByTagName('gradio-app')[0].shadowRoot; +} + +uiUpdateCallbacks = [] +function onUiUpdate(callback){ + uiUpdateCallbacks.push(callback) +} + +function uiUpdate(root){ + uiUpdateCallbacks.forEach(function(x){ + x() + }) +} + +document.addEventListener("DOMContentLoaded", function() { + var mutationObserver = new MutationObserver(function(m){ + uiUpdate(gradioApp()); + }); + mutationObserver.observe( gradioApp(), { childList:true, subtree:true }) +}); + +function selected_gallery_index(){ + var gr = gradioApp() + var buttons = gradioApp().querySelectorAll(".gallery-item") + var button = gr.querySelector(".gallery-item.\\!ring-2") + + var result = -1 + buttons.forEach(function(v, i){ if(v==button) { result = i } }) + + return result +} + +function extract_image_from_gallery(gallery){ + if(gallery.length == 1){ + return gallery[0] + } + + index = selected_gallery_index() + + if (index < 0 || index >= gallery.length){ + return [null] + } + + return gallery[index]; +} + +function extract_image_from_gallery_img2img(gallery){ + gradioApp().querySelectorAll('button')[1].click(); + return extract_image_from_gallery(gallery); +} + +function extract_image_from_gallery_extras(gallery){ + gradioApp().querySelectorAll('button')[2].click(); + return extract_image_from_gallery(gallery); +} + +function requestProgress(){ + btn = gradioApp().getElementById("check_progress"); + if(btn==null) return; + + btn.click(); +} + +function submit(){ + window.setTimeout(requestProgress, 500) + + res = [] + for(var i=0;i