Fix a bug with state change that can make index out of range

state.py

@@ -121,13 +121,21 @@ def updateCategoryOptions(token_display_map=None):
     if st.session_state.resetResult:
         return
     display_to_raw = token_display_map or {}
+
+    # Guard: ensure categorySelect has exactly one slot per category.
+    # Rapid interactions (e.g. addCategory firing mid-callback) can leave
+    # the two lists temporarily out of sync.
+    while len(st.session_state.categorySelect) < len(st.session_state.categories):
+        st.session_state.categorySelect.append([])
+    while len(st.session_state.categorySelect) > len(st.session_state.categories):
+        st.session_state.categorySelect.pop()
+
     for i, category in enumerate(st.session_state.categories):
         ms_key = f"multiselect_{category}"
         display_vals = list(st.session_state.get(ms_key, []))
-        # Translate display labels back to raw tokens for the data model
         raw_vals = [display_to_raw.get(t, t) for t in display_vals]
         st.session_state.categorySelect[i] = raw_vals
-        # Keep widget key as display labels (do not overwrite with raw)
+
     # Recompute unusedSpeakers for all files
     all_assigned_tokens = {
         token
@@ -137,12 +145,15 @@ def updateCategoryOptions(token_display_map=None):
     for fname, result in st.session_state.results.items():
         if len(result) != 2:
             continue
-        annotation, _ = result
-        unused = [
-            sp for sp in annotation.labels()
-            if f"{fname}: {sp}" not in all_assigned_tokens
-        ]
-        st.session_state.unusedSpeakers[fname] = unused
+        try:
+            annotation, _ = result
+            unused = [
+                sp for sp in annotation.labels()
+                if f"{fname}: {sp}" not in all_assigned_tokens
+            ]
+            st.session_state.unusedSpeakers[fname] = unused
+        except Exception:
+            pass
 
 
 # ---------------------------------------------------------------------------