partial success on visualzier

app.py

@@ -117,6 +117,8 @@ def visualize_trajectory(dataset, index, dataset_name=None):
         
         # Get the frames data (video file path)
         frames_data = item["frames"]
+
+        partial_success = item.get("partial_success", "N/A")
         
         # Handle video file path (could be local path or HuggingFace Hub URL)
         if isinstance(frames_data, str):
@@ -138,6 +140,8 @@ def visualize_trajectory(dataset, index, dataset_name=None):
         **Language Task:** {task}
         
         **Quality Label:** {quality_label}
+
+        **Partial Success:** {partial_success}
         
         **Data Type:** {'Robot' if is_robot else 'Human'}