From 3f12ff2830122b568dfef00418ded3db202f8597 Mon Sep 17 00:00:00 2001
From: Manav Rathi <manav@mrmr.io>
Date: Tue, 27 Aug 2024 14:01:51 +0530
Subject: [PATCH] Opt

---
 .../new/photos/components/MLSettings.tsx      | 22 ++++++++++++++++++-
 web/packages/new/photos/services/ml/index.ts  |  4 ++--
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/web/packages/new/photos/components/MLSettings.tsx b/web/packages/new/photos/components/MLSettings.tsx
index 55fc07c466..f6ed650a1f 100644
--- a/web/packages/new/photos/components/MLSettings.tsx
+++ b/web/packages/new/photos/components/MLSettings.tsx
@@ -1,12 +1,14 @@
 import { EnteDrawer } from "@/base/components/EnteDrawer";
-import { MenuItemGroup } from "@/base/components/Menu";
+import { MenuItemGroup, MenuSectionTitle } from "@/base/components/Menu";
 import { Titlebar } from "@/base/components/Titlebar";
+import { ut } from "@/base/i18n";
 import log from "@/base/log";
 import {
     disableML,
     enableML,
     mlStatusSnapshot,
     mlStatusSubscribe,
+    wipClusterEnable,
     type MLStatus,
 } from "@/new/photos/services/ml";
 import EnteSpinner from "@ente/shared/components/EnteSpinner";
@@ -295,8 +297,11 @@ const ManageML: React.FC<ManageMLProps> = ({
     onDisableML,
     setDialogBoxAttributesV2,
 }) => {
+    const [showClusterOpt, setShowClusterOpt] = useState(false);
     const { phase, nSyncedFiles, nTotalFiles } = mlStatus;
 
+    useEffect(() => void wipClusterEnable().then(setShowClusterOpt), []);
+
     let status: string;
     switch (phase) {
         case "scheduled":
@@ -372,6 +377,21 @@ const ManageML: React.FC<ManageMLProps> = ({
                     </Stack>
                 </Stack>
             </Paper>
+            {showClusterOpt && (
+                <Box>
+                    <MenuItemGroup>
+                        <EnteMenuItem
+                            label={ut("Cluster   ––– internal only option")}
+                            onClick={confirmDisableML}
+                        />
+                    </MenuItemGroup>
+                    <MenuSectionTitle
+                        title={ut(
+                            "(Re)-Create clusters locally, afresh. They will not be synced to remote.",
+                        )}
+                    />
+                </Box>
+            )}
         </Stack>
     );
 };
diff --git a/web/packages/new/photos/services/ml/index.ts b/web/packages/new/photos/services/ml/index.ts
index 7c3cc669f3..86fb3fdc75 100644
--- a/web/packages/new/photos/services/ml/index.ts
+++ b/web/packages/new/photos/services/ml/index.ts
@@ -323,8 +323,8 @@ export const indexNewUpload = (enteFile: EnteFile, uploadItem: UploadItem) => {
 /**
  * WIP! Don't enable, dragon eggs are hatching here.
  */
-export const wipClusterEnable = async () =>
-    process.env.NEXT_PUBLIC_ENTE_WIP_CL &&
+export const wipClusterEnable = async (): Promise<boolean> =>
+    !!process.env.NEXT_PUBLIC_ENTE_WIP_CL &&
     isDevBuild &&
     (await isInternalUser());