Spaces:
Sleeping
Sleeping
Add role column to downloadable csv file
Browse files
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:
|