Update state.py

state.py

@@ -113,11 +113,16 @@ def updateCategoryOptions():
     """Store tokens ('fname: SPEAKER_##') in the global categorySelect lists."""
     if st.session_state.resetResult:
         return
-    # Rebuild unusedSpeakers for every analyzed file based on current token assignments
+    # Rebuild categorySelect from widget values, then sync widget keys back to
+    # match — this keeps the data model and the displayed selection in lockstep.
     for i, category in enumerate(st.session_state.categories):
-        st.session_state.categorySelect[i] = list(
-            st.session_state[f"multiselect_{category}"]
-        )
+        ms_key = f"multiselect_{category}"
+        new_val = list(st.session_state.get(ms_key, []))
+        st.session_state.categorySelect[i] = new_val
+        # Re-assign (don't pop) so the widget renders the correct selection on
+        # the next run.  Popping causes Streamlit to fall back to [] because no
+        # `default=` is provided, wiping the visible selection.
+        st.session_state[ms_key] = new_val
     # Recompute unusedSpeakers for all files
     all_assigned_tokens = {
         token
@@ -185,7 +190,7 @@ def removeGlobalRename(index):
         _write_rename(token, "")
     st.session_state.pop(_global_rename_key(index), None)
     del st.session_state.globalRenames[index]
-    # Shift remaining widget keys down
+    # Shift remaining widget keys down and sync them to their data model
     for i in range(index, len(st.session_state.globalRenames)):
         st.session_state[_global_rename_key(i)] = list(
             st.session_state.globalRenames[i]["speakers"]
@@ -263,7 +268,11 @@ def on_grename_change(idx):
     # Write the new speakers list to the data model
     kept = [t for t in prev if t in new_set]
     entry["speakers"] = kept + [t for t in added]
-    st.session_state.pop(grkey, None)
+    # Sync the widget key to match the data model.
+    # Do NOT pop the key — popping causes Streamlit to re-initialise the widget
+    # to [] on the next render (because no `default=` is passed), erasing the
+    # selection visually even though the data model is correct.
+    st.session_state[grkey] = list(entry["speakers"])
 
     # Enforce exclusivity and write renames for newly added tokens
     for token in added:
@@ -272,7 +281,8 @@ def on_grename_change(idx):
                 continue
             if token in other_entry["speakers"]:
                 other_entry["speakers"].remove(token)
-                st.session_state.pop(_global_rename_key(other_idx), None)
+                # Sync the other entry's widget key too (same reason — don't pop)
+                st.session_state[_global_rename_key(other_idx)] = list(other_entry["speakers"])
                 _write_rename(token, "")
         _write_rename(token, name)