From ccd9826325c427f8cd31b677318abbf8f1d9be5e Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Tue, 30 Apr 2024 12:39:12 -0600
Subject: [PATCH] Handle sub labels

---
 web/src/hooks/use-camera-activity.ts | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/web/src/hooks/use-camera-activity.ts b/web/src/hooks/use-camera-activity.ts
index ef065dd81..c6f0f06ec 100644
--- a/web/src/hooks/use-camera-activity.ts
+++ b/web/src/hooks/use-camera-activity.ts
@@ -3,7 +3,7 @@ import {
   useInitialCameraState,
   useMotionActivity,
 } from "@/api/ws";
-import { CameraConfig } from "@/types/frigateConfig";
+import { ATTRIBUTE_LABELS, CameraConfig } from "@/types/frigateConfig";
 import { MotionData, ReviewSegment } from "@/types/review";
 import { useEffect, useMemo, useState } from "react";
 import { useTimelineUtils } from "./use-timeline-utils";
@@ -77,8 +77,20 @@ export function useCameraActivity(
         }
       } else {
         const newObjects = [...objects];
-        newObjects[updatedEventIndex].label =
-          updatedEvent.after.sub_label ?? updatedEvent.after.label;
+
+        let label = updatedEvent.after.label;
+
+        if (updatedEvent.after.sub_label) {
+          const sub_label = updatedEvent.after.sub_label[0];
+
+          if (ATTRIBUTE_LABELS.includes(sub_label)) {
+            label = sub_label;
+          } else {
+            label = `${label}-verified`;
+          }
+        }
+
+        newObjects[updatedEventIndex].label = label;
         newObjects[updatedEventIndex].stationary =
           updatedEvent.after.stationary;
         setObjects(newObjects);