Update app.py

app.py

@@ -450,7 +450,6 @@ if uploaded_file_paths is not None and len(uploaded_file_paths) > 0:
     valid_files = []
     file_paths = []
     file_names = []
-    # Reset valid_files?
     for uploaded_file in uploaded_file_paths:
         if not uploaded_file.name.lower().endswith(supported_file_types):
             st.error('File must be of type: {}'.format(supported_file_types))
@@ -466,22 +465,29 @@ if uploaded_file_paths is not None and len(uploaded_file_paths) > 0:
     # Save valid file names
     if len(valid_files) > 0:
         file_names = [f.name for f in valid_files]
-    while (len(st.session_state.results) < len(valid_files)):
-        st.session_state.results.append([])
-    while (len(st.session_state.summaries) < len(valid_files)):
-        st.session_state.summaries.append([])
-    while (len(st.session_state.unusedSpeakers) < len(valid_files)):
-        st.session_state.unusedSpeakers.append([])
-    while (len(st.session_state.categorySelect) < len(valid_files)):
-        tempCategories = [[] for cat in st.session_state.categories]
-        st.session_state.categorySelect.append(tempCategories)
-    while (len(st.session_state.summaries) < len(valid_files)):
-        st.session_state.summaries.append([])
-    while (len(st.session_state.speakerRenames) < len(valid_files)):
-        st.session_state.speakerRenames.append({})
-    while (len(st.session_state.speakerClips) < len(valid_files)):
-        st.session_state.speakerClips.append({})
-        
+
+    # --- Reconcile per-file state lists against the new file set ---
+    # For each new file, carry over existing state if the filename matches a
+    # previously analyzed file, otherwise insert a fresh empty slot.
+    prev_names = st.session_state.file_names
+    def _carry(old_list, empty_val_fn):
+        new_list = []
+        for name in file_names:
+            if name in prev_names:
+                idx = prev_names.index(name)
+                new_list.append(old_list[idx] if idx < len(old_list) else empty_val_fn())
+            else:
+                new_list.append(empty_val_fn())
+        return new_list
+
+    st.session_state.results       = _carry(st.session_state.results,        lambda: [])
+    st.session_state.summaries     = _carry(st.session_state.summaries,       lambda: [])
+    st.session_state.unusedSpeakers= _carry(st.session_state.unusedSpeakers,  lambda: [])
+    st.session_state.categorySelect= _carry(st.session_state.categorySelect,
+                                            lambda: [[] for _ in st.session_state.categories])
+    st.session_state.speakerRenames= _carry(st.session_state.speakerRenames,  lambda: {})
+    st.session_state.speakerClips  = _carry(st.session_state.speakerClips,    lambda: {})
+
     st.session_state.file_names = file_names
     st.session_state.valid_files = valid_files
     st.session_state.file_paths = file_paths
@@ -689,13 +695,10 @@ try:
             st.sidebar.markdown(f"**{display_label}**")
             if sp in file_clips:
                 st.sidebar.audio(file_clips[sp], format="audio/wav")
-            # IMPORTANT: label must stay fixed (use sp, not display_label) so Streamlit
-            # does not treat it as a new widget and reset its state on every rerun.
             new_name = st.sidebar.text_input(
-                sp,
+                f"Rename {display_label}",
                 placeholder="e.g. John",
-                key=widget_key,
-                label_visibility="collapsed"
+                key=widget_key
             )
             if new_name.strip():
                 st.session_state.speakerRenames[currFileIndex][sp] = new_name.strip()