Update colab.py

colab.py

@@ -174,18 +174,21 @@ class DanceGenerator:
 
         if pattern_idx < num_patterns - 1:
             t = progress * (num_patterns - 1) - pattern_idx
-            pose = self._interpolate_poses(
-                patterns[pattern_idx],
-                patterns[pattern_idx + 1],
-                t
-            )
+            # Extract pose arrays from pattern dictionaries
+            pose1 = patterns[pattern_idx]['pose']
+            pose2 = patterns[pattern_idx + 1]['pose']
+            pose = self._interpolate_poses(pose1, pose2, t)
         else:
-            pose = patterns[-1]
+            pose = patterns[-1]['pose']
 
         return pose
 
     def _interpolate_poses(self, pose1, pose2, t):
         """Smoothly interpolate between two poses"""
+        if isinstance(pose1, dict):
+            pose1 = pose1['pose']
+        if isinstance(pose2, dict):
+            pose2 = pose2['pose']
         return pose1 * (1 - t) + pose2 * t
 
     def _create_enhanced_dance_frame(self, pose_array, frame_size, add_effects=True):