duongthienz commited on
Commit
509858d
·
verified ·
1 Parent(s): 56539a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -237,7 +237,12 @@ try:
237
  speakers_dataFrame = st.session_state.summaries[currFile]["speakers_dataFrame"]
238
  currDF, _ = su.annotationToSimpleDataFrame(currAnnotation)
239
  unusedSpeakers = st.session_state.unusedSpeakers[currFile]
240
- categorySelections = st.session_state.categorySelect[currFile]
 
 
 
 
 
241
 
242
  _saved_renames = st.session_state.speakerRenames.get(currFile, {})
243
  raw_to_display = {sp: _saved_renames.get(sp, sp) for sp in speakerNames}
@@ -269,7 +274,7 @@ try:
269
  # Sidebar
270
  # -----------------------------------------------------------------------
271
 
272
- ui.render_categories_sidebar(currFile, categorySelections, all_speakers_display, raw_to_display)
273
  ui.render_rename_sidebar(currFile, speakerNames, all_speaker_tokens)
274
 
275
  # -----------------------------------------------------------------------
 
237
  speakers_dataFrame = st.session_state.summaries[currFile]["speakers_dataFrame"]
238
  currDF, _ = su.annotationToSimpleDataFrame(currAnnotation)
239
  unusedSpeakers = st.session_state.unusedSpeakers[currFile]
240
+ # categorySelect is global tokens; extract raw speaker IDs for currFile
241
+ _prefix = currFile + ": "
242
+ categorySelections = [
243
+ [t[len(_prefix):] for t in tokens if t.startswith(_prefix)]
244
+ for tokens in st.session_state.categorySelect
245
+ ]
246
 
247
  _saved_renames = st.session_state.speakerRenames.get(currFile, {})
248
  raw_to_display = {sp: _saved_renames.get(sp, sp) for sp in speakerNames}
 
274
  # Sidebar
275
  # -----------------------------------------------------------------------
276
 
277
+ ui.render_categories_sidebar(all_speaker_tokens)
278
  ui.render_rename_sidebar(currFile, speakerNames, all_speaker_tokens)
279
 
280
  # -----------------------------------------------------------------------