From 303df25cc287365b43ede22e0c082ae06aa452b2 Mon Sep 17 00:00:00 2001
From: Bwin4L <Bwin4L@users.noreply.github.com>
Date: Sat, 10 Dec 2022 22:58:06 +0100
Subject: [PATCH] Make the generated image count only count new images in the
 currently active tab

---
 javascript/notification.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/javascript/notification.js b/javascript/notification.js
index f96de313..040a3afa 100644
--- a/javascript/notification.js
+++ b/javascript/notification.js
@@ -15,7 +15,7 @@ onUiUpdate(function(){
         }
     }
 
-    const galleryPreviews = gradioApp().querySelectorAll('img.h-full.w-full.overflow-hidden');
+    const galleryPreviews = gradioApp().querySelectorAll('div[id^="tab_"][style*="display: block"] img.h-full.w-full.overflow-hidden');
 
     if (galleryPreviews == null) return;