diff options
| author | siutin <osiutino@gmail.com> | 2023-02-06 03:55:31 +0800 |
|---|---|---|
| committer | siutin <osiutino@gmail.com> | 2023-03-30 17:20:09 +0800 |
| commit | 4242e194e417ec5008d09ec6d756594ac65f77bd (patch) | |
| tree | a39f18034e3bef43db01b98a52145e06603124b3 /modules/progress.py | |
| parent | 9407f1731aa8c112ffc0efaa611a76f7fead3d0c (diff) | |
add a button to restore the current progress
Diffstat (limited to 'modules/progress.py')
| -rw-r--r-- | modules/progress.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/modules/progress.py b/modules/progress.py index 27a336ad..36963c92 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -48,6 +48,20 @@ def set_last_task_result(id_job, result): last_task_result = result
+def restore_progress_call(task_tag):
+ if current_task is None or not current_task[5:-1].startswith(task_tag):
+
+ # image, generation_info, html_info, html_log
+ return tuple(list([None, None, None, None]))
+
+ else:
+
+ t_task = current_task
+ while t_task != last_task_id:
+ time.sleep(2.5)
+ return last_task_result
+
+
class CurrentTaskResponse(BaseModel):
current_task: str = Field(default=None, title="Task ID", description="id of the current progress task")
|