From f6eeb6abcec4fb61b930fe520b3f9848ce30c903 Mon Sep 17 00:00:00 2001 From: Manav Rathi Date: Fri, 15 Nov 2024 14:04:10 +0530 Subject: [PATCH] Apply rejections to suggestion base clusters --- web/packages/new/photos/services/ml/people.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/web/packages/new/photos/services/ml/people.ts b/web/packages/new/photos/services/ml/people.ts index 349336336d..a531f9d09a 100644 --- 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 => { 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), );