Update sonogram.py

sonogram.py

@@ -232,18 +232,12 @@ class Sonogram():
             duration = 0
             tracking = False
             for stepClass,timeIncrement,members in zip(timeStepClass,st,timeStepMembers):
-                print(currList)
-                print(stepClass,timeIncrement,members)
                 if stepClass == c:
-                    print(1)
                     if currMembers == members:
-                        print(2)
                         
                         duration += min(stepSize,maxTime-timeIncrement)
                     else:
-                        print(3)
                         if tracking:
-                            print(4)
                             singleDimList.append((currMembers,Segment(start,start+duration)))
                             currList.append((currMembers,Segment(start,start+duration)))
                             if start+duration > endTime:
@@ -257,9 +251,7 @@ class Sonogram():
                         duration += min(stepSize,maxTime-timeIncrement)
                         tracking = True
                 else:
-                    print(5)
                     if tracking:
-                        print(6)
                         singleDimList.append((currMembers,Segment(start,start+duration)))
                         currList.append((currMembers,Segment(start,start+duration)))
                         if start+duration > endTime:
@@ -276,11 +268,8 @@ class Sonogram():
             categorySegmentList.append(currList)
         # Check where we left off
         if endTime != maxTime:
-            print(f'End {endTime} : Max {maxTime}')
-            # Create silence at end if needed
-            if maxTime-st[-1] > stepSize:
-                singleDimList.append((None,Segment(start,start+duration)))
-                categorySegmentList[2].append((None,Segment(st[-1] + stepSize,maxTime)))
+            singleDimList.append((None,Segment(endTime,maxTime)))
+            categorySegmentList[2].append((None,Segment(endTime,maxTime)))
 
         # For debug
         singleDimList = sorted(singleDimList,key=lambda index : index[1].start)