duongthienz commited on
Commit
170c898
·
verified ·
1 Parent(s): 8814211

Add role column to downloadable csv file

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -293,7 +293,16 @@ try:
293
 
294
  with dataTab:
295
  displayDF = apply_speaker_renames_to_df(currDF, currFile, column="Resource")
296
- # Clean up for download: drop Task, rename Resource -> Speaker, sort by Start
 
 
 
 
 
 
 
 
 
297
  displayDF = displayDF.drop(columns=["Task"], errors="ignore")
298
  displayDF = displayDF.rename(columns={"Resource": "Speaker"})
299
  if "Start" in displayDF.columns:
 
293
 
294
  with dataTab:
295
  displayDF = apply_speaker_renames_to_df(currDF, currFile, column="Resource")
296
+ # Clean up for download: drop Task, rename Resource -> Speaker, sort by Start, add Role
297
+ # Build raw-speaker -> role lookup before renaming the column
298
+ raw_to_role = {
299
+ raw_sp: st.session_state.categories[i]
300
+ for i, tokens in enumerate(st.session_state.categorySelect)
301
+ for token in tokens
302
+ for raw_sp in [token.split(": ", 1)[1]]
303
+ if token.startswith(f"{currFile}: ")
304
+ }
305
+ displayDF["Role"] = displayDF["Resource"].map(raw_to_role).fillna("")
306
  displayDF = displayDF.drop(columns=["Task"], errors="ignore")
307
  displayDF = displayDF.rename(columns={"Resource": "Speaker"})
308
  if "Start" in displayDF.columns: