Merge branch 'deployments/ood.antenna.insectai.org' of github.com:RolnickLab/antenna into feat/better-cluster-data

Author: mihow

ami/main/api/views.py

@@ -1188,6 +1188,24 @@ def filter_queryset(self, request, queryset, view):
         return queryset
 
 
+class TaxonUnknownSpeciesFilter(filters.BaseFilterBackend):
+    """
+    Filter taxa that is or is not marked as "Unknown species".
+    """
+
+    query_param = "unknown_species"
+
+    def filter_queryset(self, request: Request, queryset, view):
+        if self.query_param in request.query_params:
+            unknown_species = BooleanField(required=False).clean(request.query_params.get(self.query_param))
+            if unknown_species:
+                queryset = queryset.filter(unknown_species=True)
+            else:
+                queryset = queryset.exclude(unknown_species=True)
+
+        return queryset
+
+
 class TaxonTagFilter(filters.BaseFilterBackend):
     """FilterBackend that allows OR-based filtering of taxa by tag ID."""
 
@@ -1222,6 +1240,7 @@ class TaxonViewSet(DefaultViewSet, ProjectMixin):
         CustomTaxonFilter,
         TaxonCollectionFilter,
         TaxonTaxaListFilter,
+        TaxonUnknownSpeciesFilter,
         TaxonTagFilter,
         TagInverseFilter,
     ]

ui/src/components/filtering/filter-control.tsx

@@ -16,6 +16,7 @@ import { TaxaListFilter } from './filters/taxa-list-filter'
 import { TaxonFilter } from './filters/taxon-filter'
 import { TypeFilter } from './filters/type-filter'
 import { FilterProps } from './filters/types'
+import { UnknownSpeciesFilter } from './filters/unknown-species-filter'
 import { VerificationStatusFilter } from './filters/verification-status-filter'
 import { VerifiedByFilter } from './filters/verified-by-filter'
 
@@ -43,6 +44,7 @@ const ComponentMap: {
   taxa_list_id: TaxaListFilter,
   verified_by_me: VerifiedByFilter,
   verified: VerificationStatusFilter,
+  unknown_species: UnknownSpeciesFilter,
 }
 
 interface FilterControlProps {

ui/src/components/filtering/filters/unknown-species-filter.tsx

@@ -0,0 +1,30 @@
+import { Select } from 'nova-ui-kit'
+import { booleanToString, stringToBoolean } from '../utils'
+import { FilterProps } from './types'
+
+const OPTIONS = [
+  { value: true, label: 'Unknown species' },
+  { value: false, label: 'Known species' },
+]
+
+export const UnknownSpeciesFilter = ({ value: string, onAdd }: FilterProps) => {
+  const value = stringToBoolean(string)
+
+  return (
+    <Select.Root value={booleanToString(value)} onValueChange={onAdd}>
+      <Select.Trigger>
+        <Select.Value placeholder="Select a value" />
+      </Select.Trigger>
+      <Select.Content className="max-h-72">
+        {OPTIONS.map((option) => (
+          <Select.Item
+            key={booleanToString(option.value)}
+            value={booleanToString(option.value)}
+          >
+            {option.label}
+          </Select.Item>
+        ))}
+      </Select.Content>
+    </Select.Root>
+  )
+}

ui/src/pages/species/species.tsx

@@ -47,6 +47,7 @@ export const Species = () => {
           <FilterControl field="event" readonly />
           <FilterControl field="deployment" />
           <FilterControl field="taxon" />
+          <FilterControl field="unknown_species" />
           {taxaLists.length > 0 && (
             <FilterControl data={taxaLists} field="taxa_list_id" />
           )}

ui/src/utils/useFilters.ts

@@ -100,6 +100,10 @@ export const AVAILABLE_FILTERS: {
     label: 'Taxon',
     field: 'taxon',
   },
+  {
+    label: 'Known status',
+    field: 'unknown_species',
+  },
   {
     label: 'Taxa list',
     field: 'taxa_list_id',