From eba910e2023744b8b882245df52f63a782b2d04a Mon Sep 17 00:00:00 2001
From: Manav Rathi <manav@mrmr.io>
Date: Wed, 15 May 2024 13:30:26 +0530
Subject: [PATCH] mlwm

---
 web/apps/photos/src/services/logout.ts           |  2 +-
 .../services/machineLearning/mlWorkManager.ts    | 16 ++++++----------
 2 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/web/apps/photos/src/services/logout.ts b/web/apps/photos/src/services/logout.ts
index 73db5cf3fa..7431923c84 100644
--- a/web/apps/photos/src/services/logout.ts
+++ b/web/apps/photos/src/services/logout.ts
@@ -31,7 +31,7 @@ export const photosLogout = async () => {
     const electron = globalThis.electron;
     if (electron) {
         try {
-            await mlWorkManager.setMlSearchEnabled(false);
+            await mlWorkManager.logout();
         } catch (e) {
             log.error("Ignoring error during logout (ML)", e);
         }
diff --git a/web/apps/photos/src/services/machineLearning/mlWorkManager.ts b/web/apps/photos/src/services/machineLearning/mlWorkManager.ts
index cacb6d2376..4b7f8e3f46 100644
--- a/web/apps/photos/src/services/machineLearning/mlWorkManager.ts
+++ b/web/apps/photos/src/services/machineLearning/mlWorkManager.ts
@@ -175,16 +175,12 @@ class MLWorkManager {
         }
     }
 
-    private async logoutHandler() {
-        log.info("logoutHandler");
-        try {
-            this.stopSyncJob();
-            this.mlSyncJob = undefined;
-            await this.terminateLiveSyncWorker();
-            await mlIDbStorage.clearMLDB();
-        } catch (e) {
-            log.error("Failed in ML logout Handler", e);
-        }
+    async logout() {
+        this.setMlSearchEnabled(false);
+        this.stopSyncJob();
+        this.mlSyncJob = undefined;
+        await this.terminateLiveSyncWorker();
+        await mlIDbStorage.clearMLDB();
     }
 
     private async fileUploadedHandler(arg: {