From b26efff8c496309329cd1982aee55e81bf81a655 Mon Sep 17 00:00:00 2001
From: ruocaled <ruocaled@gmail.com>
Date: Fri, 14 Oct 2022 17:14:59 -0700
Subject: [PATCH] allow re-open for multiple images gallery

---
 javascript/progressbar.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/javascript/progressbar.js b/javascript/progressbar.js
index 196fe507..574fd549 100644
--- a/javascript/progressbar.js
+++ b/javascript/progressbar.js
@@ -67,12 +67,13 @@ function check_gallery(id_gallery){
         if(galleryObservers[id_gallery]){
             galleryObservers[id_gallery].disconnect();
         }
+        let prevSelectedIndex = selected_gallery_index();
         galleryObservers[id_gallery] = new MutationObserver(function (){
             let galleryButtons = gradioApp().querySelectorAll('#'+id_gallery+' .gallery-item')
             let galleryBtnSelected = gradioApp().querySelector('#'+id_gallery+' .gallery-item.\\!ring-2')
-            if (galleryButtons.length === 1 && !galleryBtnSelected) {
-                //automatically open when there is only 1 gallery btn, and was previously selected
-                galleryButtons[0].click();
+            if (prevSelectedIndex !== -1 && galleryButtons.length>prevSelectedIndex && !galleryBtnSelected) {
+                //automatically re-open previously selected index (if exists)
+                galleryButtons[prevSelectedIndex].click();
             }
         })
         galleryObservers[id_gallery].observe( gallery, { childList:true, subtree:false })