close drawer on job start

Author: hotzenklotz

frontend/javascripts/viewer/view/ai_jobs/run_ai_model/ai_image_segmentation_job_context.tsx

@@ -11,9 +11,11 @@ import Toast from "libs/toast";
 import { computeArrayFromBoundingBox } from "libs/utils";
 import type React from "react";
 import { createContext, useCallback, useContext, useEffect, useState } from "react";
+import { useDispatch } from "react-redux";
 import { APIJobType, type AiModel } from "types/api_types";
 import { ControlModeEnum } from "viewer/constants";
 import { getColorLayers } from "viewer/model/accessors/dataset_accessor";
+import { setAIJobModalStateAction } from "viewer/model/actions/ui_actions";
 import type { UserBoundingBox } from "viewer/store";
 
 interface RunAiModelJobContextType {
@@ -41,6 +43,8 @@ export const RunAiModelJobContextProvider: React.FC<{ children: React.ReactNode
   const [newDatasetName, setNewDatasetName] = useState("");
   const [selectedLayerName, setSelectedLayerName] = useState<string | null>(null);
 
+  const dispatch = useDispatch();
+
   const dataset = useWkSelector((state) => state.dataset);
   const annotationId = useWkSelector((state) => state.annotation.annotationId);
   const isViewMode = useWkSelector(
@@ -76,8 +80,8 @@ export const RunAiModelJobContextProvider: React.FC<{ children: React.ReactNode
     const maybeAnnotationId = isViewMode ? {} : { annotationId };
 
     try {
-      if ("isOwnedByUsersOrganization" in selectedModel) {
-        // Custom model
+      if ("traininJob" in selectedModel) {
+        // Custom models
         const commonInferenceArgs = {
           ...maybeAnnotationId,
           aiModelId: selectedModel.id as string,
@@ -97,7 +101,7 @@ export const RunAiModelJobContextProvider: React.FC<{ children: React.ReactNode
           await runNeuronModelInferenceWithAiModelJob(commonInferenceArgs);
         }
       } else {
-        // Pre-trained model
+        // Pre-trained models
         switch (selectedJobType) {
           case APIJobType.INFER_NEURONS:
             await startNeuronInferralJob(
@@ -124,6 +128,7 @@ export const RunAiModelJobContextProvider: React.FC<{ children: React.ReactNode
         }
       }
       Toast.success("Analysis started successfully!");
+      dispatch(setAIJobModalStateAction("invisible"));
     } catch (error) {
       console.error(error);
       Toast.error("Failed to start analysis.");

frontend/javascripts/viewer/view/ai_jobs/run_ai_model/ai_model_selector.tsx

@@ -171,7 +171,9 @@ export const AiModelSelector: React.FC = () => {
                 title={
                   <Space>
                     <Text strong>{item.name}</Text>
-                    <Tag>CUSTOM</Tag>
+                    <Tag>
+                      {item.category === APIAiModelCategory.EM_NEURONS ? "NEURONS" : "INSTANCES"}
+                    </Tag>
                   </Space>
                 }
                 description={item.comment}