From a8ecf3f31d420e9c250addc1acf2046f5aa5fe1d Mon Sep 17 00:00:00 2001
From: Daddy Raegen <54722924+DaddyRaegen@users.noreply.github.com>
Date: Fri, 6 Mar 2026 20:33:21 -0500
Subject: [PATCH] refactor: encapsulate task clearing behind
 TaskManager.clear_all()

---
 backend/main.py        | 3 +--
 backend/utils/tasks.py | 5 +++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index 3f58aa8b..c9d7e7f7 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -1616,8 +1616,7 @@ async def clear_all_tasks():
     task_manager = get_task_manager()
     progress_manager = get_progress_manager()
 
-    task_manager._active_downloads.clear()
-    task_manager._active_generations.clear()
+    task_manager.clear_all()
 
     with progress_manager._lock:
         progress_manager._progress.clear()
diff --git a/backend/utils/tasks.py b/backend/utils/tasks.py
index da13011a..8baf71c3 100644
--- a/backend/utils/tasks.py
+++ b/backend/utils/tasks.py
@@ -76,6 +76,11 @@ class TaskManager:
         """Cancel/dismiss a download task (removes it from active list)."""
         return self._active_downloads.pop(model_name, None) is not None
 
+    def clear_all(self) -> None:
+        """Clear all download and generation tasks."""
+        self._active_downloads.clear()
+        self._active_generations.clear()
+
     def is_download_active(self, model_name: str) -> bool:
         """Check if a download is active."""
         return model_name in self._active_downloads