update

app.py

@@ -50,16 +50,14 @@ def visualize_trajectory(dataset, index):
         metadata = f"""
         ## Trajectory Information
         
-        **Task:** {task}
+        **Language Task:** {task}
         
-        **Optimality:** {optimal}
+        **Optimal:** {optimal}
         
-        **Data Type:** {'🤖 Robot' if is_robot else '👤 Human'}
+        **Data Type:** {'Robot' if is_robot else 'Human'}
         
         **Source:** {data_source}
-        
-        **Frames:** {frames_info}
-        
+                
         **Trajectory ID:** {item.get('id', 'N/A')}
         """
         
@@ -116,13 +114,14 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
             prev_btn = gr.Button("⬅️ Previous", variant="secondary")
         
         with gr.Column(scale=2):
-            # Slider for navigation
+            # Slider for navigation with dynamic max
             slider = gr.Slider(
                 minimum=0,
-                maximum=100,  # Will be updated when dataset loads
+                maximum=0,
                 step=1,
                 value=0,
-                label="Trajectory Index"
+                label="Select a dataset first",
+                interactive=False
             )
         
         with gr.Column(scale=1):
@@ -141,7 +140,7 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
         if dataset is not None:
             max_index = len(dataset) - 1
             info = f"**Dataset Info:**\n- **Total Trajectories:** {len(dataset)}\n- **Features:** {list(dataset.features.keys())}"
-            # Return the dataset length for slider configuration
+            # Return the dataset length for number input configuration
             return dataset, status, info, 0, max_index
         else:
             return None, status, "", 0, 0
@@ -150,12 +149,14 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
         """Update the displayed trajectory."""
         if dataset is None:
             return None, "No dataset loaded", "No dataset loaded", None
-        # Ensure index is within bounds
-        if index >= len(dataset):
+        # Ensure index is within bounds and is a valid number
+        if index is None or not isinstance(index, (int, float)):
+            index = 0
+        elif index >= len(dataset):
             index = len(dataset) - 1
         elif index < 0:
             index = 0
-        return visualize_trajectory(dataset, index)
+        return visualize_trajectory(dataset, int(index))
     
     def next_trajectory(dataset, current_idx):
         """Go to next trajectory."""
@@ -173,11 +174,33 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
         video, metadata, title, image = visualize_trajectory(dataset, prev_idx)
         return prev_idx, video, metadata, title, image
     
+    def update_slider_range(dataset):
+        """Update the slider with new maximum value based on dataset length."""
+        if dataset is not None:
+            max_value = len(dataset) - 1
+            return gr.update(
+                maximum=max_value,
+                value=0,  # Reset to beginning
+                label=f"Trajectory Index (0 to {max_value})",
+                interactive=True
+            )
+        else:
+            return gr.update(
+                maximum=0,
+                value=0,
+                label="Select a dataset first",
+                interactive=False
+            )
+    
     # Connect the components
     load_btn.click(
         fn=load_dataset,
         inputs=[dataset_name_input, config_name_input],
-        outputs=[current_dataset, status_output, dataset_info, current_index, slider]
+        outputs=[current_dataset, status_output, dataset_info, current_index]
+    ).then(
+        fn=update_slider_range,
+        inputs=current_dataset,
+        outputs=slider
     )
     
     slider.change(
@@ -213,7 +236,11 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     # Load initial dataset
     demo.load(
         fn=lambda: load_dataset("aliangdw/rfm", "libero_10"),
-        outputs=[current_dataset, status_output, dataset_info, current_index, slider]
+        outputs=[current_dataset, status_output, dataset_info, current_index]
+    ).then(
+        fn=update_slider_range,
+        inputs=current_dataset,
+        outputs=slider
     )
 
 # Launch the app