Update sonogram.py

sonogram.py

@@ -211,12 +211,12 @@ class Sonogram():
             timeStepMembers.append(memberSet)
                 
         for i,item in enumerate(timeStepAggregate):
-            categoryId = max(item, key=item.get)
-            timeStepClass.append(categoryId)
+            cat = max(item, key=item.get)
+            timeStepClass.append(cat)
             # For group decisions, members include all in window
-            if categoryId == 0: 
+            if cat == 'group': 
                 timeStepMembers[i] = '+'.join(timeStepMembers[i])
-            elif categoryId == 1:
+            elif cat == 'individual':
                 timeStepMembers[i] = timeStepMembers[i][0]
             else:
                 timeStepMembers[i] = None
@@ -233,12 +233,17 @@ class Sonogram():
             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:
@@ -252,7 +257,9 @@ 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: