update

app.py

@@ -119,7 +119,7 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
             # Slider for navigation
             slider = gr.Slider(
                 minimum=0,
-                maximum=0,
+                maximum=100,  # Will be updated when dataset loads
                 step=1,
                 value=0,
                 label="Trajectory Index"
@@ -141,7 +141,8 @@ 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 dataset, status, info, max_index, 0
+            # Return the dataset length for slider configuration
+            return dataset, status, info, 0, max_index
         else:
             return None, status, "", 0, 0
     
@@ -149,6 +150,11 @@ 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):
+            index = len(dataset) - 1
+        elif index < 0:
+            index = 0
         return visualize_trajectory(dataset, index)
     
     def next_trajectory(dataset, current_idx):
@@ -171,13 +177,17 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     load_btn.click(
         fn=load_dataset,
         inputs=[dataset_name_input, config_name_input],
-        outputs=[current_dataset, status_output, dataset_info, slider, current_index]
+        outputs=[current_dataset, status_output, dataset_info, current_index, slider]
     )
     
     slider.change(
         fn=lambda dataset, idx: update_trajectory(dataset, idx),
         inputs=[current_dataset, slider],
         outputs=[video_output, metadata_output, title_output, image_output]
+    ).then(
+        fn=lambda dataset, idx: idx if dataset is None or idx < len(dataset) else len(dataset) - 1,
+        inputs=[current_dataset, slider],
+        outputs=[current_index]
     )
     
     next_btn.click(
@@ -203,7 +213,7 @@ 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, slider, current_index]
+        outputs=[current_dataset, status_output, dataset_info, current_index, slider]
     )
 
 # Launch the app