diff --git a/javascript/images_history.js b/javascript/image_browser.js similarity index 57% rename from javascript/images_history.js rename to javascript/image_browser.js index c3ab670..3f70467 100644 --- a/javascript/images_history.js +++ b/javascript/image_browser.js @@ -1,6 +1,6 @@ -var images_history_click_image = function(){ +var image_browser_click_image = function(){ if (!this.classList?.contains("transform")){ - var gallery = images_history_get_parent_by_class(this, "images_history_cantainor"); + var gallery = image_browser_get_parent_by_class(this, "image_browser_container"); var buttons = gallery.querySelectorAll(".gallery-item"); var i = 0; var hidden_list = []; @@ -11,13 +11,13 @@ var images_history_click_image = function(){ i += 1; }) if (hidden_list.length > 0){ - setTimeout(images_history_hide_buttons, 10, hidden_list, gallery); + setTimeout(image_browser_hide_buttons, 10, hidden_list, gallery); } } - images_history_set_image_info(this); + image_browser_set_image_info(this); } -function images_history_get_parent_by_class(item, class_name){ +function image_browser_get_parent_by_class(item, class_name){ var parent = item.parentElement; while(!parent.classList.contains(class_name)){ parent = parent.parentElement; @@ -25,7 +25,7 @@ function images_history_get_parent_by_class(item, class_name){ return parent; } -function images_history_get_parent_by_tagname(item, tagname){ +function image_browser_get_parent_by_tagname(item, tagname){ var parent = item.parentElement; tagname = tagname.toUpperCase() while(parent.tagName != tagname){ @@ -34,7 +34,7 @@ function images_history_get_parent_by_tagname(item, tagname){ return parent; } -function images_history_hide_buttons(hidden_list, gallery){ +function image_browser_hide_buttons(hidden_list, gallery){ var buttons = gallery.querySelectorAll(".gallery-item"); var num = 0; buttons.forEach(function(e){ @@ -43,15 +43,15 @@ function images_history_hide_buttons(hidden_list, gallery){ } }); if (num == hidden_list.length){ - setTimeout(images_history_hide_buttons, 10, hidden_list, gallery); + setTimeout(image_browser_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){ - var buttons = images_history_get_parent_by_tagname(button, "DIV").querySelectorAll(".gallery-item"); +function image_browser_set_image_info(button){ + var buttons = image_browser_get_parent_by_tagname(button, "DIV").querySelectorAll(".gallery-item"); var index = -1; var i = 0; buttons.forEach(function(e){ @@ -62,8 +62,8 @@ function images_history_set_image_info(button){ i += 1; } }); - var gallery = images_history_get_parent_by_class(button, "images_history_cantainor"); - var set_btn = gallery.querySelector(".images_history_set_index"); + var gallery = image_browser_get_parent_by_class(button, "image_browser_container"); + var set_btn = gallery.querySelector(".image_browser_set_index"); var curr_idx = set_btn.getAttribute("img_index", index); if (curr_idx != index) { set_btn.setAttribute("img_index", index); @@ -72,18 +72,18 @@ function images_history_set_image_info(button){ } -function images_history_get_current_img(tabname, img_index, page_index){ +function image_browser_get_current_img(tabname, img_index, page_index){ return [ tabname, - gradioApp().getElementById(tabname + '_images_history_set_index').getAttribute("img_index"), + gradioApp().getElementById(tabname + '_image_browser_set_index').getAttribute("img_index"), page_index ]; } -function images_history_delete(del_num, tabname, image_index){ +function image_browser_delete(del_num, tabname, image_index){ image_index = parseInt(image_index); - var tab = gradioApp().getElementById(tabname + '_images_history'); - var set_btn = tab.querySelector(".images_history_set_index"); + var tab = gradioApp().getElementById(tabname + '_image_browser'); + var set_btn = tab.querySelector(".image_browser_set_index"); var buttons = []; tab.querySelectorAll(".gallery-item").forEach(function(e){ if (e.style.display != 'none'){ @@ -94,7 +94,7 @@ function images_history_delete(del_num, tabname, image_index){ del_num = Math.min(img_num - image_index, del_num) if (img_num <= del_num){ setTimeout(function(tabname){ - gradioApp().getElementById(tabname + '_images_history_renew_page').click(); + gradioApp().getElementById(tabname + '_image_browser_renew_page').click(); }, 30, tabname); } else { var next_img @@ -114,78 +114,78 @@ function images_history_delete(del_num, tabname, image_index){ } -function images_history_turnpage(tabname){ - var buttons = gradioApp().getElementById(tabname + '_images_history').querySelectorAll(".gallery-item"); +function image_browser_turnpage(tabname){ + var buttons = gradioApp().getElementById(tabname + '_image_browser').querySelectorAll(".gallery-item"); buttons.forEach(function(elem) { elem.style.display = 'block'; }); } -function images_history_init(){ - var tabnames = gradioApp().getElementById("images_history_tabnames_list") +function image_browser_init(){ + var tabnames = gradioApp().getElementById("image_browser_tabnames_list") if (tabnames){ - images_history_tab_list = tabnames.querySelector("textarea").value.split(",") - for (var i in images_history_tab_list ){ - var tab = images_history_tab_list[i]; - gradioApp().getElementById(tab + '_images_history').classList.add("images_history_cantainor"); - gradioApp().getElementById(tab + '_images_history_set_index').classList.add("images_history_set_index"); - gradioApp().getElementById(tab + '_images_history_del_button').classList.add("images_history_del_button"); - gradioApp().getElementById(tab + '_images_history_gallery').classList.add("images_history_gallery"); + image_browser_tab_list = tabnames.querySelector("textarea").value.split(",") + for (var i in image_browser_tab_list ){ + var tab = image_browser_tab_list[i]; + gradioApp().getElementById(tab + '_image_browser').classList.add("image_browser_container"); + gradioApp().getElementById(tab + '_image_browser_set_index').classList.add("image_browser_set_index"); + gradioApp().getElementById(tab + '_image_browser_del_button').classList.add("image_browser_del_button"); + gradioApp().getElementById(tab + '_image_browser_gallery').classList.add("image_browser_gallery"); } //preload - var tab_btns = gradioApp().getElementById("images_history_tab").querySelector("div").querySelectorAll("button"); - for (var i in images_history_tab_list){ - var tabname = images_history_tab_list[i] + var tab_btns = gradioApp().getElementById("image_browser_tabs_container").querySelector("div").querySelectorAll("button"); + for (var i in image_browser_tab_list){ + var tabname = image_browser_tab_list[i] tab_btns[i].setAttribute("tabname", tabname); tab_btns[i].addEventListener('click', function(){ - var tabs_box = gradioApp().getElementById("images_history_tab"); + var tabs_box = gradioApp().getElementById("image_browser_tabs_container"); if (!tabs_box.classList.contains(this.getAttribute("tabname"))) { - gradioApp().getElementById(this.getAttribute("tabname") + "_images_history_renew_page").click(); + gradioApp().getElementById(this.getAttribute("tabname") + "_image_browser_renew_page").click(); tabs_box.classList.add(this.getAttribute("tabname")) } }); } - if (gradioApp().getElementById("images_history_preload").querySelector("input").checked ){ + if (gradioApp().getElementById("image_browser_preload").querySelector("input").checked ){ setTimeout(function(){tab_btns[0].click()}, 100); } } else { - setTimeout(images_history_init, 500); + setTimeout(image_browser_init, 500); } } let timer -var images_history_tab_list = ""; -setTimeout(images_history_init, 500); +var image_browser_tab_list = ""; +setTimeout(image_browser_init, 500); document.addEventListener("DOMContentLoaded", function() { var mutationObserver = new MutationObserver(function(m){ - if (images_history_tab_list != ""){ + if (image_browser_tab_list != ""){ - for (var i in images_history_tab_list ){ - let tabname = images_history_tab_list[i] - var buttons = gradioApp().querySelectorAll('#' + tabname + '_images_history .gallery-item'); + for (var i in image_browser_tab_list ){ + let tabname = image_browser_tab_list[i] + var buttons = gradioApp().querySelectorAll('#' + tabname + '_image_browser .gallery-item'); buttons.forEach(function(button){ - button.addEventListener('click', images_history_click_image, true); + button.addEventListener('click', image_browser_click_image, true); document.onkeyup = function(e) { - if (!checkImageBrowserActive()) { + if (!image_browser_active()) { return; } clearTimeout(timer) timer = setTimeout(() => { - var gallery_btn = gradioApp().getElementById(getCurrentTabName() +"_images_history_gallery").getElementsByClassName('gallery-item !flex-none !h-9 !w-9 transition-all duration-75 !ring-2 !ring-orange-500 hover:!ring-orange-500 svelte-1g9btlg'); + var gallery_btn = gradioApp().getElementById(image_browser_current_tab() + "_image_browser_gallery").getElementsByClassName('gallery-item !flex-none !h-9 !w-9 transition-all duration-75 !ring-2 !ring-orange-500 hover:!ring-orange-500 svelte-1g9btlg'); gallery_btn = gallery_btn && gallery_btn.length > 0 ? gallery_btn[0] : null; if (gallery_btn) { - images_history_click_image.call(gallery_btn) + image_browser_click_image.call(gallery_btn) } }, 500); } }); - var cls_btn = gradioApp().getElementById(tabname + '_images_history_gallery').querySelector("svg"); + var cls_btn = gradioApp().getElementById(tabname + '_image_browser_gallery').querySelector("svg"); if (cls_btn){ cls_btn.addEventListener('click', function(){ - gradioApp().getElementById(tabname + '_images_history_renew_page').click(); + gradioApp().getElementById(tabname + '_image_browser_renew_page').click(); }, false); } @@ -195,34 +195,34 @@ document.addEventListener("DOMContentLoaded", function() { mutationObserver.observe(gradioApp(), { childList:true, subtree:true }); }); -function getCurrentTabName() { - var tabs = gradioApp().getElementById("images_history_tab").querySelectorAll('[id$="_images_history_container"]'); +function image_browser_current_tab() { + var tabs = gradioApp().getElementById("image_browser_tabs_container").querySelectorAll('[id$="_image_browser_container"]'); for (const element of tabs) { if (element.style.display === "block") { const id = element.id; - const index = id.indexOf("_images_history_container"); + const index = id.indexOf("_image_browser_container"); const tabname = id.substring(0, index); return tabname; } } } -function checkImageBrowserActive() { - var ext_active = gradioApp().getElementById("tab_images_history"); +function image_browser_active() { + var ext_active = gradioApp().getElementById("tab_image_browser"); return ext_active && ext_active.style.display !== "none"; } gradioApp().addEventListener("keydown", function(event) { // If we are not on the Image Browser Extension, dont listen for keypresses - if (!checkImageBrowserActive()) { + if (!image_browser_active()) { return; } // Listens for keypresses 0-5 and updates the corresponding ranking (0 is the last option, None) if (event.code >= "Digit0" && event.code <= "Digit5") { var selectedValue = event.code.charAt(event.code.length - 1); - var radioInputs = gradioApp().getElementById( getCurrentTabName() + "_images_ranking").getElementsByTagName("input"); + var radioInputs = gradioApp().getElementById( image_browser_current_tab() + "_images_ranking").getElementsByTagName("input"); for (const input of radioInputs) { if (input.value === selectedValue || (selectedValue === '0' && input === radioInputs[radioInputs.length - 1])) { input.checked = true; diff --git a/scripts/images_history.py b/scripts/image_browser.py similarity index 87% rename from scripts/images_history.py rename to scripts/image_browser.py index 378a68e..72ced5a 100644 --- a/scripts/images_history.py +++ b/scripts/image_browser.py @@ -43,11 +43,11 @@ init = True logger = logging.getLogger(__name__) logger_mode = logging.ERROR -if hasattr(opts, "images_logger_warning"): - if opts.images_logger_warning: +if hasattr(opts, "image_browser_logger_warning"): + if opts.image_browser_logger_warning: logger_mode = logging.WARNING -if hasattr(opts, "images_logger_debug"): - if opts.images_logger_debug: +if hasattr(opts, "image_browser_logger_debug"): + if opts.image_browser_logger_debug: logger_mode = logging.DEBUG logger.setLevel(logger_mode) console_handler = logging.StreamHandler() @@ -55,7 +55,7 @@ console_handler.setLevel(logger_mode) logger.addHandler(console_handler) def delete_recycle(filename): - if opts.images_delete_recycle: + if opts.image_browser_delete_recycle: send2trash(filename) else: os.remove(filename) @@ -144,34 +144,34 @@ def reduplicative_file_move(src, dst): name = os.path.basename(src) save_name = os.path.join(dst, name) src_txt_exists = False - if opts.images_txt_files: + if opts.image_browser_txt_files: src_txt = totxt(src) if os.path.exists(src_txt): src_txt_exists = True if not os.path.exists(save_name): - if opts.images_copy_image: + if opts.image_browser_copy_image: shutil.copy2(src, dst) - if opts.images_txt_files and src_txt_exists: + if opts.image_browser_txt_files and src_txt_exists: shutil.copy2(src_txt, dst) else: shutil.move(src, dst) - if opts.images_txt_files and src_txt_exists: + if opts.image_browser_txt_files and src_txt_exists: shutil.move(src_txt, dst) else: name = same_name_file(name, dst) - if opts.images_copy_image: + if opts.image_browser_copy_image: shutil.copy2(src, os.path.join(dst, name)) - if opts.images_txt_files and src_txt_exists: + if opts.image_browser_txt_files and src_txt_exists: shutil.copy2(src_txt, totxt(os.path.join(dst, name))) else: shutil.move(src, os.path.join(dst, name)) - if opts.images_txt_files and src_txt_exists: + if opts.image_browser_txt_files and src_txt_exists: shutil.move(src_txt, totxt(os.path.join(dst, name))) def save_image(file_name, filenames, page_index, turn_page_switch): if file_name is not None and os.path.exists(file_name): reduplicative_file_move(file_name, opts.outdir_save) - if opts.images_copy_image: + if opts.image_browser_copy_image: message = "