From e4dcdeb03084835c2e00c7c1af6bcbb0295c2ff1 Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Tue, 14 May 2024 16:42:13 +0530
Subject: [PATCH] [mob] Minor fix to reflect correct status for indexing

---
 mobile/lib/events/embedding_updated_event.dart                | 2 ++
 .../semantic_search/semantic_search_service.dart              | 2 ++
 mobile/lib/ui/settings/machine_learning_settings_page.dart    | 4 ++--
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/mobile/lib/events/embedding_updated_event.dart b/mobile/lib/events/embedding_updated_event.dart
index 9021b8b50..736b85c17 100644
--- a/mobile/lib/events/embedding_updated_event.dart
+++ b/mobile/lib/events/embedding_updated_event.dart
@@ -1,3 +1,5 @@
 import "package:photos/events/event.dart";
 
 class EmbeddingUpdatedEvent extends Event {}
+
+class EmbeddingCacheUpdatedEvent extends Event {}
diff --git a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
index 18c77b43b..99aa3a011 100644
--- a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
+++ b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
@@ -190,6 +190,7 @@ class SemanticSearchService {
     _logger.info(
       "Loading ${_cachedEmbeddings.length} took: ${(endTime.millisecondsSinceEpoch - startTime.millisecondsSinceEpoch)}ms",
     );
+    Bus.instance.fire(EmbeddingCacheUpdatedEvent());
     _logger.info("Cached embeddings: " + _cachedEmbeddings.length.toString());
   }
 
@@ -227,6 +228,7 @@ class SemanticSearchService {
         .getOwnedFileIDs(Configuration.instance.getUserID()!);
     final embeddedFileIDs =
         await EmbeddingsDB.instance.getFileIDs(_currentModel);
+
     uploadedFileIDs.removeWhere(
       (id) => embeddedFileIDs.contains(id),
     );
diff --git a/mobile/lib/ui/settings/machine_learning_settings_page.dart b/mobile/lib/ui/settings/machine_learning_settings_page.dart
index 3306ea36f..a0b72ae09 100644
--- a/mobile/lib/ui/settings/machine_learning_settings_page.dart
+++ b/mobile/lib/ui/settings/machine_learning_settings_page.dart
@@ -228,13 +228,13 @@ class MagicSearchIndexStatsWidget extends StatefulWidget {
 class _MagicSearchIndexStatsWidgetState
     extends State<MagicSearchIndexStatsWidget> {
   IndexStatus? _status;
-  late StreamSubscription<EmbeddingUpdatedEvent> _eventSubscription;
+  late StreamSubscription<EmbeddingCacheUpdatedEvent> _eventSubscription;
 
   @override
   void initState() {
     super.initState();
     _eventSubscription =
-        Bus.instance.on<EmbeddingUpdatedEvent>().listen((event) {
+        Bus.instance.on<EmbeddingCacheUpdatedEvent>().listen((event) {
       _fetchIndexStatus();
     });
     _fetchIndexStatus();