From 9407f1731aa8c112ffc0efaa611a76f7fead3d0c Mon Sep 17 00:00:00 2001
From: siutin <osiutino@gmail.com>
Date: Mon, 6 Feb 2023 03:53:05 +0800
Subject: store the last generated result

---
 modules/progress.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'modules/progress.py')

diff --git a/modules/progress.py b/modules/progress.py
index 05032ac5..27a336ad 100644
--- a/modules/progress.py
+++ b/modules/progress.py
@@ -37,6 +37,16 @@ def finish_task(id_task):
 def add_task_to_queue(id_job):
     pending_tasks[id_job] = time.time()
 
+last_task_id = None
+last_task_result = None
+
+def set_last_task_result(id_job, result):
+  global last_task_id
+  global last_task_result
+
+  last_task_id = id_job
+  last_task_result = result
+
 
 class CurrentTaskResponse(BaseModel):
   current_task: str = Field(default=None, title="Task ID", description="id of the current progress task")
-- 
cgit v1.2.3