Update app.py

app.py

@@ -68,9 +68,7 @@ def processFile(filePath):
     print("Detecting speakers")
     annotations = pipeline({"waveform": waveformEnhanced, "sample_rate": sampleRate})
     print("Speakers Detected")
-    speakerList = su.annotationToSpeakerList(annotations)
-    print("Speakers split into list")
-    return (speakerList, annotations, int(waveformEnhanced.shape[-1]/sampleRate))
+    return (annotations, int(waveformEnhanced.shape[-1]/sampleRate))
 
 def addCategory():
     newCategory = st.session_state.categoryInput
@@ -128,7 +126,7 @@ def analyze(inFileName):
         print(f"Found at index {currFileIndex}")
         if len(st.session_state.results) > currFileIndex and len(st.session_state.summaries) > currFileIndex and len(st.session_state.results[currFileIndex]) > 0:
             # Handle 
-            currSpeakerList, currAnnotation, currTotalTime = st.session_state.results[currFileIndex]
+            currAnnotation, currTotalTime = st.session_state.results[currFileIndex]
             speakerNames = currAnnotation.labels()
     
             # Update other categories
@@ -363,7 +361,7 @@ else:
             start_time = time.time()
             totalFiles = len(valid_files)
             for i in range(totalFiles):
-                if len(st.session_state.results) > i and len(st.session_state.results[i]) > 0  and st.session_state.results[i][0] == file_names[i]:
+                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'):
@@ -376,16 +374,16 @@ else:
                         for segment in annotations.itersegments():
                             if segment.end > totalSeconds:
                                 totalSeconds = segment.end
-                        st.session_state.results[i] = (speakerList,annotations, totalSeconds)
+                        st.session_state.results[i] = (annotations, totalSeconds)
                         st.session_state.summaries[i] = {}
                         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}'):
-                        speakerList, annotations, totalSeconds = processFile(file_paths[i])
+                        annotations, totalSeconds = processFile(file_paths[i])
                         print(f"Finished processing {file_paths[i]}")
-                        st.session_state.results[i] = (speakerList,annotations, totalSeconds)
+                        st.session_state.results[i] = (annotations, totalSeconds)
                         print("Results saved")
                         st.session_state.summaries[i] = {}
                         print("Summaries saved")
@@ -405,7 +403,7 @@ try:
     currFileIndex = file_names.index(currFile)
     if len(st.session_state.results) > currFileIndex and len(st.session_state.summaries) > currFileIndex and len(st.session_state.results[currFileIndex]) > 0:
         # Handle 
-        currSpeakerList, currAnnotation, currTotalTime = st.session_state.results[currFileIndex]
+        currAnnotation, currTotalTime = st.session_state.results[currFileIndex]
         speakerNames = currAnnotation.labels()
         # Update other categories
         unusedSpeakers = st.session_state.unusedSpeakers[currFileIndex]
@@ -570,7 +568,7 @@ if len(st.session_state.results) > 0:
             }
             allCategories = copy.deepcopy(st.session_state.categories)
             for i in indices:
-                currSpeakerList, currAnnotation, currTotalTime = st.session_state.results[i]
+                currAnnotation, currTotalTime = st.session_state.results[i]
                 categorySelections = st.session_state["categorySelect"][i]
                 catSummary,extraCats = su.calcCategories(currAnnotation,categorySelections)
                 st.session_state.summaries[i]["categories"] = (catSummary,extraCats)