add role column in rename speaker tab

ui.py

@@ -220,11 +220,20 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
         if entry["name"] not in st.session_state.inline_rename_history:
             st.session_state.inline_rename_history.append(entry["name"])
 
+    def _get_role(sp):
+        """Return the role name assigned to this speaker, or empty string."""
+        token = f"{currFile}: {sp}"
+        for i, tokens in enumerate(st.session_state.categorySelect):
+            if token in tokens:
+                return st.session_state.categories[i]
+        return ""
+
     # Header
-    header_cols = st.columns([3, 3, 1])
+    header_cols = st.columns([3, 2, 2, 1])
     header_cols[0].markdown("**Speaker**")
-    header_cols[1].markdown("**Audio Sample**")
-    header_cols[2].markdown("** **", unsafe_allow_html=True)
+    header_cols[1].markdown("**Role**")
+    header_cols[2].markdown("**Audio Sample**")
+    header_cols[3].markdown("** **", unsafe_allow_html=True)
     st.markdown("<hr style='margin-top:2px; margin-bottom:4px;'>", unsafe_allow_html=True)
 
     if not has_samples:
@@ -235,7 +244,7 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
         edit_key     = (currFile, sp)
         is_editing   = st.session_state.inline_rename_active.get(edit_key, False)
 
-        row_cols = st.columns([3, 3, 1])
+        row_cols = st.columns([3, 2, 2, 1])
 
         # --- Speaker cell ---
         if is_editing:
@@ -252,21 +261,6 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
             _NEW_OPTION = "＋ Enter a new name"
 
             with row_cols[0]:
-                # Wrap in a form so pressing Enter submits (confirms) the rename.
-                # Change the default "Press Enter to submit form" helper text via CSS.
-                st.markdown(
-                    "<style>"
-                    # target the small helper text Streamlit injects into every form
-                    "div[data-testid='stForm'] small {"
-                    "  visibility: hidden;"  # hide original text …
-                    "}"
-                    "div[data-testid='stForm'] small::after {"
-                    "  content: 'Press Enter to apply';"
-                    "  visibility: visible;"
-                    "}"
-                    "</style>",
-                    unsafe_allow_html=True,
-                )
                 # The selectbox MUST live outside the form: inside a form,
                 # widget state is only committed on submit, so choosing
                 # _NEW_OPTION would never make the text input appear until
@@ -298,6 +292,25 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
                     if not show_input:
                         new_name = chosen
 
+                # Hide Streamlit's built-in "Press Enter to submit form" hint
+                # and replace it. Targets both <small> (older Streamlit) and
+                # <p> (newer Streamlit). font-size:0 collapses the original;
+                # ::after injects the replacement.
+                st.markdown(
+                    "<style>"
+                    "div[data-testid='stForm'] small,"
+                    "div[data-testid='stForm'] p:last-of-type {"
+                    "  font-size: 0 !important;"
+                    "}"
+                    "div[data-testid='stForm'] small::after,"
+                    "div[data-testid='stForm'] p:last-of-type::after {"
+                    "  content: 'Press Enter to apply';"
+                    "  font-size: 0.8rem;"
+                    "  color: gray;"
+                    "}"
+                    "</style>",
+                    unsafe_allow_html=True,
+                )
                 with st.form(key=form_key, border=False):
                     if not history:
                         # No history — plain text input is the only field
@@ -318,6 +331,14 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
                             placeholder=f"Rename {sp}…",
                         )
 
+                    # A disabled first submit button suppresses Streamlit's
+                    # "Press Enter to submit form" hint (built-in behaviour)
+                    # while still allowing Enter to submit via the real buttons.
+                    st.form_submit_button(
+                        "", disabled=True,
+                        help=None,
+                        label_visibility="collapsed",
+                    )
                     _, btn_col1, btn_col2, btn_col3 = st.columns([4, 1, 1, 1])
                     confirmed_submit = btn_col1.form_submit_button("✓", help="Confirm rename (or press Enter)")
                     cancel_clicked   = btn_col2.form_submit_button("✕", help="Cancel")
@@ -354,19 +375,22 @@ def render_speaker_samples_tab(speakerNames, raw_to_display, currFile):
                     st.session_state.inline_rename_active[edit_key] = True
                     st.rerun()
 
+        # --- Role cell ---
+        row_cols[1].write(_get_role(sp))
+
         # --- Audio sample cell ---
         if sp in file_samples:
-            row_cols[1].audio(file_samples[sp], format="audio/wav")
+            row_cols[2].audio(file_samples[sp], format="audio/wav")
             sp_segs = st.session_state.speakerSegments.get(currFile, {}).get(sp, [])
             if has_waveform and sp_segs:
-                if row_cols[2].button(
+                if row_cols[3].button(
                     "↺", key=f"sample_randomize_{currFile}_{sp}",
                     help="Try a different audio sample for this speaker",
                 ):
                     randomize_speaker_clip(currFile, sp)
                     st.rerun()
         else:
-            row_cols[1].write("—")
+            row_cols[2].write("—")
 
 
 # ---------------------------------------------------------------------------