From 4e7e506e4d593da8cc88050fd100d99ac03c0fa3 Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Fri, 16 Aug 2024 18:00:05 +0530
Subject: [PATCH] [Fix] Avoid queueing same fileID again

---
 mobile/lib/utils/ml_util.dart | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/mobile/lib/utils/ml_util.dart b/mobile/lib/utils/ml_util.dart
index d50555d38a..ddb885136b 100644
--- a/mobile/lib/utils/ml_util.dart
+++ b/mobile/lib/utils/ml_util.dart
@@ -73,6 +73,7 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
       await MLDataDB.instance.getIndexedFileIds();
   final Map<int, int> clipIndexedFileIDs =
       await MLDataDB.instance.clipIndexedFileWithVersion();
+  final Set<int> queuedFiledIDs = {};
 
   // Get all regular files and all hidden files
   final enteFiles = await SearchService.instance.getAllFiles();
@@ -86,6 +87,11 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
     if (_skipAnalysisEnteFile(enteFile)) {
       continue;
     }
+    if (queuedFiledIDs.contains(enteFile.uploadedFileID)) {
+      continue;
+    }
+    queuedFiledIDs.add(enteFile.uploadedFileID!);
+
     final shouldRunFaces =
         _shouldRunIndexing(enteFile, faceIndexedFileIDs, faceMlVersion);
     final shouldRunClip =
@@ -108,6 +114,10 @@ Future<List<FileMLInstruction>> getFilesForMlIndexing() async {
     if (_skipAnalysisEnteFile(enteFile)) {
       continue;
     }
+    if (queuedFiledIDs.contains(enteFile.uploadedFileID)) {
+      continue;
+    }
+    queuedFiledIDs.add(enteFile.uploadedFileID!);
     final shouldRunFaces =
         _shouldRunIndexing(enteFile, faceIndexedFileIDs, faceMlVersion);
     final shouldRunClip =