Update ui.py

ui.py

@@ -114,14 +114,31 @@ def render_rename_sidebar(currFile, speakerNames, all_speaker_tokens):
     st.sidebar.divider()
 
     def _on_grename_change(idx):
-        # Write the widget's new value into the data model first
-        st.session_state.globalRenames[idx]["speakers"] = list(
-            st.session_state[_global_rename_key(idx)]
-        )
-        # Remove the changed entry's widget key so it re-seeds cleanly on
-        # the next render (avoids Streamlit "cannot modify instantiated widget" errors)
+        # Determine which tokens were just added (not which ones are currently held)
+        # so exclusivity enforcement only removes newly claimed tokens from other entries.
+        prev_speakers = set(st.session_state.globalRenames[idx]["speakers"])
+        new_speakers  = list(st.session_state[_global_rename_key(idx)])
+        newly_added   = set(new_speakers) - prev_speakers
+
+        # Write the widget's new value into the data model
+        st.session_state.globalRenames[idx]["speakers"] = new_speakers
+        # Pop widget key so it re-seeds cleanly on the next render pass
         st.session_state.pop(_global_rename_key(idx), None)
-        enforceGlobalRenameExclusivity(idx)
+
+        if newly_added:
+            # Record these tokens in the graveyard so applyGlobalRenames can
+            # revert them if they are later removed from all globalRenames entries.
+            st.session_state.globally_tracked_tokens.update(newly_added)
+            # Only enforce exclusivity for tokens that were just added
+            for other_idx, entry in enumerate(st.session_state.globalRenames):
+                if other_idx == idx:
+                    continue
+                cleaned = [t for t in entry["speakers"] if t not in newly_added]
+                if cleaned != entry["speakers"]:
+                    entry["speakers"] = cleaned
+                    st.session_state.pop(_global_rename_key(other_idx), None)
+
+        applyGlobalRenames()
 
     # Build claimed set from the data model (entry["speakers"]) as the canonical
     # source of truth.  Widget keys may be absent after a mutation (we pop them