Update app.py

app.py

@@ -286,11 +286,17 @@ def updateMultiSelect():
     if result:
         currAnnotation, _ = result
         speakerNames = list(currAnnotation.labels())
+
+        # Always restore rename widgets from the persistent speakerRenames dict
+        # so that coming back to a file after visiting another shows saved names.
+        saved_renames = st.session_state.speakerRenames.get(fileName, {})
         raw_to_display = {}
         for sp in speakerNames:
             wk = f"rename_{fileName}_{sp}"
-            live = st.session_state.get(wk, "").strip()
-            raw_to_display[sp] = live if live else sp
+            saved = saved_renames.get(sp, "")
+            st.session_state[wk] = saved  # unconditionally restore
+            raw_to_display[sp] = saved if saved else sp
+
         for i, category in enumerate(st.session_state['categories']):
             raw_choices = st.session_state['categorySelect'][fileName][i]
             st.session_state[f'multiselect_{category}'] = [raw_to_display.get(sp, sp) for sp in raw_choices]
@@ -773,7 +779,8 @@ try:
         current_renames = st.session_state.speakerRenames[currFile]
         for sp in speakerNames:
             widget_key = f"rename_{currFile}_{sp}"
-            # Seed the widget state once; never pass value= so reruns don't overwrite it
+            # Seed the widget state on first render only; updateMultiSelect handles
+            # restoring it unconditionally whenever the user switches files.
             if widget_key not in st.session_state:
                 st.session_state[widget_key] = current_renames.get(sp, "")
             live_name = st.session_state[widget_key].strip()