From 5f2568320fc663c2bc418e94b7c36ccdfb9be7b0 Mon Sep 17 00:00:00 2001
From: bluelovers <codelovers@users.sourceforge.net>
Date: Wed, 8 May 2024 01:25:41 +0800
Subject: [PATCH] chore: add ImageViewer Navigation Index label

---
 javascript/imageviewer.js | 30 +++++++++++++++++++++---------
 style.css                 | 13 +++++++++++++
 2 files changed, 34 insertions(+), 9 deletions(-)

diff --git a/javascript/imageviewer.js b/javascript/imageviewer.js
index d4d4f016d..98e9d548a 100644
--- a/javascript/imageviewer.js
+++ b/javascript/imageviewer.js
@@ -23,6 +23,19 @@ function showModal(event) {
         gradioApp().getElementById("modal_save").style.display = "none";
     }
     event.stopPropagation();
+
+    setTimeout(() => {
+        updateModalNavigationIndex(selected_gallery_index(), all_gallery_buttons().length);
+    }, 10);
+}
+
+/**
+ * @param idx {number}
+ * @param length {number}
+ */
+function updateModalNavigationIndex(idx, length) {
+    const modalTitle = gradioApp().getElementById("lightboxModal").querySelector('.modalTitle');
+    modalTitle.textContent = `${idx + 1}／${length}`;
 }
 
 function negmod(n, m) {
@@ -51,17 +64,11 @@ function modalImageSwitch(offset) {
     var galleryButtons = all_gallery_buttons();
 
     if (galleryButtons.length > 1) {
-        var currentButton = selected_gallery_button();
-
-        var result = -1;
-        galleryButtons.forEach(function(v, i) {
-            if (v == currentButton) {
-                result = i;
-            }
-        });
+        var result = selected_gallery_index();
 
         if (result != -1) {
-            var nextButton = galleryButtons[negmod((result + offset), galleryButtons.length)];
+            let idx = negmod((result + offset), galleryButtons.length);
+            var nextButton = galleryButtons[idx];
             nextButton.click();
             const modalImage = gradioApp().getElementById("modalImage");
             const modal = gradioApp().getElementById("lightboxModal");
@@ -69,6 +76,7 @@ function modalImageSwitch(offset) {
             if (modalImage.style.display === 'none') {
                 modal.style.setProperty('background-image', `url(${modalImage.src})`);
             }
+            updateModalNavigationIndex(idx, galleryButtons.length);
             setTimeout(function() {
                 modal.focus();
             }, 10);
@@ -216,6 +224,10 @@ document.addEventListener("DOMContentLoaded", function() {
     modalSave.title = "Save Image(s)";
     modalControls.appendChild(modalSave);
 
+    const modalTitle = document.createElement("span");
+    modalTitle.className = "modalTitle";
+    modalControls.appendChild(modalTitle);
+
     const modalClose = document.createElement('span');
     modalClose.className = 'modalClose cursor';
     modalClose.innerHTML = '&times;';
diff --git a/style.css b/style.css
index f6a89b8f9..4e214b3db 100644
--- a/style.css
+++ b/style.css
@@ -673,6 +673,7 @@ table.popup-table .link{
     user-select: none;
     -webkit-user-select: none;
     flex-direction: column;
+    font-family: var(--font);
 }
 
 .modalControls {
@@ -706,6 +707,18 @@ table.popup-table .link{
     text-decoration: none;
 }
 
+.modalControls .modalTitle {
+    display: block;
+    margin-left: auto;
+    width: 100% !important;
+    white-space: nowrap;
+    text-align: center;
+}
+
+.modalControls .modalTitle:hover {
+    color: white;
+}
+
 #lightboxModal > img {
     display: block;
     margin: auto;