chore: moved all session state init to relevant modules

src/classifier/classifier_image.py

@@ -12,6 +12,17 @@ from utils.grid_maker import gridder
 from utils.metadata_handler import metadata2md
 from input.input_observation import InputObservation
 
+def init_classifier_session_states() -> None:
+    '''
+    Initialise the session state variables used in classification
+    '''
+    if "classify_whale_done" not in st.session_state:
+        st.session_state.classify_whale_done = {}
+
+    if "whale_prediction1" not in st.session_state:
+        st.session_state.whale_prediction1 = {}
+
+
 def add_classifier_header() -> None:
     """
     Add brief explainer text about cetacean classification to the tab 

src/input/input_handling.py

@@ -370,6 +370,10 @@ def init_input_data_session_states() -> None:
     if "files" not in st.session_state:
         st.session_state.files = {}
 
+    if "public_observations" not in st.session_state:
+        st.session_state.public_observations = {}
+
+
     
 def add_input_UI_elements() -> None:
     '''

src/main.py

@@ -22,12 +22,11 @@ from input.input_handling import dbg_show_observation_hashes
 
 from maps.alps_map import present_alps_map
 from maps.obs_map import present_obs_map
-from utils.st_logs import setup_logging, parse_log_buffer
-from utils.workflow_state import WorkflowFSM, FSM_STATES
-from utils.workflow_ui import refresh_progress_display, init_workflow_viz
+from utils.st_logs import parse_log_buffer, init_logging_session_states
+from utils.workflow_ui import refresh_progress_display, init_workflow_viz, init_workflow_session_states
 from hf_push_observations import push_all_observations
 
-from classifier.classifier_image import cetacean_just_classify, cetacean_show_results_and_review, cetacean_show_results
+from classifier.classifier_image import cetacean_just_classify, cetacean_show_results_and_review, cetacean_show_results, init_classifier_session_states
 from classifier.classifier_hotdog import hotdog_classify
 
 
@@ -55,29 +54,17 @@ g_logger.setLevel(LOG_LEVEL)
 st.set_page_config(layout="wide")
 
 # initialise various session state variables
-if "handler" not in st.session_state:
-    st.session_state['handler'] = setup_logging()
-
-
-if "public_observations" not in st.session_state:
-    st.session_state.public_observations = {}
-
-if "classify_whale_done" not in st.session_state:
-    st.session_state.classify_whale_done = {}
-
-if "whale_prediction1" not in st.session_state:
-    st.session_state.whale_prediction1 = {}
+init_logging_session_states() # logging should be early 
+init_workflow_session_states() 
 
+# TODO: this is obselete, now we have the st_logs functionality.
 if "tab_log" not in st.session_state:
     st.session_state.tab_log = None
     
-if "workflow_fsm" not in st.session_state:
-    # create and init the state machine
-    st.session_state.workflow_fsm = WorkflowFSM(FSM_STATES)
-    
 init_input_data_session_states()
 init_input_container_states()
 init_workflow_viz()
+init_classifier_session_states()
 
     
     

src/utils/st_logs.py

@@ -100,6 +100,16 @@ class StreamlitLogHandler(logging.Handler):
         self.log_area.empty()  # Clear previous logs
         self.buffer.clear()
 
+
+def init_logging_session_states():
+    """
+    Initialise the session state variables for logging.
+    """
+    
+    if "handler" not in st.session_state:
+        st.session_state['handler'] = setup_logging()
+
+
 # Set up logging to capture all info level logs from the root logger
 @st.cache_resource
 def setup_logging(level:int=logging.INFO, buffer_len:int=15) -> StreamlitLogHandler:
@@ -126,6 +136,7 @@ def setup_logging(level:int=logging.INFO, buffer_len:int=15) -> StreamlitLogHand
     #    st.session_state['handler'] = handler
     return handler
 
+
 def parse_log_buffer(log_contents: deque) -> List[dict]:
     """
     Convert log buffer to a list of dictionaries for use with a streamlit datatable.

src/utils/workflow_ui.py

@@ -1,4 +1,14 @@
 import streamlit as st
+from utils.workflow_state import WorkflowFSM, FSM_STATES
+
+def init_workflow_session_states():
+    """
+    Initialise the session state variables for the workflow state machine
+    """
+    
+    if "workflow_fsm" not in st.session_state:
+        # create and init the state machine
+        st.session_state.workflow_fsm = WorkflowFSM(FSM_STATES)
 
 def refresh_progress_display() -> None:
     """