fix Replace preview

fix broken Replace preview for extra networks tabs edit metadata caused by #11808
AUTOMATIC1111 · Jun 30, 2024 · ed76a0e · ed76a0e
1 parent a30b19d
commit ed76a0e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/javascript/ui.js b/javascript/ui.js
@@ -26,6 +26,11 @@ function selected_gallery_index() {
     return all_gallery_buttons().findIndex(elem => elem.classList.contains('selected'));
 }
 
+function selected_gallery_index_gallery_container(gallery_container) {
+    var tabGalleryButtons = gradioApp().querySelectorAll(`#${gallery_container} .thumbnail-item.thumbnail-small`);
+    return Array.from(tabGalleryButtons).findIndex(elem => elem.classList.contains('selected'));
+}
+
 function extract_image_from_gallery(gallery) {
     if (gallery.length == 0) {
         return [null];

diff --git a/modules/ui_extra_networks_user_metadata.py b/modules/ui_extra_networks_user_metadata.py
@@ -194,7 +194,7 @@ def save_preview(self, index, gallery, name):
     def setup_ui(self, gallery):
         self.button_replace_preview.click(
             fn=self.save_preview,
-            _js="function(x, y, z){return [selected_gallery_index(), y, z]}",
+            _js=f"function(x, y, z){{return [selected_gallery_index_gallery_container('{self.tabname + '_gallery_container'}'), y, z]}}",
             inputs=[self.edit_name_input, gallery, self.edit_name_input],
             outputs=[self.html_preview, self.html_status]
         ).then(