Upload dance_generator.py

dance_generator.py

@@ -249,4 +249,35 @@ class DanceGenerator:
             (23, 24), (23, 25), (24, 26),  # Legs
             (25, 27), (26, 28), (27, 29), (28, 30),
             (29, 31), (30, 32)
-        ]
+        ]
+        
+    def _smooth_transition(self, prev_pose, current_pose, smoothing_factor=0.3):
+        """Create smooth transition between poses"""
+        if prev_pose is None or current_pose is None:
+            return current_pose
+            
+        # Interpolate between previous and current pose
+        smoothed_pose = (1 - smoothing_factor) * prev_pose + smoothing_factor * current_pose
+        
+        # Ensure the smoothed pose maintains proper proportions
+        # Normalize joint positions relative to hip center
+        hip_center_idx = 23  # Index for hip center landmark
+        
+        prev_hip = prev_pose[hip_center_idx]
+        current_hip = current_pose[hip_center_idx]
+        smoothed_hip = smoothed_pose[hip_center_idx]
+        
+        # Adjust positions relative to hip center
+        for i in range(len(smoothed_pose)):
+            if i != hip_center_idx:
+                # Calculate relative positions
+                prev_relative = prev_pose[i] - prev_hip
+                current_relative = current_pose[i] - current_hip
+                
+                # Interpolate relative positions
+                smoothed_relative = (1 - smoothing_factor) * prev_relative + smoothing_factor * current_relative
+                
+                # Update smoothed pose
+                smoothed_pose[i] = smoothed_hip + smoothed_relative
+        
+        return smoothed_pose