From f18d4413e8ad0c2eb7cad4e97c860520266a5542 Mon Sep 17 00:00:00 2001
From: ManInDark <61268856+ManInDark@users.noreply.github.com>
Date: Fri, 30 Jun 2023 23:21:07 +0200
Subject: [PATCH] Applied changes proposed by @cmdr2

---
 ui/media/js/main.js | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index 4cced947..7bef003f 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1179,7 +1179,7 @@ function abortTask(task) {
         }
     })
     if (task.batchesDone > 0) {
-        document.title = "Stopped - Easy Diffusion"
+        document.title = "Easy Diffusion"
     }
 }
 
@@ -1290,7 +1290,10 @@ function onTaskCompleted(task, reqBody, instance, outputContainer, stepUpdate) {
 }
 
 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);
+    let all_tasks = [...document.querySelectorAll("div .imageTaskContainer").entries()].map(c => htmlTaskMap.get(c[1]))
+    let tasks_to_be_run = all_tasks.filter(task => task.isProcessing)
+    let img_remaining_per_task = tasks_to_be_run.map(task => task.numOutputsTotal - Math.max(0, (task.batchesDone - 1) * task.reqBody.num_outputs))
+    let img_remaining = img_remaining_per_task.reduce((total, value) => total + value, 0);
     if (img_remaining > 0) {
         document.title = `${img_remaining} - Easy Diffusion`;
     } else {