Update colab.py

colab.py

@@ -191,64 +191,65 @@ class DanceGenerator:
     def _create_enhanced_dance_frame(self, pose_array, frame_size, add_effects=True):
         """Create enhanced visualization frame with effects"""
         height, width = frame_size
-        frame = np.zeros((height, width, 3), dtype=np.uint8)
-
+        # Change background from black to light gray for better visibility
+        frame = np.ones((height, width, 3), dtype=np.uint8) * 240  # Light gray background
+        
         # Convert coordinates
         points = (pose_array[:, :2] * [width, height]).astype(int)
-
-        # Draw enhanced skeleton
+        
+        # Draw enhanced skeleton with thicker lines and more visible colors
         connections = self._get_pose_connections()
         for connection in connections:
             start_idx, end_idx = connection
             if start_idx < len(points) and end_idx < len(points):
-                # Draw glowing lines
                 if add_effects:
                     self._draw_glowing_line(
                         frame,
                         points[start_idx],
                         points[end_idx],
-                        (0, 255, 0)
+                        (0, 100, 255),  # Orange color for skeleton
+                        thickness=4
                     )
                 else:
                     cv2.line(frame,
                             tuple(points[start_idx]),
                             tuple(points[end_idx]),
-                            (0, 255, 0), 2)
+                            (0, 100, 255), 4)
 
-        # Draw enhanced joints
+        # Draw enhanced joints with larger radius
         for point in points:
             if add_effects:
-                self._draw_glowing_point(frame, point, (0, 0, 255))
+                self._draw_glowing_point(frame, point, (255, 0, 0), radius=6)  # Blue joints
             else:
-                cv2.circle(frame, tuple(point), 4, (0, 0, 255), -1)
+                cv2.circle(frame, tuple(point), 6, (255, 0, 0), -1)
 
         return frame
 
-    def _draw_glowing_line(self, frame, start, end, color, thickness=2):
-        """Draw a line with glow effect"""
-        # Draw main line
-        cv2.line(frame, tuple(start), tuple(end), color, thickness)
-
-        # Draw glow
+    def _draw_glowing_line(self, frame, start, end, color, thickness=4):
+        """Draw a line with enhanced glow effect"""
+        # Draw outer glow
         for i in range(3):
-            alpha = 0.3 - i * 0.1
-            thickness = thickness + 2
+            alpha = 0.5 - i * 0.15
+            thick = thickness + (i * 4)
             cv2.line(frame, tuple(start), tuple(end),
                     tuple([int(c * alpha) for c in color]),
-                    thickness)
-
-    def _draw_glowing_point(self, frame, point, color, radius=4):
-        """Draw a point with glow effect"""
-        # Draw main point
-        cv2.circle(frame, tuple(point), radius, color, -1)
+                    thick)
+        
+        # Draw main line
+        cv2.line(frame, tuple(start), tuple(end), color, thickness)
 
-        # Draw glow
+    def _draw_glowing_point(self, frame, point, color, radius=6):
+        """Draw a point with enhanced glow effect"""
+        # Draw outer glow
         for i in range(3):
-            alpha = 0.3 - i * 0.1
-            r = radius + i * 2
+            alpha = 0.5 - i * 0.15
+            r = radius + (i * 3)
             cv2.circle(frame, tuple(point), r,
                       tuple([int(c * alpha) for c in color]),
                       -1)
+        
+        # Draw main point
+        cv2.circle(frame, tuple(point), radius, color, -1)
 
     def _landmarks_to_array(self, landmarks):
         """Convert MediaPipe landmarks to numpy array"""