duongthienz commited on
Commit
70f7192
·
verified ·
1 Parent(s): 21f8bdd

Fix a bug with custom role

Browse files
Files changed (1) hide show
  1. state.py +4 -12
state.py CHANGED
@@ -68,19 +68,11 @@ def init_session_state():
68
  # ---------------------------------------------------------------------------
69
 
70
  def get_display_name(speaker, fileName):
71
- """Return the best display name for a speaker, in priority order:
72
- 1. Explicit rename (speakerRenames) set by rename sidebar or rename tab
73
- 2. Role assignment (categorySelect) e.g. "Instructor" or "Students"
74
- 3. Original raw label — e.g. "SPEAKER_00"
75
  """
76
- explicit = st.session_state.speakerRenames.get(fileName, {}).get(speaker)
77
- if explicit:
78
- return explicit
79
- token = f"{fileName}: {speaker}"
80
- for i, tokens in enumerate(st.session_state.categorySelect):
81
- if token in tokens:
82
- return st.session_state.categories[i]
83
- return speaker
84
 
85
 
86
  def apply_speaker_renames_to_df(df, fileName, column="task"):
 
68
  # ---------------------------------------------------------------------------
69
 
70
  def get_display_name(speaker, fileName):
71
+ """Return the user-assigned display name for a speaker, or the original label.
72
+ Role assignments (categorySelect) are intentionally excluded roles are for
73
+ grouping in charts, not for renaming speakers.
 
74
  """
75
+ return st.session_state.speakerRenames.get(fileName, {}).get(speaker, speaker)
 
 
 
 
 
 
 
76
 
77
 
78
  def apply_speaker_renames_to_df(df, fileName, column="task"):