Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Fix a bug with custom role
Browse files
state.py
CHANGED
|
@@ -68,19 +68,11 @@ def init_session_state():
|
|
| 68 |
# ---------------------------------------------------------------------------
|
| 69 |
|
| 70 |
def get_display_name(speaker, fileName):
|
| 71 |
-
"""Return the
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
3. Original raw label — e.g. "SPEAKER_00"
|
| 75 |
"""
|
| 76 |
-
|
| 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"):
|