update

app.py

@@ -36,15 +36,15 @@ def visualize_trajectory(dataset, index):
         is_robot = item["is_robot"]
         data_source = item["data_source"]
         
-        # Get the frames data (datasets.Audio object with bytes)
+        # Get the frames data (video file path)
         frames_data = item["frames"]
         
-        # Extract video bytes from datasets.Audio object
-        if hasattr(frames_data, 'bytes') and frames_data["bytes"] is not None:
-            video_bytes = frames_data["bytes"]
-            frames_info = f"Video bytes: {len(video_bytes)} bytes"
+        # Handle video file path (could be local path or HuggingFace Hub URL)
+        if isinstance(frames_data, str):
+            video_path = frames_data
+            frames_info = f"Video file: {video_path}"
         else:
-            return None, "Error: No video bytes found", "Error: No video bytes found", None
+            return None, "Error: Invalid video path", "Error: Invalid video path", None
         
         # Create metadata
         metadata = f"""
@@ -63,8 +63,8 @@ def visualize_trajectory(dataset, index):
         **Trajectory ID:** {item.get('id', 'N/A')}
         """
         
-        # Return video bytes for Gradio to display
-        return video_bytes, metadata, f"Trajectory {index}", None
+        # Return video path for Gradio to display
+        return video_path, metadata, f"Trajectory {index}", None
         
     except Exception as e:
         return None, f"Error: {str(e)}", f"Error: {str(e)}", None