Update app.py

app.py

@@ -310,12 +310,12 @@ def process_video(video_path, process_seconds=20, conf_threshold=0.2, max_det=8)
     for i in range(0, len(filtered_points), frame_interval):
         current_points = filtered_points[:i+1]
         
-        # 轨迹动画帧
+        # 轨迹动画帧 - 修改颜色方案，与静态图保持一致
         frame_trajectory = base_trajectory.copy()
         if len(current_points) > 1:
             points = np.array(current_points, dtype=np.int32)
             for j in range(len(points) - 1):
-                ratio = j / (len(points) - 1)
+                ratio = j / (len(current_points) - 1)  # 修改这里，使用当前总长度
                 color = (
                     int((1 - ratio) * 255),  # B
                     50,                       # G
@@ -327,7 +327,7 @@ def process_video(video_path, process_seconds=20, conf_threshold=0.2, max_det=8)
             cv2.circle(frame_trajectory, tuple(points[-1]), 8, (0, 0, 255), -1)
         trajectory_frames.append(frame_trajectory)
         
-        # 热图动画帧
+        # 热图动画帧 - 修改颜色方案，使用与静态图相同的配色
         frame_heatmap = base_heatmap.copy()
         for x, y in current_points:
             if 0 <= x < width and 0 <= y < height:
@@ -339,6 +339,7 @@ def process_video(video_path, process_seconds=20, conf_threshold=0.2, max_det=8)
         if np.max(frame_heatmap) > 0:
             frame_heatmap_norm = cv2.normalize(frame_heatmap, None, 0, 255, cv2.NORM_MINMAX)
             frame_heatmap_color = cv2.applyColorMap(frame_heatmap_norm.astype(np.uint8), cv2.COLORMAP_JET)
+            # 添加白色背景，与静态图保持一致
             frame_heatmap_color = cv2.addWeighted(frame_heatmap_color, 0.7, np.full_like(frame_heatmap_color, 255), 0.3, 0)
             heatmap_frames.append(frame_heatmap_color)