Enhance log_simulation function with color support for vertices and camera trajectory

app.py

@@ -1,7 +1,6 @@
 import json
 import numpy as np
 import rerun as rr
-from rerun.components import Material
 import spaces
 import gradio as gr
 from gradio_rerun import Rerun
@@ -81,7 +80,8 @@ def log_simulation(simulation_data: Dict[str, Any]) -> None:
             rr.Mesh3D(
                 vertex_positions=vertices,
                 indices=faces,
-                mesh_material=Material(albedo_factor=subject_color)
+                # Apply color to all vertices
+                colors=np.tile(subject_color, (len(vertices), 1))
             ),
             timeless=True
         )
@@ -96,11 +96,15 @@ def log_simulation(simulation_data: Dict[str, Any]) -> None:
         [vector3_to_numpy(frame['position']) for frame in camera_frames])
     rr.log(
         "world/camera_trajectory",
-        rr.Points3D(camera_positions),
+        rr.Points3D(
+            camera_positions,
+            # Cyan color for trajectory
+            colors=np.full((len(camera_positions), 4), [0.0, 0.8, 0.8, 1.0])
+        ),
         timeless=True
     )
 
-    # Log camera positions over time
+    # Log camera movement over time
     for frame_idx, camera_frame in enumerate(camera_frames):
         rr.set_time_sequence("frame", frame_idx)
 
@@ -126,6 +130,12 @@ def log_simulation(simulation_data: Dict[str, Any]) -> None:
             )
         )
 
+        # Log frame number
+        rr.log(
+            "metadata/current_frame",
+            rr.TextDocument(f"Frame: {frame_idx + 1}/{len(camera_frames)}"),
+        )
+
 
 def load_simulation_data(file) -> Tuple[Optional[List[Dict[str, Any]]], Optional[List[str]]]:
     """Load simulation data from JSON file and return simulations with their descriptions"""