Update sonogram_utility.py

sonogram_utility.py

@@ -374,7 +374,10 @@ def calcSpeakingTypes(myAnnotation,maxTime):
                 continue
             # Add to one voice
             oneVoice.append((currID,currSegment))
-    for _,timeSlot in multiVoice:
+    ovAnnotation = Annotation()
+    mvAnnotation = Annotation()
+    for currID,timeSlot in multiVoice:
+        mvAnnotation[timeSlot] = currID
         copyOfNo = copy.deepcopy(noVoice)
         for emptySlot in noVoice:
             if checkForOverlap(timeSlot,emptySlot) is None:
@@ -383,7 +386,8 @@ def calcSpeakingTypes(myAnnotation,maxTime):
                 copyOfNo.remove(emptySlot)
                 copyOfNo += removeOverlap(emptySlot,timeSlot)
         noVoice = copyOfNo
-    for _,timeSlot in oneVoice:
+    for currID,timeSlot in oneVoice:
+        ovAnnotation[timeSlot] = currID
         copyOfNo = copy.deepcopy(noVoice)
         for emptySlot in noVoice:
             if checkForOverlap(timeSlot,emptySlot) is None:
@@ -392,7 +396,11 @@ def calcSpeakingTypes(myAnnotation,maxTime):
                 copyOfNo.remove(emptySlot)
                 copyOfNo += removeOverlap(emptySlot,timeSlot)
         noVoice = copyOfNo
-    return noVoice, oneVoice, multiVoice
+    nvAnnotation = Annotation()
+    for emptySlot in noVoice:
+        nvAnnotation[emptySlot] = "None"
+    
+    return nvAnnotation, ovAnnotation, mvAnnotation
 
 def timeToString(timeInSeconds):
     if isinstance(timeInSeconds,list):