From 53c8aa25cb0bbcbcd37c5f7737062b672c038db7 Mon Sep 17 00:00:00 2001
From: leccelecce <24962424+leccelecce@users.noreply.github.com>
Date: Wed, 26 Mar 2025 21:44:15 +0000
Subject: [PATCH] Fix missing restart notification in Classification settings
 (#17397)

* Use different message key to fix missing restart message

* Move addMessage to finally block
---
 .../views/settings/ClassificationSettingsView.tsx    |  4 ++--
 web/src/views/settings/FrigatePlusSettingsView.tsx   | 12 ++++++------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/web/src/views/settings/ClassificationSettingsView.tsx b/web/src/views/settings/ClassificationSettingsView.tsx
index b40fde19f..24c3a9107 100644
--- a/web/src/views/settings/ClassificationSettingsView.tsx
+++ b/web/src/views/settings/ClassificationSettingsView.tsx
@@ -176,8 +176,8 @@ export default function ClassificationSettingsView({
       })
       .finally(() => {
         addMessage(
-          "search_settings",
-          `Restart Required (Classification settings changed)`,
+          "search_settings_restart",
+          `Restart required (Classification settings changed)`,
           undefined,
           "search_settings",
         );
diff --git a/web/src/views/settings/FrigatePlusSettingsView.tsx b/web/src/views/settings/FrigatePlusSettingsView.tsx
index f29ebd604..49426bbe9 100644
--- a/web/src/views/settings/FrigatePlusSettingsView.tsx
+++ b/web/src/views/settings/FrigatePlusSettingsView.tsx
@@ -131,12 +131,6 @@ export default function FrigatePlusSettingsView({
             position: "top-center",
           });
           setChangedValue(false);
-          addMessage(
-            "plus_restart",
-            "Restart required (Frigate+ model changed)",
-            undefined,
-            "plus_restart",
-          );
           updateConfig();
         } else {
           toast.error(
@@ -160,6 +154,12 @@ export default function FrigatePlusSettingsView({
         );
       })
       .finally(() => {
+        addMessage(
+          "plus_restart",
+          "Restart required (Frigate+ model changed)",
+          undefined,
+          "plus_restart",
+        );
         setIsLoading(false);
       });
   }, [updateConfig, addMessage, frigatePlusSettings, t]);