kenmirrors/ente
1
0
Fork 0
mirror of https://github.com/ente-io/ente.git synced 2025-08-08 07:28:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

[mob][photos] Store ML results on remote first

Browse Source
This commit is contained in:
laurenspriem 2024-08-21 18:31:55 +02:00
parent 87a17c0034
commit 5ef4b7160e
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mobile/lib/services/machine_learning/ml_service.dart
View File

@ -400,8 +400,8 @@ class MLService {
DataType.mlData,
);
// Faces results
final List<Face> faces = [];
if (result.facesRan) {
final List<Face> faces = [];
if (result.faces!.isEmpty) {
faces.add(Face.empty(result.fileId));
_logger.info("no face detected, storing empty for ${result.fileId}");
@ -418,7 +418,6 @@ class MLService {
}
_logger.info("storing ${faces.length} faces for ${result.fileId}");
}
await MLDataDB.instance.bulkInsertFaces(faces);
dataEntity.putFace(
RemoteFaceEmbedding(
faces,
@ -431,9 +430,6 @@ class MLService {
}
// Clip results
if (result.clipRan) {
await SemanticSearchService.storeClipImageResult(
result.clip!,
);
dataEntity.putClip(
RemoteClipEmbedding(
result.clip!.embedding,
@ -442,11 +438,20 @@ class MLService {
),
);
}
// Storing all results on remote
// Storing results on remote
await FileDataService.instance.putFileData(
instruction.file,
dataEntity,
);
_logger.info("Results for file ${result.fileId} stored on remote");
// Storing results locally
if (result.facesRan) await MLDataDB.instance.bulkInsertFaces(faces);
if (result.clipRan) {
await SemanticSearchService.storeClipImageResult(
result.clip!,
);
}
_logger.info("Results for file ${result.fileId} stored locally");
return actuallyRanML;
} catch (e, s) {
bool acceptedIssue = false;
@ -459,7 +464,8 @@ class MLService {
);
acceptedIssue = true;
}
if (errorString.contains('InvalidImageFormatException: Error decoding image')) {
if (errorString
.contains('InvalidImageFormatException: Error decoding image')) {
_logger.severe(
'InvalidImageFormatException while processing image with ID ${instruction.file.uploadedFileID}, storing empty results so indexing does not get stuck',
e,