TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][photos] keep face crops for clusters as well

Browse Source
This commit is contained in:
laurenspriem
2024-11-23 16:07:11 +05:30
parent d1e82b9261
commit dd8cef3ddf
1 changed files with 27 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
mobile/lib/ui/viewer/search/result/person_face_widget.dart
View File

@@ -88,6 +88,7 @@ class _PersonFaceWidgetState extends State<PersonFaceWidget> {
try {
EnteFile? fileForFaceCrop = widget.file;
String? personAvatarFaceID;
Iterable<String>? allFaces;
if (widget.personId != null) {
final PersonEntity? personEntity =
await PersonService.instance.getPerson(widget.personId!);
@@ -99,30 +100,35 @@ class _PersonFaceWidgetState extends State<PersonFaceWidget> {
if (tryCache != null) return tryCache;
}
if (personAvatarFaceID == null && widget.cannotTrustFile) {
final allFaces =
allFaces =
await MLDataDB.instance.getFaceIDsForPerson(widget.personId!);
final allFileIDs = allFaces.map((e) => getFileIdFromFaceId(e));
final fileIDToCreationTime =
await FilesDB.instance.getFileIDToCreationTime();
// Get the file with the most recent creation time
final recentFileID = allFileIDs.reduce((a, b) {
final aTime = fileIDToCreationTime[a];
final bTime = fileIDToCreationTime[b];
if (aTime == null) {
return b;
}
if (bTime == null) {
return a;
}
return (aTime >= bTime) ? a : b;
});
if (fileForFaceCrop.uploadedFileID != recentFileID) {
fileForFaceCrop =
await FilesDB.instance.getAnyUploadedFile(recentFileID);
if (fileForFaceCrop == null) return null;
}
}
}
} else if (widget.clusterID != null && widget.cannotTrustFile) {
allFaces =
await MLDataDB.instance.getFaceIDsForCluster(widget.clusterID!);
}
if (allFaces != null) {
final allFileIDs = allFaces.map((e) => getFileIdFromFaceId(e));
final fileIDToCreationTime =
await FilesDB.instance.getFileIDToCreationTime();
// Get the file with the most recent creation time
final recentFileID = allFileIDs.reduce((a, b) {
final aTime = fileIDToCreationTime[a];
final bTime = fileIDToCreationTime[b];
if (aTime == null) {
return b;
}
if (bTime == null) {
return a;
}
return (aTime >= bTime) ? a : b;
});
if (fileForFaceCrop.uploadedFileID != recentFileID) {
fileForFaceCrop =
await FilesDB.instance.getAnyUploadedFile(recentFileID);
if (fileForFaceCrop == null) return null;
}
}
final Face? face = await MLDataDB.instance.getCoverFaceForPerson(