Update app.py

app.py

@@ -210,7 +210,7 @@ def calcCategories(annotation,maxTime):
                 copyOfNo.remove(emptySlot)
                 copyOfNo += removeOverlap(emptySlot,timeSlot)
         noVoice = copyOfNo
-    return noVoice, oneVoice, multiVoice
+    return noVoice, oneVoice, multiVoice, rawData 
 
 def sumTimes(timeList):
     totalTime = 0
@@ -355,12 +355,12 @@ for i, tab in enumerate(audio_tabs):
             currDF = all_dataFrame
 
             # TBD CLEAN THIS UP!!!
-            noVoice2, oneVoice2, multiVoice2 = calcCategories(annotations,totalSeconds)
+            noVoice2, oneVoice2, multiVoice2, rawSample = calcCategories(annotations,totalSeconds)
             noVoice2.sort()
             oneVoice2.sort()
             multiVoice2.sort()
-            sList,timeList = sumTimesPerSpeaker(oneVoice)
-            multiSpeakerList, multiTimeList = sumMultiTimesPerSpeaker(multiVoice)
+            sList,timeList = sumTimesPerSpeaker(oneVoice2)
+            multiSpeakerList, multiTimeList = sumMultiTimesPerSpeaker(multiVoice2)
             
             multiVoice = annotations.get_overlap()
             singleVoice = annotations.extrude(multiVoice).get_timeline()
@@ -378,8 +378,8 @@ for i, tab in enumerate(audio_tabs):
 
             df4: pd.DataFrame = pd.DataFrame(
                 {
-                    "values": [sumTimes(rawSample["speaker 1"]),sumTimes(rawSample["speaker 2"]),sumTimes(rawSample["speaker 3"])],
-                    "names": ["speaker 1","speaker 2","speaker 3"]
+                    "values": [sumTimes(rawSample[key]) for key in rawSample.keys()],
+                    "names": [key for key in rawSample.keys()]
                 }
             )
             df4.name = "df4"