[mod,fix] passes default category kind if no selected category

piximi · Jun 27, 2024 · aba95b9 · aba95b9
1 parent 2eb6af5
commit aba95b9
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/hooks/useAnnotationState/useAnnotationState.ts b/src/hooks/useAnnotationState/useAnnotationState.ts
@@ -39,14 +39,8 @@ export const useAnnotationState = (annotationTool: AnnotationTool) => {
 
   const onAnnotated = useMemo(() => {
     const func = () => {
-      dispatch(
-        annotatorSlice.actions.setAnnotationState({
-          annotationState: AnnotationState.Annotated,
-          kind: selectedCategory?.kind,
-          annotationTool,
-        })
-      );
       if (selectionMode !== AnnotationMode.New) return;
+
       if (!selectedCategory) {
         if (!defaultSelectedCategory) return;
         dispatch(
@@ -66,6 +60,14 @@ export const useAnnotationState = (annotationTool: AnnotationTool) => {
           activeImageId!
         );
       }
+
+      dispatch(
+        annotatorSlice.actions.setAnnotationState({
+          annotationState: AnnotationState.Annotated,
+          kind: selectedCategory?.kind ?? defaultSelectedCategory?.kind,
+          annotationTool,
+        })
+      );
     };
     return func;
   }, [