From 19405ee960339bde924ff6137d87d95968f6b579 Mon Sep 17 00:00:00 2001
From: ManInDark <61268856+ManInDark@users.noreply.github.com>
Date: Thu, 29 Jun 2023 00:28:17 +0200
Subject: [PATCH] Changed from showing percentage to how many images are left
 to create

---
 ui/media/js/main.js | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 9bfb8ef3..6c138061 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1144,10 +1144,7 @@ function getTaskUpdater(task, reqBody, outputContainer) {
                 stepUpdate.total_steps * (batchCount - task.batchesDone) // Initial value at (unstarted task count * Nbr of steps)
             )
             const percent = Math.min(100, 100 * (overallStepCount / totalSteps)).toFixed(0)
-            if (SHOW_PROGRESS) {
-                document.title = `${percent}% - Easy Diffusion` 
-            }
-
+            
             const timeTaken = stepUpdate.step_time // sec
             const stepsRemaining = Math.max(0, totalSteps - overallStepCount)
             const timeRemaining = timeTaken < 0 ? "" : millisecondsToStr(stepsRemaining * timeTaken * 1000)
@@ -1289,6 +1286,15 @@ function onTaskCompleted(task, reqBody, instance, outputContainer, stepUpdate) {
         playSound()
     }
     if (SHOW_PROGRESS) {
+        updateTitle()
+    }
+}
+
+function updateTitle() {
+    let img_remaining = [...document.querySelectorAll("div .imageTaskContainer").entries()].map(c => htmlTaskMap.get(c[1])).filter(task => task.isProcessing).map(task => task.numOutputsTotal - Math.max(0, (task.batchesDone - 1) * task.reqBody.num_outputs)).reduce((total, value) => total + value, 0);
+    if (img_remaining > 0) {
+        document.title = `${img_remaining} - Easy Diffusion`;
+    } else {
         document.title = "Completed - Easy Diffusion"
     }
 }
@@ -1384,6 +1390,9 @@ async function onTaskStart(task) {
     renderButtons.style.display = "flex"
     renameMakeImageButton()
     updateInitialText()
+    if (SHOW_PROGRESS) {
+        updateTitle();
+    }
 }
 
 /* Hover effect for the init image in the task list */