Fix a bug with Role

app.py

@@ -230,7 +230,7 @@ try:
 
     st.header(f"Analysis of file {currFile}")
     TAB_NAMES = ["Rename Speaker", "Roles", "Categories",
-                 "Treemap", "Timeline", "Time Spoken", "Data"]
+                 "Treemap", "Timeline", "Time Spoken", "Download"]
     renameTab, pie2, sunburst1, treemap1, timeline, bar1, dataTab = st.tabs(TAB_NAMES)
 
     currAnnotation, currTotalTime = st.session_state.results[currFile]
@@ -312,17 +312,17 @@ try:
     # -----------------------------------------------------------------------
 
     with dataTab:
-        displayDF = apply_speaker_renames_to_df(currDF, currFile, column="Resource")
-        # Clean up for download: drop Task, rename Resource -> Speaker, sort by Start, add Role
-        # Build raw-speaker -> role lookup before renaming the column
+        # Build raw-speaker -> role lookup against the RAW currDF (before renames
+        # are applied) so the map keys match the original SPEAKER_## labels.
         raw_to_role = {
-            raw_sp: st.session_state.categories[i]
+            token.split(": ", 1)[1]: st.session_state.categories[i]
             for i, tokens in enumerate(st.session_state.categorySelect)
             for token in tokens
-            for raw_sp in [token.split(": ", 1)[1]]
             if token.startswith(f"{currFile}: ")
         }
+        displayDF = currDF.copy()
         displayDF["Role"] = displayDF["Resource"].map(raw_to_role).fillna("")
+        displayDF = apply_speaker_renames_to_df(displayDF, currFile, column="Resource")
         displayDF = displayDF.drop(columns=["Task"], errors="ignore")
         displayDF = displayDF.rename(columns={"Resource": "Speaker"})
         if "Start" in displayDF.columns: