Update app.py

app.py

@@ -367,6 +367,10 @@ if 'unusedSpeakers' not in st.session_state:
     st.session_state.unusedSpeakers = []
 if 'file_names' not in st.session_state:
     st.session_state.file_names = []
+if 'valid_files' not in st.session_state:
+    st.session_state.valid_files = []
+if 'file_paths' not in st.session_state:
+    st.session_state.file_paths = []
 if 'showSummary' not in st.session_state:
     st.session_state.showSummary = 'No'
     
@@ -388,12 +392,12 @@ uploaded_file_paths = st.file_uploader("Upload an audio of classroom activity to
 supported_file_types = ('.wav','.mp3','.mp4','.txt','.rttm','.csv')
 viewChoices = ["Voice Categories","Custom Categories","Detailed Voice Categories","Voice Category Treemap","Speaker Timeline","Time per Speaker"]
 
-valid_files = []
-file_paths = []
+valid_files = st.session_state.valid_files
+file_paths = st.session_state.file_paths
 currDF = None
 temp_dir = tempfile.mkdtemp()
 
-if uploaded_file_paths is not None:
+if uploaded_file_paths is not None and len(uploaded_file_paths) > 0:
     print("Found file paths")
     valid_files = []
     file_paths = []
@@ -429,6 +433,8 @@ if uploaded_file_paths is not None:
         st.session_state.speakerRenames.append({})
         
     st.session_state.file_names = file_names
+    st.session_state.valid_files = valid_files
+    st.session_state.file_paths = file_paths
 
 file_names = st.session_state.file_names
     
@@ -517,6 +523,8 @@ if st.sidebar.button("Load Demo Example"):
     valid_files=[sampleUpload]
     file_paths=[sampleUpload.path]
     file_names=[sampleUpload.name]
+    st.session_state.valid_files = valid_files
+    st.session_state.file_paths = file_paths
     start_time = time.time()
     st.session_state.file_names = file_names
     # Save valid file names