1 files changed, 124 insertions, 0 deletions
diff --git a/javascript/images_history.js b/javascript/images_history.js
new file mode 100644
index 00000000..9a3e00a0
--- /dev/null
+++ b/javascript/images_history.js
@@ -0,0 +1,124 @@
+images_history_tab_list = ["txt2img", "img2img", "extras"]
+function images_history_init(){ 
+    if (gradioApp().getElementById('txt2img_images_history_first_page') == null) {
+        setTimeout(images_history_init, 500)
+    } else {        
+        for (i in images_history_tab_list ){
+            tab = images_history_tab_list[i]
+            gradioApp().getElementById(tab + '_images_history').classList.add("images_history_gallery")
+            gradioApp().getElementById(tab + '_images_history_set_index').classList.add("images_history_set_index")      
+            
+        }
+        gradioApp().getElementById("txt2img_images_history_first_page").click()     
+    }    
+}
+setTimeout(images_history_init, 500)
+var images_history_button_actions = function(){
+    if (!this.classList.contains("transform")){        
+        gallery = this.parentElement
+        while(!gallery.classList.contains("images_history_gallery")){gallery = gallery.parentElement}        
+        buttons = gallery.querySelectorAll(".gallery-item")
+        i = 0
+        hidden_list = []
+        buttons.forEach(function(e){
+            if (e.style.display == "none"){
+                hidden_list.push(i)
+            }
+            i += 1
+        })
+        if (hidden_list.length > 0){
+            setTimeout(images_history_hide_buttons, 10, hidden_list, gallery)
+        }
+        
+    }    
+    images_history_set_image_info(this) 
+
+}
+onUiUpdate(function(){
+    for (i in images_history_tab_list ){
+        tab = images_history_tab_list[i]
+        buttons = gradioApp().querySelectorAll('#' + tab + '_images_history .gallery-item')
+        buttons.forEach(function(bnt){    
+            bnt.addEventListener('click', images_history_button_actions, true)
+         });
+    }
+})
+function images_history_hide_buttons(hidden_list, gallery){
+    buttons = gallery.querySelectorAll(".gallery-item")
+    num = 0
+    buttons.forEach(function(e){
+        if (e.style.display == "none"){
+            num += 1
+        }
+    })
+    if (num == hidden_list.length){
+        setTimeout(images_history_hide_buttons, 10, hidden_list, gallery)
+    } 
+    for( i in hidden_list){
+        buttons[hidden_list[i]].style.display = "none"
+    }    
+}
+
+function images_history_set_image_info(button){
+    item = button.parentElement
+    while(item.tagName != "DIV"){item = item.parentElement}    
+    buttons = item.querySelectorAll(".gallery-item")
+    index = -1
+    i = 0
+    buttons.forEach(function(e){
+        if(e==button){index = i}
+        if(e.style.display != "none"){
+            i += 1
+        }        
+    })
+    gallery = button.parentElement
+    while(!gallery.classList.contains("images_history_gallery")){gallery = gallery.parentElement}    
+    set_btn = gallery.querySelector(".images_history_set_index")
+    set_btn.setAttribute("img_index", index)
+    set_btn.click()
+}
+
+function images_history_get_current_img(tabname, image_path, files){
+    s =  gradioApp().getElementById(tabname + '_images_history_set_index').getAttribute("img_index")
+    return [s, image_path, files]
+}
+
+function images_history_delete(tabname, img_path, img_file_name, page_index, filenames, image_index){
+    image_index = parseInt(image_index)
+    tab = gradioApp().getElementById(tabname + '_images_history')
+    set_btn = tab.querySelector(".images_history_set_index")
+    buttons = []
+    tab.querySelectorAll(".gallery-item").forEach(function(e){
+        if (e.style.display != 'none'){
+            buttons.push(e)
+        }
+    })
+
+    
+    img_num = buttons.length / 2
+    if (img_num == 1){
+            setTimeout(function(tabname){
+                gradioApp().getElementById(tabname + '_images_history_renew_page').click()
+            }, 30, tabname) 
+    } else {
+        buttons[image_index].style.display = 'none'
+        buttons[image_index + img_num].style.display = 'none'
+        if (image_index  >= img_num - 1){
+            console.log(buttons.length, img_num)
+            btn = buttons[img_num - 2]
+        } else {
+            btn = buttons[image_index + 1]           
+        }    
+        setTimeout(function(btn){btn.click()}, 30, btn)
+    }  
+    
+    return [tabname, img_path, img_file_name, page_index, filenames, image_index]
+}
+
+function images_history_turnpage(img_path, page_index, image_index, tabname){
+    buttons = gradioApp().getElementById(tabname + '_images_history').querySelectorAll(".gallery-item")
+    buttons.forEach(function(elem) {
+         elem.style.display = 'block'
+    })
+    return [img_path, page_index, image_index, tabname]
+}