From 5ff9f408eba1ff97bbbe121da0e1c0199c8df8ae Mon Sep 17 00:00:00 2001
From: Manav Rathi <manav@mrmr.io>
Date: Thu, 11 Jul 2024 12:07:27 +0530
Subject: [PATCH] Touchups

---
 web/packages/new/photos/services/ml/index.ts | 25 +++++++++-----------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/web/packages/new/photos/services/ml/index.ts b/web/packages/new/photos/services/ml/index.ts
index c599368fc5..e0084045d1 100644
--- a/web/packages/new/photos/services/ml/index.ts
+++ b/web/packages/new/photos/services/ml/index.ts
@@ -3,10 +3,6 @@
  */
 
 import { FILE_TYPE } from "@/media/file-type";
-import {
-    isBetaUser,
-    isInternalUser,
-} from "@/new/photos/services/feature-flags";
 import type { EnteFile } from "@/new/photos/types/file";
 import { isDesktop } from "@/next/app";
 import { blobCache } from "@/next/blob-cache";
@@ -97,12 +93,17 @@ export const terminateMLWorker = () => {
     }
 };
 
+/** Gatekeep some common entry points to catch accidental invocations. */
+const ensureDesktop = () => {
+    // ML currently only works when we're running in our desktop app.
+    if (!isDesktop) throw new Error("ML subsystem can only be used on desktop");
+};
+
 /**
  * Initialize the ML subsystem if the user has enabled it in preferences.
  */
 export const initML = () => {
-    // ML currently only works when we're running in our desktop app.
-    if (!isDesktop) return;
+    ensureDesktop();
     _isMLEnabledLocal = isMLEnabledLocally();
 };
 
@@ -118,13 +119,6 @@ export const logoutML = async () => {
     await clearMLDB();
 };
 
-/**
- * Return true if we should show an UI option to the user to allow them to
- * enable ML.
- */
-export const canEnableML = async () =>
-    (await isInternalUser()) || (await isBetaUser());
-
 /**
  * Return true if the user has enabled machine learning in their preferences.
  *
@@ -251,7 +245,10 @@ const updateIsMLEnabledRemote = (enabled: boolean) =>
  * This function does not wait for these processes to run to completion, and
  * returns immediately.
  */
-export const triggerMLSync = () => void mlSync();
+export const triggerMLSync = () => {
+    ensureDesktop();
+    void mlSync();
+};
 
 const mlSync = async () => {
     _isMLEnabledRemote = await getIsMLEnabledRemote();