From ac04ceadce1d91174d8cffee9a202bb517e9019a Mon Sep 17 00:00:00 2001
From: Manav Rathi <manav@mrmr.io>
Date: Thu, 26 Sep 2024 21:04:48 +0530
Subject: [PATCH] Delete

---
 .../components/Gallery/PeopleHeader.tsx       | 34 +++++++++++++++++--
 web/packages/new/photos/services/ml/index.ts  | 11 ++++++
 2 files changed, 43 insertions(+), 2 deletions(-)

diff --git a/web/packages/new/photos/components/Gallery/PeopleHeader.tsx b/web/packages/new/photos/components/Gallery/PeopleHeader.tsx
index ea845b6b20..013e034c0e 100644
--- a/web/packages/new/photos/components/Gallery/PeopleHeader.tsx
+++ b/web/packages/new/photos/components/Gallery/PeopleHeader.tsx
@@ -8,7 +8,11 @@
  */
 
 import { pt } from "@/base/i18n";
-import { addPerson, renamePerson } from "@/new/photos/services/ml/";
+import {
+    addPerson,
+    deletePerson,
+    renamePerson,
+} from "@/new/photos/services/ml/";
 import { type Person } from "@/new/photos/services/ml/people";
 import OverflowMenu from "@ente/shared/components/OverflowMenu/menu";
 import { OverflowMenuOption } from "@ente/shared/components/OverflowMenu/option";
@@ -16,6 +20,7 @@ import AddIcon from "@mui/icons-material/Add";
 import EditIcon from "@mui/icons-material/Edit";
 import MoreHoriz from "@mui/icons-material/MoreHoriz";
 import { IconButton, Stack, Tooltip } from "@mui/material";
+import { ClearIcon } from "@mui/x-date-pickers";
 import { t } from "i18next";
 import React, { useState } from "react";
 import type { FaceCluster } from "../../services/ml/cluster";
@@ -23,6 +28,7 @@ import type { CGroup } from "../../services/user-entity";
 import type { NewAppContextPhotos } from "../../types/context";
 import { SpaceBetweenFlex } from "../mui-custom";
 import { NameInputDialog } from "../NameInputDialog";
+import { useWrapLoadError } from "../use-wrap";
 import { GalleryItemsHeaderAdapter, GalleryItemsSummary } from "./ListHeader";
 
 interface PeopleHeaderProps {
@@ -69,7 +75,10 @@ const CGroupPersonOptions: React.FC<CGroupPersonOptionsProps> = ({
     cgroup,
     appContext,
 }) => {
-    const { startLoading, finishLoading } = appContext;
+    const { startLoading, finishLoading, setDialogBoxAttributesV2 } =
+        appContext;
+
+    const wrap = useWrapLoadError(appContext);
 
     const [openAddNameInput, setOpenAddNameInput] = useState(false);
 
@@ -84,6 +93,20 @@ const CGroupPersonOptions: React.FC<CGroupPersonOptionsProps> = ({
         }
     };
 
+    const handleDeletePerson = () =>
+        setDialogBoxAttributesV2({
+            title: pt("Reset person?"),
+            content: pt(
+                "The name, face groupings and suggestions for this person will be reset",
+            ),
+            close: { text: t("cancel") },
+            proceed: {
+                text: t("RESET"),
+                action: wrap(() => deletePerson(cgroup)),
+            },
+            buttonDirection: "row",
+        });
+
     return (
         <>
             <OverflowMenu
@@ -97,6 +120,13 @@ const CGroupPersonOptions: React.FC<CGroupPersonOptionsProps> = ({
                 >
                     {t("rename")}
                 </OverflowMenuOption>
+                <OverflowMenuOption
+                    startIcon={<ClearIcon />}
+                    centerAlign
+                    onClick={handleDeletePerson}
+                >
+                    {pt("Reset")}
+                </OverflowMenuOption>
             </OverflowMenu>
 
             <NameInputDialog
diff --git a/web/packages/new/photos/services/ml/index.ts b/web/packages/new/photos/services/ml/index.ts
index 4858aded3a..ddb88dd69c 100644
--- a/web/packages/new/photos/services/ml/index.ts
+++ b/web/packages/new/photos/services/ml/index.ts
@@ -25,6 +25,7 @@ import {
     updateOrCreateUserEntities,
     type CGroup,
 } from "../user-entity";
+import { deleteUserEntity } from "../user-entity/remote";
 import type { FaceCluster } from "./cluster";
 import { regenerateFaceCrops } from "./crop";
 import { clearMLDB, getFaceIndex, getIndexableAndIndexedCounts } from "./db";
@@ -675,3 +676,13 @@ export const renamePerson = async (name: string, cgroup: CGroup) => {
     );
     return mlSync();
 };
+
+/**
+ * Delete an existing person.
+ *
+ * @param cgroup The existing cgroup underlying the person.
+ */
+export const deletePerson = async ({ id }: CGroup) => {
+    await deleteUserEntity(id);
+    return mlSync();
+};