Create a docker image when pushing the pro_version branch

Author: BenCretois

.github/workflows/docker.yml

@@ -2,7 +2,9 @@ name: Create and publish a Docker image
 
 on:
   push:
-    branches: main
+    branches: 
+      - main
+      - pro_version
 
 env:
   REGISTRY: ghcr.io

src/dashboard.py

@@ -7,18 +7,18 @@
 
 import streamlit as st
 
-from ui_utils import setup_page_config
-from session_manager import initialize_pro_session, get_or_load_pro_clip
-from selection_handlers import get_pro_user_selections
-from ui_components import (
+from ui.ui_utils import setup_page_config
+from session.session_manager import initialize_pro_session, get_or_load_pro_clip
+from handlers.selection_handlers import get_pro_user_selections
+from ui.ui_components import (
     render_pro_page_header,
     render_pro_help_section,
     render_pro_clip_section,
     render_pro_empty_validation_placeholder,
     render_pro_all_validated_placeholder,
 )
-from validation_handlers import render_pro_validation_form
-from queries import get_top_species_for_database
+from handlers.validation_handlers import render_pro_validation_form
+from database.queries import get_top_species_for_database
 
 
 def main():

src/database/__init__.py

@@ -45,7 +45,7 @@ def get_top_species_for_database():
     return [row[0] for row in conn.execute(query).fetchall()]
 
 
-@st.cache_data(ttl=600, show_spinner="Loading assigned clips...")  # Cache for 10 minutes
+@st.cache_data(ttl=1800, show_spinner="Loading assigned clips...")  # Cache for 10 minutes
 def get_assigned_clips_for_user(user_id):
     """
     Get all assigned clips for a specific userID.

src/queries.py src/database/queries.pysrc/queries.py renamed to src/database/queries.py

@@ -2,8 +2,8 @@
 
 import streamlit as st
 from config import LANGUAGE_MAPPING
-from queries import check_user_has_annotations
-from ui_utils import render_sidebar_logo
+from database.queries import check_user_has_annotations
+from ui.ui_utils import render_sidebar_logo
 
 
 def render_pro_authentication():

src/handlers/__init__.py

@@ -4,7 +4,7 @@
 import pandas as pd
 import streamlit as st
 
-from queries import get_remaining_pro_clips_count, get_top_species_for_database
+from database.queries import get_remaining_pro_clips_count, get_top_species_for_database
 from utils import get_species_display_names, load_species_translations
 
 
@@ -152,16 +152,21 @@ def _handle_pro_validation_submission(result, selections, selected_species, othe
 
     all_identified_species = selected_species + additional_species
 
+    # Parse arrays from parquet to ensure they're proper lists
+    birdnet_species = _parse_array_string(result.get("species_array", []))
+    birdnet_confidences = _parse_array_string(result.get("confidence_array", []))
+    birdnet_uncertainties = _parse_array_string(result.get("uncertainty_array", []))
+    
     # Prepare validation data
     validation_data = {
         "filename": result["filename"],
         "userID": selections["user_id"],
         "deployment_id": result.get("deployment_id", ""),
-        "birdnet_species_detected": result.get("species_array", []),
-        "birdnet_confidences": result.get("confidence_array", []),
-        "birdnet_uncertainties": result.get("uncertainty_array", []),  # Array of uncertainties
+        "birdnet_species_detected": birdnet_species,
+        "birdnet_confidences": birdnet_confidences,
+        "birdnet_uncertainties": birdnet_uncertainties,
         "start_time": result["start_time"],
-        "identified_species": all_identified_species,  # List of species user confirmed/added
+        "identified_species": all_identified_species,
         "species_count": len(all_identified_species),  # Number of species identified
         "user_confidence": user_confidence,
         "user_notes": user_notes,

src/selection_handlers.py src/handlers/selection_handlers.pysrc/selection_handlers.py renamed to src/handlers/selection_handlers.py

@@ -1,8 +1,8 @@
 """Expert Mode Session Management."""
 
 import streamlit as st
-from queries import get_random_assigned_clip
-from session_utils import init_base_session, init_state_vars, check_params_changed
+from database.queries import get_random_assigned_clip
+from session.session_utils import init_base_session, init_state_vars, check_params_changed
 
 
 def initialize_pro_session():
@@ -25,7 +25,7 @@ def clear_pro_clip_state():
     st.session_state.expert_validated_clips_session = set()
     st.session_state.expert_remaining_count = None
 
-    from queries import get_validated_pro_clips, get_remaining_pro_clips_count
+    from database.queries import get_validated_pro_clips, get_remaining_pro_clips_count
     get_validated_pro_clips.clear()
     get_remaining_pro_clips_count.clear()
 
@@ -46,7 +46,7 @@ def get_or_load_pro_clip(selections):
     if st.session_state.expert_current_clip is None:
         # Initialize remaining count on first load
         if st.session_state.expert_remaining_count is None:
-            from queries import get_remaining_pro_clips_count
+            from database.queries import get_remaining_pro_clips_count
             st.session_state.expert_remaining_count = get_remaining_pro_clips_count(selections["user_id"])
 
         st.session_state.expert_current_clip = get_random_assigned_clip(