Update sonogram_utility.py

sonogram_utility.py

@@ -216,25 +216,26 @@ def sumSegments(segmentList):
         total += s.duration
     return total
 
-def sumTimes(annotation):
-    return annotation.get_timeline(False).duration()
+def sumTimes(myAnnotation):
+    return myAnnotation.get_timeline(False).duration()
     
-def sumTimesPerSpeaker(timeSlotList):
+def sumTimesPerSpeaker(myAnnotation):
     speakerList = []
     timeList = []
-    for speaker,timeSlot in timeSlotList:
+    for speaker in myAnnotation.labels():
         if speaker not in speakerList:
             speakerList.append(speaker)
             timeList.append(0)
-        timeList[speakerList.index(speaker)] += timeSlot.duration
+        timeList[speakerList.index(speaker)] += sumTimes(myAnnotation.subset([speaker]))
     return speakerList, timeList
     
-def sumMultiTimesPerSpeaker(timeSlotList):
+def sumMultiTimesPerSpeaker(myAnnotation):
     speakerList = []
     timeList = []
-    sList,tList = sumTimesPerSpeaker(timeSlotList)
+    sList,tList = sumTimesPerSpeaker(myAnnotation)
     for i,speakerGroup in enumerate(sList):
-        for speaker in speakerGroup:
+        speakerSplit = speakerGroup.split('+')
+        for speaker in speakerSplit:
             if speaker not in speakerList:
                 speakerList.append(speaker)
                 timeList.append(0)