Update ui.py

ui.py

@@ -51,9 +51,8 @@ def render_chart(fig, tab, pdf_path, svg_path, pdf_name, svg_name, pdf_key, svg_
 # Sidebar — categories section
 # ---------------------------------------------------------------------------
 
-def render_categories_sidebar(currFile, categorySelections, all_speakers_display,
-                               raw_to_display):
-    """Render category multiselects, remove buttons, and Add custom role input."""
+def render_categories_sidebar(all_speaker_tokens):
+    """Render cross-file role multiselects using speaker tokens, matching the rename sidebar pattern."""
     st.sidebar.subheader("Add Speaker's Role")
     st.sidebar.markdown(
         "<p style='font-size:0.85rem; color:gray; margin-bottom:2px;'>"
@@ -62,12 +61,18 @@ def render_categories_sidebar(currFile, categorySelections, all_speakers_display
         unsafe_allow_html=True,
     )
     st.sidebar.divider()
+
+    # Tokens already claimed by any role, for exclusivity filtering
+    all_claimed = {
+        token
+        for tokens in st.session_state.categorySelect
+        for token in tokens
+    }
+
     for i, category in enumerate(st.session_state.categories):
-        ms_key       = f"multiselect_{category}"
-        speakerSet   = categorySelections[i]
-        default_disp = [raw_to_display.get(sp, sp) for sp in speakerSet]
+        ms_key = f"multiselect_{category}"
         if ms_key not in st.session_state:
-            st.session_state[ms_key] = default_disp
+            st.session_state[ms_key] = list(st.session_state.categorySelect[i]) if i < len(st.session_state.categorySelect) else []
         label_col, trash_col = st.sidebar.columns([5, 1])
         label_col.markdown(f"**{category}**")
         trash_col.button(
@@ -75,9 +80,14 @@ def render_categories_sidebar(currFile, categorySelections, all_speakers_display
             on_click=removeCategory, args=(i,),
             help=f"Remove {category}",
         )
+        this_role_tokens = set(st.session_state.categorySelect[i]) if i < len(st.session_state.categorySelect) else set()
+        available_tokens = [
+            t for t in all_speaker_tokens
+            if t not in all_claimed or t in this_role_tokens
+        ]
         st.sidebar.multiselect(
-            category, all_speakers_display,
-            key=ms_key, on_change=updateCategoryOptions, args=(currFile,),
+            category, available_tokens,
+            key=ms_key, on_change=updateCategoryOptions,
             label_visibility="collapsed",
         )
     st.sidebar.text_input("Add custom role", key="categoryInput", on_change=addCategory)