Update visualization/visualizer.py

visualization/visualizer.py

@@ -24,15 +24,27 @@ def visualize_simulation(file, simulation_index: Optional[int]) -> Optional[str]
         simulation = simulations[simulation_index]
 
         logger = SimulationLogger()
+
+        # Log metadata
         logger.log_metadata(simulation['instructions'])
+
+        # Log subjects (individual frames)
         logger.log_subjects(simulation['subjects'])
 
-        if 'helper_keyframes' in simulation:
-            logger.log_helper_keyframes(simulation['helper_keyframes'])
-        
+        # Log subject trajectories (full center path)
+        logger.log_subject_trajectories(simulation['subjects'])
+
+        # Log camera trajectory (full path)
         logger.log_camera_trajectory(simulation['cameraFrames'])
+
+        # Log individual camera frames
         logger.log_camera_frames(simulation['cameraFrames'])
 
+        # Log helper keyframes if available
+        if 'helper_keyframes' in simulation:
+            logger.log_helper_keyframes(simulation['helper_keyframes'])
+
+        # Save the visualization data
         rr.save(rrd_path)
 
         return rrd_path