Apply rejections to suggestion base clusters

2024-11-15 14:04:10 +05:30
parent c577ccd7e4
commit f6eeb6abce
1 changed files with 6 additions and 1 deletions
--- a/web/packages/new/photos/services/ml/people.ts
+++ b/web/packages/new/photos/services/ml/people.ts
@@ -446,7 +446,12 @@ export const _suggestionsAndChoicesForPerson = async (
 ): Promise<PersonSuggestionsAndChoices> => {
    const startTime = Date.now();

-    const personClusters = person.cgroup.data.assigned;
+    const rejectedFaceIDs = new Set(person.cgroup.data.rejectedFaceIDs);
+    const personClusters = person.cgroup.data.assigned.map((cluster) => ({
+        ...cluster,
+        faces: cluster.faces.filter((id) => !rejectedFaceIDs.has(id)),
+    }));
+
    const rejectedClusterIDs = new Set(
        await savedRejectedClustersForCGroup(person.cgroup.id),
    );