From 1731aeb87dceefac6d838972ed9743cfdcfeae11 Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Fri, 16 Aug 2024 15:13:01 +0530
Subject: [PATCH] [mob] Fix state refresh when model is being downloaded

---
 .../lib/ui/settings/machine_learning_settings_page.dart  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mobile/lib/ui/settings/machine_learning_settings_page.dart b/mobile/lib/ui/settings/machine_learning_settings_page.dart
index f75fd00b36..910f4da683 100644
--- a/mobile/lib/ui/settings/machine_learning_settings_page.dart
+++ b/mobile/lib/ui/settings/machine_learning_settings_page.dart
@@ -35,12 +35,20 @@ class MachineLearningSettingsPage extends StatefulWidget {
 class _MachineLearningSettingsPageState
     extends State<MachineLearningSettingsPage> {
   final EnteWakeLock _wakeLock = EnteWakeLock();
+  Timer? _timer;
 
   @override
   void initState() {
     super.initState();
     _wakeLock.enable();
     MachineLearningController.instance.forceOverrideML(turnOn: true);
+    if (!MLService.instance.areModelsDownloaded) {
+      _timer = Timer.periodic(const Duration(seconds: 10), (timer) {
+        if (mounted) {
+          setState(() {});
+        }
+      });
+    }
   }
 
   @override
@@ -48,6 +56,7 @@ class _MachineLearningSettingsPageState
     super.dispose();
     _wakeLock.disable();
     MachineLearningController.instance.forceOverrideML(turnOn: false);
+    _timer?.cancel();
   }
 
   @override