add configuration setting for creating pulse annotations

Author: BryonLewis

bats_ai/core/admin/configuration.py

@@ -11,6 +11,7 @@ class ConfigurationAdmin(admin.ModelAdmin):
         "display_pulse_annotations",
         "display_sequence_annotations",
         "run_inference_on_upload",
+        "create_pulse_annotations_from_batbot",
         "spectrogram_x_stretch",
         "spectrogram_view",
     ]

bats_ai/core/models/configuration.py

@@ -22,6 +22,7 @@ class AvailableColorScheme(models.TextChoices):
     display_pulse_annotations = models.BooleanField(default=True)
     display_sequence_annotations = models.BooleanField(default=True)
     run_inference_on_upload = models.BooleanField(default=True)
+    create_pulse_annotations_from_batbot = models.BooleanField(default=False)
     spectrogram_x_stretch = models.DecimalField(default=2.5, max_digits=3, decimal_places=2)
     spectrogram_view = models.CharField(
         max_length=12, choices=SpectrogramViewMode, default=SpectrogramViewMode.COMPRESSED

bats_ai/core/views/configuration.py

@@ -26,6 +26,7 @@ class ConfigurationSchema(Schema):
     display_sequence_annotations: bool
     is_admin: bool | None = None
     run_inference_on_upload: bool
+    create_pulse_annotations_from_batbot: bool
     spectrogram_x_stretch: float
     spectrogram_view: Configuration.SpectrogramViewMode
     default_color_scheme: Configuration.AvailableColorScheme
@@ -44,6 +45,7 @@ def get_configuration(request):
         display_pulse_annotations=config.display_pulse_annotations,
         display_sequence_annotations=config.display_sequence_annotations,
         run_inference_on_upload=config.run_inference_on_upload,
+        create_pulse_annotations_from_batbot=config.create_pulse_annotations_from_batbot,
         spectrogram_x_stretch=config.spectrogram_x_stretch,
         spectrogram_view=config.spectrogram_view,
         default_color_scheme=config.default_color_scheme,
@@ -62,7 +64,7 @@ def update_configuration(request, payload: ConfigurationSchema):
     config = Configuration.objects.first()
     if not config:
         return JsonResponse({"error": "No configuration found"}, status=404)
-    for attr, value in payload.dict().items():
+    for attr, value in payload.dict(exclude={"is_admin"}).items():
         setattr(config, attr, value)
     config.save()
     return ConfigurationSchema.from_orm(config)

client/src/api/api.ts

@@ -653,6 +653,7 @@ export interface ConfigurationSettings {
   display_pulse_annotations: boolean;
   display_sequence_annotations: boolean;
   run_inference_on_upload: boolean;
+  create_pulse_annotations_from_batbot: boolean;
   spectrogram_x_stretch: number;
   spectrogram_view: "compressed" | "uncompressed";
   is_admin?: boolean;

client/src/use/useState.ts

@@ -70,6 +70,7 @@ const configuration: Ref<Configuration> = ref({
   spectrogram_view: "compressed",
   spectrogram_x_stretch: 2.5,
   run_inference_on_upload: true,
+  create_pulse_annotations_from_batbot: false,
   default_color_scheme: "inferno",
   default_spectrogram_background_color: "rgb(0, 0, 0)",
   is_admin: false,

client/src/views/Admin.vue

@@ -28,6 +28,8 @@ export default defineComponent({
       displaySequenceAnnotations:
         configuration.value.display_sequence_annotations,
       runInferenceOnUpload: configuration.value.run_inference_on_upload,
+      createPulseAnnotationsFromBatbot:
+        configuration.value.create_pulse_annotations_from_batbot,
       spectrogramXStretch: configuration.value.spectrogram_x_stretch,
       spectrogramView: configuration.value.spectrogram_view,
       defaultColorScheme: configuration.value.default_color_scheme,
@@ -54,6 +56,8 @@ export default defineComponent({
         configuration.value.display_sequence_annotations;
       settings.runInferenceOnUpload =
         configuration.value.run_inference_on_upload;
+      settings.createPulseAnnotationsFromBatbot =
+        configuration.value.create_pulse_annotations_from_batbot;
       settings.spectrogramXStretch = configuration.value.spectrogram_x_stretch;
       settings.defaultColorScheme = configuration.value.default_color_scheme;
       settings.defaultBackgroundColor =
@@ -74,6 +78,8 @@ export default defineComponent({
         display_pulse_annotations: settings.displayPulseAnnotations,
         display_sequence_annotations: settings.displaySequenceAnnotations,
         run_inference_on_upload: settings.runInferenceOnUpload,
+        create_pulse_annotations_from_batbot:
+          settings.createPulseAnnotationsFromBatbot,
         spectrogram_x_stretch: settings.spectrogramXStretch,
         default_color_scheme: settings.defaultColorScheme,
         default_spectrogram_background_color: settings.defaultBackgroundColor,
@@ -148,6 +154,13 @@ export default defineComponent({
               label="Run Inference on Upload"
             />
           </v-row>
+          <v-row>
+            <v-switch
+              v-model="settings.createPulseAnnotationsFromBatbot"
+              :color="settings.createPulseAnnotationsFromBatbot ? 'primary' : ''"
+              label="Create pulse annotations from BatBot on import"
+            />
+          </v-row>
           <v-row>
             <v-col cols="3">
               <div class="text-caption">Stretch compressed spectrogram</div>