From 6aaa7b15194aeac84ad28de13b170bc4676b7d13 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Thu, 7 Nov 2024 14:15:25 -0600
Subject: [PATCH] Ensure the regenerate button shows when genai is only enabled
 at the camera level

---
 frigate/api/event.py                                     | 4 +++-
 frigate/genai/__init__.py                                | 9 ++++-----
 web/src/components/overlay/detail/SearchDetailDialog.tsx | 2 +-
 web/src/types/frigateConfig.ts                           | 7 +++++++
 4 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/frigate/api/event.py b/frigate/api/event.py
index ac414cdde..cf0ac26cc 100644
--- a/frigate/api/event.py
+++ b/frigate/api/event.py
@@ -996,9 +996,11 @@ def regenerate_description(
             status_code=404,
         )
 
+    camera_config = request.app.frigate_config.cameras[event.camera]
+
     if (
         request.app.frigate_config.semantic_search.enabled
-        and request.app.frigate_config.genai.enabled
+        and camera_config.genai.enabled
     ):
         request.app.event_metadata_updater.publish((event.id, params.source))
 
diff --git a/frigate/genai/__init__.py b/frigate/genai/__init__.py
index e2d509383..74fae9fea 100644
--- a/frigate/genai/__init__.py
+++ b/frigate/genai/__init__.py
@@ -54,11 +54,10 @@ class GenAIClient:
 
 def get_genai_client(genai_config: GenAIConfig) -> Optional[GenAIClient]:
     """Get the GenAI client."""
-    if genai_config.enabled:
-        load_providers()
-        provider = PROVIDERS.get(genai_config.provider)
-        if provider:
-            return provider(genai_config)
+    load_providers()
+    provider = PROVIDERS.get(genai_config.provider)
+    if provider:
+        return provider(genai_config)
     return None
 
 
diff --git a/web/src/components/overlay/detail/SearchDetailDialog.tsx b/web/src/components/overlay/detail/SearchDetailDialog.tsx
index f158df329..f56074a52 100644
--- a/web/src/components/overlay/detail/SearchDetailDialog.tsx
+++ b/web/src/components/overlay/detail/SearchDetailDialog.tsx
@@ -477,7 +477,7 @@ function ObjectDetailsTab({
           onChange={(e) => setDesc(e.target.value)}
         />
         <div className="flex w-full flex-row justify-end gap-2">
-          {config?.genai.enabled && (
+          {config?.cameras[search.camera].genai.enabled && (
             <div className="flex items-center">
               <Button
                 className="rounded-r-none border-r-0"
diff --git a/web/src/types/frigateConfig.ts b/web/src/types/frigateConfig.ts
index 1d54b4e7e..5c5971fc0 100644
--- a/web/src/types/frigateConfig.ts
+++ b/web/src/types/frigateConfig.ts
@@ -77,6 +77,13 @@ export interface CameraConfig {
     cmd: string;
     roles: string[];
   }[];
+  genai: {
+    enabled: string;
+    prompt: string;
+    object_prompts: { [key: string]: string };
+    required_zones: string[];
+    objects: string[];
+  };
   live: {
     height: number;
     quality: number;