Update sonogram_utility.py

sonogram_utility.py

@@ -339,7 +339,30 @@ def annotationToDataFrame(myAnnotation):
     df = pd.DataFrame(dataList)
     return df, timeSummary
     
+def annotationToSimpleDataFrame(myAnnotation):
+    dataList = []
+    speakerDict = {}
+    for currSpeaker in myAnnotation.labels():
+        if currSpeaker not in speakerDict.keys():
+            speakerDict[currSpeaker] = []
+        for currSegment in myAnnotation.subset([currSpeaker]).itersegments():
+            speakerDict[currSpeaker].append(currSegment)
 
+    timeSummary = {}
+    for key in speakerDict.keys():
+        if key not in timeSummary.keys():
+            timeSummary[key] = 0
+        for speakingSegment in speakerDict[key]:
+            timeSummary[key] += speakingSegment.duration
+    
+    for key in speakerDict.keys():
+        for k, speakingSegment in enumerate(speakerDict[key]):
+            speakerName = key
+            startPoint = speakingSegment.start
+            endPoint = speakingSegment.end
+            dataList.append(dict(Task=speakerName + f".{k}", Start=startPoint, Finish=endPoint, Resource=speakerName))
+    df = pd.DataFrame(dataList)
+    return df, timeSummary
 
 def calcCategories(myAnnotation,categories):
     categorySlots = []