From cf1e8fcb303a21ab626fc1e8b3bc95bb780e8758 Mon Sep 17 00:00:00 2001
From: Kalle <kalekkiorg@gmail.com>
Date: Thu, 13 Oct 2022 00:12:20 +0300
Subject: Correct img gen count in notification

Display correct count of images generated in browser notification regardless of "Show grid in results for web" setting.
---
 javascript/notification.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'javascript/notification.js')

diff --git a/javascript/notification.js b/javascript/notification.js
index bdf614ad..f96de313 100644
--- a/javascript/notification.js
+++ b/javascript/notification.js
@@ -36,7 +36,7 @@ onUiUpdate(function(){
     const notification = new Notification(
         'Stable Diffusion',
         {
-            body: `Generated ${imgs.size > 1 ? imgs.size - 1 : 1} image${imgs.size > 1 ? 's' : ''}`,
+            body: `Generated ${imgs.size > 1 ? imgs.size - opts.return_grid : 1} image${imgs.size > 1 ? 's' : ''}`,
             icon: headImg,
             image: headImg,
         }
-- 
cgit v1.2.3


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: 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(-)

(limited to 'javascript/notification.js')

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;
 
-- 
cgit v1.2.3