Update app.py

app.py

@@ -39,13 +39,14 @@ class TrainingState:
             self.logs.pop(0)
             
     def complete_process(self):
-        if self.start_time is not None:
-            elapsed = time.time() - self.start_time
-            self.add_log(f"{self.status.capitalize()} completed in {elapsed:.1f} seconds!")
-        else:
-            self.add_log(f"{self.status.capitalize()} completed!")
-        self.status = "idle"
-        self.progress = 100
+    current_status = self.status  # save before setting to idle
+    if self.start_time is not None:
+        elapsed = time.time() - self.start_time
+        self.add_log(f"{current_status.capitalize()} completed in {elapsed:.1f} seconds!")
+    else:
+        self.add_log(f"{current_status.capitalize()} completed!")
+    self.status = "idle"
+    self.progress = 100
         
     def get_status(self):
         status_map = {