Add in RTTM TXT viewer functionality

app.py

@@ -290,7 +290,7 @@ if 'results' not in st.session_state:
 if 'summaries' not in st.session_state:
     st.session_state.summaries = []
 if 'categories' not in st.session_state:
-    st.session_state.categories = ["Lecturer","Audience"]
+    st.session_state.categories = ["Instructor","Audience"]
     st.session_state.categorySelect = []
 # Single Use
 if 'removeCategory' not in st.session_state:
@@ -314,7 +314,7 @@ if not isGPU:
 
 uploaded_file_paths = st.file_uploader("Upload an audio of classroom activity to analyze", accept_multiple_files=True)
 
-supported_file_types = ('.wav','.mp3','.mp4','.txt')
+supported_file_types = ('.wav','.mp3','.mp4','.txt','.rttm')
 
 valid_files = []
 file_paths = []
@@ -328,7 +328,7 @@ if uploaded_file_paths is not None:
     file_names = []
     # Reset valid_files?
     for uploaded_file in uploaded_file_paths:
-        if not uploaded_file.name.endswith(supported_file_types):
+        if not uploaded_file.name.lower().endswith(supported_file_types):
             st.error('File must be of type: {}'.format(supported_file_types))
             uploaded_file = None
         else:
@@ -353,12 +353,7 @@ if uploaded_file_paths is not None:
         st.session_state.categorySelect.append(tempCategories)
     while (len(st.session_state.summaries) < len(valid_files)):
         st.session_state.summaries.append([])
-    maybeRemove = '''
-    # Clear replaced files
-    for i in range(len(valid_files)):
-        if len(st.session_state.results[i]) > 0 and st.session_state.results[i][0] != file_names[i]:
-            st.session_state.results[i] = []
-            st.session_state.summaries[i] = []'''
+        
     st.session_state.file_names = file_names
 
 file_names = st.session_state.file_names
@@ -377,11 +372,25 @@ else:
                 if len(st.session_state.results) > i and len(st.session_state.results[i]) > 0:
                     continue
                 # Text files use sample data
-                if file_paths[i].endswith('.txt'):
+                if file_paths[i].lower().endswith('.txt'):
                     with st.spinner(text=f'Loading Demo File {i+1} of {totalFiles}'):
-                        time.sleep(1)
                         # RTTM load as filler
-                        speakerList, annotations = su.loadAudioRTTM(sample_data[i])
+                        speakerList, annotations = su.loadAudioTXT(file_paths[i])
+                        print(annotations)
+                        # Approximate total seconds
+                        totalSeconds = 0
+                        for segment in annotations.itersegments():
+                            if segment.end > totalSeconds:
+                                totalSeconds = segment.end
+                        st.session_state.results[i] = (annotations, totalSeconds)
+                        st.session_state.summaries[i] = {}
+                        speakerNames = annotations.labels()
+                        st.session_state.unusedSpeakers[i] = speakerNames
+                elif file_paths[i].lower().endswith('.rttm'):
+                    with st.spinner(text=f'Loading File {i+1} of {totalFiles}'):
+                        # RTTM load as filler
+                        speakerList, annotations = su.loadAudioRTTM(file_paths[i])
+                        print(annotations)
                         # Approximate total seconds
                         totalSeconds = 0
                         for segment in annotations.itersegments():
@@ -392,7 +401,6 @@ else:
                         speakerNames = annotations.labels()
                         st.session_state.unusedSpeakers[i] = speakerNames
                 else:
-                    #st.info(file_paths[i])
                     with st.spinner(text=f'Processing File {i+1} of {totalFiles}'):
                         annotations, totalSeconds = processFile(file_paths[i])
                         print(f"Finished processing {file_paths[i]}")
@@ -406,6 +414,7 @@ else:
                 with st.spinner(text=f'Analyzing File {i+1} of {totalFiles}'):
                     analyze(file_names[i])
                     print(f"Finished analyzing {file_paths[i]}")
+            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!")
  
 currFile = st.sidebar.selectbox('Current File', file_names,on_change=updateMultiSelect,key="select_currFile")