diff --git a/javascript/progressbar.js b/javascript/progressbar.js index f10dc174..a627b724 100644 --- a/javascript/progressbar.js +++ b/javascript/progressbar.js @@ -1,15 +1,6 @@ -function gradioApp(){ - return document.getElementsByTagName('gradio-app')[0].shadowRoot; -} - global_progressbar = null -uiUpdateCallbacks = [] -function onUiUpdate(callback){ - uiUpdateCallbacks.push(callback) -} - -function uiUpdate(root){ +onUiUpdate(function(){ progressbar = root.getElementById('progressbar') if(progressbar!= null && progressbar != global_progressbar){ global_progressbar = progressbar @@ -30,87 +21,9 @@ function uiUpdate(root){ img2img_preview.style.width = img2img_gallery.clientWidth + "px" img2img_preview.style.height = img2img_gallery.clientHeight + "px" } - + window.setTimeout(requestProgress, 500) }); mutationObserver.observe( progressbar, { childList:true, subtree:true }) } - - - 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