Add multi-file demo option

Demo button renamed. Added multi-file demo button. Adjusted "Analyze all" button to a toggle. Multi-file demo begins by artificially loading local rttms and then uses the "Analyze all" pathway to process them.

app.py

@@ -475,70 +475,6 @@ if uploaded_file_paths is not None and len(uploaded_file_paths) > 0:
 file_names = st.session_state.file_names
 file_paths_dict = st.session_state.file_paths  # dict {fname: path}
 
-if len(file_names) == 0:
-    st.text("Upload file(s) to enable analysis")
-else:
-    if st.button("Analyze All New Audio",key=f"button_all"):
-        if len(file_names) == 0:
-            st.error('Upload file(s) first!')
-        else:
-            print("Start analyzing")
-            start_time = time.time()
-            totalFiles = len(file_names)
-            for i, fname in enumerate(file_names):
-                fpath = file_paths_dict.get(fname, "")
-                if st.session_state.results.get(fname):
-                    continue
-                if fpath.lower().endswith('.txt'):
-                    with st.spinner(text=f'Loading Demo File {i+1} of {totalFiles}'):
-                        speakerList, annotations = su.loadAudioTXT(fpath)
-                        printV(annotations,4)
-                        totalSeconds = 0
-                        for segment in annotations.itersegments():
-                            if segment.end > totalSeconds:
-                                totalSeconds = segment.end
-                        st.session_state.results[fname] = (annotations, totalSeconds)
-                        st.session_state.summaries[fname] = {}
-                        st.session_state.unusedSpeakers[fname] = list(annotations.labels())
-                elif fpath.lower().endswith('.rttm'):
-                    with st.spinner(text=f'Loading File {i+1} of {totalFiles}'):
-                        speakerList, annotations = su.loadAudioRTTM(fpath)
-                        printV(annotations,4)
-                        totalSeconds = 0
-                        for segment in annotations.itersegments():
-                            if segment.end > totalSeconds:
-                                totalSeconds = segment.end
-                        st.session_state.results[fname] = (annotations, totalSeconds)
-                        st.session_state.summaries[fname] = {}
-                        st.session_state.unusedSpeakers[fname] = list(annotations.labels())
-                elif fpath.lower().endswith('.csv'):
-                    with st.spinner(text=f'Loading File {i+1} of {totalFiles}'):
-                        speakerList, annotations = su.loadAudioCSV(fpath)
-                        printV(annotations,4)
-                        totalSeconds = 0
-                        for segment in annotations.itersegments():
-                            if segment.end > totalSeconds:
-                                totalSeconds = segment.end
-                        st.session_state.results[fname] = (annotations, totalSeconds)
-                        st.session_state.summaries[fname] = {}
-                        st.session_state.unusedSpeakers[fname] = list(annotations.labels())
-                else:
-                    with st.spinner(text=f'Processing File {i+1} of {totalFiles}'):
-                        annotations, totalSeconds, waveform, sample_rate = processFile(fpath)
-                        print(f"Finished processing {fpath}")
-                        st.session_state.results[fname] = (annotations, totalSeconds)
-                        st.session_state.summaries[fname] = {}
-                        st.session_state.unusedSpeakers[fname] = list(annotations.labels())
-                    with st.spinner(text=f'Generating speaker clips for File {i+1} of {totalFiles}'):
-                        generate_speaker_clips(annotations, waveform, sample_rate, fname)
-                        del waveform
-                        print(f"Speaker clips generated for {fpath}")
-                with st.spinner(text=f'Analyzing File {i+1} of {totalFiles}'):
-                    analyze(fname)
-                    print(f"Finished analyzing {fpath}")
-            print(f"Took {time.time() - start_time} seconds to analyze {totalFiles} files!")
-            st.success(f"Took {time.time() - start_time} seconds to analyze {totalFiles} files!")
-
 class FakeUpload:
     def __init__(self,filepath):
         self.path = filepath
@@ -546,7 +482,7 @@ class FakeUpload:
         ]
 demoPath = "sample.rttm"
 isDemo = False
-if st.sidebar.button("Load Demo Example"):
+if st.sidebar.button("Single File Demo"):
     demoName = demoPath.split('/')[-1]
     start_time = time.time()
     if demoName not in st.session_state.file_names:
@@ -569,12 +505,116 @@ if st.sidebar.button("Load Demo Example"):
         st.session_state.results[demoName] = (annotations, totalSeconds)
         st.session_state.summaries[demoName] = {}
         st.session_state.unusedSpeakers[demoName] = list(annotations.labels())
-        with st.spinner(text=f'Analyzing Demo Data'):
-            analyze(demoName)
+    with st.spinner(text=f'Analyzing Demo Data'):
+        analyze(demoName)
     st.success(f"Took {time.time() - start_time} seconds to analyze the demo file!")
     st.session_state.select_currFile = demoName
     isDemo = True
-            
+
+multiFileDemoPaths = ["audioSamples/media-afc-cal-afc1986022_sr01a05.rttm","audioSamples/media-afc-cal-afc1986022_sr34a01.rttm","audioSamples/media-afc-cal-afc1986022_sr14b02.rttm",
+            "audioSamples/media-afc-cal-afc1986022_sr52a02.rttm","audioSamples/media-afc-cal-afc1986022_sr14b01.rttm"]
+# TODO: prepare audio for playback of audio
+multiFileAudioPaths = ["audioSamples/media-afc-cal-afc1986022_sr01a05.mp3","audioSamples/media-afc-cal-afc1986022_sr34a01.mp3","audioSamples/media-afc-cal-afc1986022_sr14b02.mp3",
+            "audioSamples/media-afc-cal-afc1986022_sr52a02.mp3","audioSamples/media-afc-cal-afc1986022_sr14b01.mp3"]
+
+if st.sidebar.button("Multiple Files Demo"):
+    for demoPath in multiFileDemoPaths:
+        demoName = demoPath.split('/')[-1]
+        start_time = time.time()
+        if demoName not in st.session_state.file_names:
+            st.session_state.file_names.append(demoName)
+        st.session_state.file_paths[demoName] = demoPath
+        st.session_state.results.setdefault(demoName, [])
+        st.session_state.summaries.setdefault(demoName, {})
+        st.session_state.unusedSpeakers.setdefault(demoName, [])
+        st.session_state.categorySelect.setdefault(demoName, [[] for _ in st.session_state.categories])
+        st.session_state.speakerRenames.setdefault(demoName, {})
+        st.session_state.speakerClips.setdefault(demoName, {})
+        file_names = st.session_state.file_names
+    
+        with st.spinner(text=f'Loading Demo Sample'):
+            speakerList, annotations = su.loadAudioRTTM(demoPath)
+            totalSeconds = 0
+            for segment in annotations.itersegments():
+                if segment.end > totalSeconds:
+                    totalSeconds = segment.end
+            st.session_state.results[demoName] = (annotations, totalSeconds)
+            st.session_state.summaries[demoName] = {}
+            st.session_state.unusedSpeakers[demoName] = list(annotations.labels())
+        # TODO: Remove if not necessary
+        #st.session_state.select_currFile = demoName
+        isDemo = True
+        analyzeAllToggle = True
+
+analyzeAllToggle = False
+if len(file_names) == 0:
+    st.text("Upload file(s) to enable analysis")
+else:
+    if st.button("Analyze All New Audio",key=f"button_all"):
+        if len(file_names) == 0:
+            st.error('Upload file(s) first!')
+        else:
+            analyzeAllToggle = True
+
+if analyzeAllToggle == True:
+    print("Start analyzing")
+    start_time = time.time()
+    totalFiles = len(file_names)
+    for i, fname in enumerate(file_names):
+        fpath = file_paths_dict.get(fname, "")
+        if st.session_state.results.get(fname):
+            continue
+        if fpath.lower().endswith('.txt'):
+            with st.spinner(text=f'Loading Demo File {i+1} of {totalFiles}'):
+                speakerList, annotations = su.loadAudioTXT(fpath)
+                printV(annotations,4)
+                totalSeconds = 0
+                for segment in annotations.itersegments():
+                    if segment.end > totalSeconds:
+                        totalSeconds = segment.end
+                st.session_state.results[fname] = (annotations, totalSeconds)
+                st.session_state.summaries[fname] = {}
+                st.session_state.unusedSpeakers[fname] = list(annotations.labels())
+        elif fpath.lower().endswith('.rttm'):
+            with st.spinner(text=f'Loading File {i+1} of {totalFiles}'):
+                speakerList, annotations = su.loadAudioRTTM(fpath)
+                printV(annotations,4)
+                totalSeconds = 0
+                for segment in annotations.itersegments():
+                    if segment.end > totalSeconds:
+                        totalSeconds = segment.end
+                st.session_state.results[fname] = (annotations, totalSeconds)
+                st.session_state.summaries[fname] = {}
+                st.session_state.unusedSpeakers[fname] = list(annotations.labels())
+        elif fpath.lower().endswith('.csv'):
+            with st.spinner(text=f'Loading File {i+1} of {totalFiles}'):
+                speakerList, annotations = su.loadAudioCSV(fpath)
+                printV(annotations,4)
+                totalSeconds = 0
+                for segment in annotations.itersegments():
+                    if segment.end > totalSeconds:
+                        totalSeconds = segment.end
+                st.session_state.results[fname] = (annotations, totalSeconds)
+                st.session_state.summaries[fname] = {}
+                st.session_state.unusedSpeakers[fname] = list(annotations.labels())
+        else:
+            with st.spinner(text=f'Processing File {i+1} of {totalFiles}'):
+                annotations, totalSeconds, waveform, sample_rate = processFile(fpath)
+                print(f"Finished processing {fpath}")
+                st.session_state.results[fname] = (annotations, totalSeconds)
+                st.session_state.summaries[fname] = {}
+                st.session_state.unusedSpeakers[fname] = list(annotations.labels())
+            with st.spinner(text=f'Generating speaker clips for File {i+1} of {totalFiles}'):
+                generate_speaker_clips(annotations, waveform, sample_rate, fname)
+                del waveform
+                print(f"Speaker clips generated for {fpath}")
+        with st.spinner(text=f'Analyzing File {i+1} of {totalFiles}'):
+            analyze(fname)
+            print(f"Finished analyzing {fpath}")
+    print(f"Took {time.time() - start_time} seconds to analyze {totalFiles} files!")
+    st.success(f"Took {time.time() - start_time} seconds to analyze {totalFiles} files!")
+    analyzeAllToggle = False
+
 currFile = st.sidebar.selectbox('Current File', file_names, on_change=updateMultiSelect, key="select_currFile")
 
 if isDemo: