UI improvements with admin

.gitignore

@@ -13,3 +13,4 @@ temp*.png
 staticfiles/
 .env
 dev/.env.prod.docker-compose
+**/*.egg-info/**

bats_ai/core/migrations/0014_configuration_run_inference_on_upload_and_more.py

@@ -0,0 +1,31 @@
+# Generated by Django 4.1.13 on 2025-03-26 21:22
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('core', '0013_configuration'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='configuration',
+            name='run_inference_on_upload',
+            field=models.BooleanField(default=True),
+        ),
+        migrations.AddField(
+            model_name='configuration',
+            name='spectrogram_view',
+            field=models.CharField(
+                choices=[('compressed', 'Compressed'), ('uncompressed', 'Uncompressed')],
+                default='compressed',
+                max_length=12,
+            ),
+        ),
+        migrations.AddField(
+            model_name='configuration',
+            name='spectrogram_x_stretch',
+            field=models.DecimalField(decimal_places=2, default=2.5, max_digits=3),
+        ),
+    ]

bats_ai/core/models/configuration.py

@@ -5,8 +5,17 @@
 
 # Define the Configuration model
 class Configuration(models.Model):
+    class SpectrogramViewMode(models.TextChoices):
+        COMPRESSED = 'compressed'
+        UNCOMPRESSED = 'uncompressed'
+
     display_pulse_annotations = models.BooleanField(default=True)
     display_sequence_annotations = models.BooleanField(default=True)
+    run_inference_on_upload = models.BooleanField(default=True)
+    spectrogram_x_stretch = models.DecimalField(default=2.5, max_digits=3, decimal_places=2)
+    spectrogram_view = models.CharField(
+        max_length=12, choices=SpectrogramViewMode.choices, default=SpectrogramViewMode.COMPRESSED
+    )
 
     def save(self, *args, **kwargs):
         # Ensure only one instance of Configuration exists

bats_ai/core/tasks.py

@@ -11,6 +11,7 @@
 from bats_ai.core.models import (
     Annotations,
     CompressedSpectrogram,
+    Configuration,
     Recording,
     RecordingAnnotation,
     Species,
@@ -187,7 +188,9 @@ def recording_compute_spectrogram(recording_id: int):
                 spectrogram_id = spectrogram_id_temp
     if spectrogram_id is not None:
         compressed_spectro = generate_compress_spectrogram(recording_id, spectrogram_id)
-        predict(compressed_spectro.pk)
+        config = Configuration.objects.first()
+        if not config or not config.run_inference_on_upload:
+            predict(compressed_spectro.pk)
 
 
 @shared_task

bats_ai/core/views/configuration.py

@@ -17,6 +17,9 @@ class ConfigurationSchema(Schema):
     display_pulse_annotations: bool
     display_sequence_annotations: bool
     is_admin: bool | None = None
+    run_inference_on_upload: bool
+    spectrogram_x_stretch: float
+    spectrogram_view: Configuration.SpectrogramViewMode
 
 
 # Endpoint to retrieve the configuration status
@@ -28,6 +31,9 @@ def get_configuration(request):
     return ConfigurationSchema(
         display_pulse_annotations=config.display_pulse_annotations,
         display_sequence_annotations=config.display_sequence_annotations,
+        run_inference_on_upload=config.run_inference_on_upload,
+        spectrogram_x_stretch=config.spectrogram_x_stretch,
+        spectrogram_view=config.spectrogram_view,
         is_admin=request.user.is_authenticated and request.user.is_superuser,
     )