From 329360aa5bd8eadbbf041cacec3d6f5fd7db6e5d Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 21 Oct 2022 15:18:05 +0530
Subject: [PATCH] Show the number of images it'll make in the Make Image button

---
 ui/index.html       |  2 +-
 ui/media/js/main.js | 19 +++++++++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/ui/index.html b/ui/index.html
index b714ff1a..62eb32c6 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -267,7 +267,7 @@
 <script src="media/js/inpainting-editor.js?v=1"></script>
 <script src="media/js/image-modifiers.js?v=3"></script>
 <script src="media/js/auto-save.js?v=1"></script>
-<script src="media/js/main.js?v=3"></script>
+<script src="media/js/main.js?v=4"></script>
 <script src="media/js/themes.js?v=1"></script>
 <script>
 async function init() {
diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 239b1354..6300ea4a 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -711,7 +711,7 @@ async function checkTasks() {
         setStatus('request', 'done', 'success')
         setTimeout(checkTasks, 500)
         stopImageBtn.style.display = 'none'
-        makeImageBtn.innerHTML = 'Make Image'
+        renameMakeImageButton()
 
         currentTask = null
 
@@ -728,7 +728,7 @@ async function checkTasks() {
     setStatus('request', 'fetching..')
 
     stopImageBtn.style.display = 'block'
-    makeImageBtn.innerHTML = 'Enqueue Next Image'
+    renameMakeImageButton()
     bellPending = true
 
     previewTools.style.display = 'block'
@@ -1123,6 +1123,21 @@ diskPathField.addEventListener('change', handleStringSettingChange(DISK_PATH_KEY
 widthField.addEventListener('change', onDimensionChange)
 heightField.addEventListener('change', onDimensionChange)
 
+function renameMakeImageButton() {
+    let totalImages = Math.max(parseInt(numOutputsTotalField.value), parseInt(numOutputsParallelField.value))
+    let imageLabel = 'Image'
+    if (totalImages > 1) {
+        imageLabel = totalImages + ' Images'
+    }
+    if (taskQueue.length == 0) {
+        makeImageBtn.innerText = 'Make ' + imageLabel
+    } else {
+        makeImageBtn.innerText = 'Enqueue Next ' + imageLabel
+    }
+}
+numOutputsTotalField.addEventListener('change', renameMakeImageButton)
+numOutputsParallelField.addEventListener('change', renameMakeImageButton)
+
 function onDimensionChange() {
     if (!maskSetting.checked) {
         return