update

app.py

@@ -20,7 +20,7 @@ def get_available_configs(dataset_name):
         print(f"Error getting configs: {e}")
         return []
 
-def visualize_trajectory(dataset, index):
+def visualize_trajectory(dataset, index, dataset_name=None):
     """
     Function to retrieve a trajectory and its metadata from the dataset.
     """
@@ -41,7 +41,11 @@ def visualize_trajectory(dataset, index):
         
         # Handle video file path (could be local path or HuggingFace Hub URL)
         if isinstance(frames_data, str):
-            video_path = f"https://huggingface.co/datasets/aliangdw/rfm/resolve/main/{frames_data}"
+            # Use dynamic dataset name if provided, otherwise fallback to default
+            if dataset_name:
+                video_path = f"https://huggingface.co/datasets/{dataset_name}/resolve/main/{frames_data}"
+            else:
+                video_path = f"https://huggingface.co/datasets/aliangdw/rfm/resolve/main/{frames_data}"
             frames_info = f"Video file: {video_path}"
         else:
             return None, "Error: Invalid video path", "Error: Invalid video path", None
@@ -145,7 +149,7 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
         else:
             return None, status, "", 0, 0
     
-    def update_trajectory(dataset, index):
+    def update_trajectory(dataset, index, dataset_name=None):
         """Update the displayed trajectory."""
         if dataset is None:
             return None, "No dataset loaded", "No dataset loaded", None
@@ -156,22 +160,22 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
             index = len(dataset) - 1
         elif index < 0:
             index = 0
-        return visualize_trajectory(dataset, int(index))
+        return visualize_trajectory(dataset, int(index), dataset_name)
     
-    def next_trajectory(dataset, current_idx):
+    def next_trajectory(dataset, current_idx, dataset_name=None):
         """Go to next trajectory."""
         if dataset is None:
             return current_idx, None, "No dataset loaded", "No dataset loaded", None
         next_idx = min(current_idx + 1, len(dataset) - 1)
-        video, metadata, title, image = visualize_trajectory(dataset, next_idx)
+        video, metadata, title, image = visualize_trajectory(dataset, next_idx, dataset_name)
         return next_idx, video, metadata, title, image
     
-    def prev_trajectory(dataset, current_idx):
+    def prev_trajectory(dataset, current_idx, dataset_name=None):
         """Go to previous trajectory."""
         if dataset is None:
             return current_idx, None, "No dataset loaded", "No dataset loaded", None
         prev_idx = max(current_idx - 1, 0)
-        video, metadata, title, image = visualize_trajectory(dataset, prev_idx)
+        video, metadata, title, image = visualize_trajectory(dataset, prev_idx, dataset_name)
         return prev_idx, video, metadata, title, image
     
     def update_slider_range(dataset):
@@ -204,8 +208,8 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     )
     
     slider.change(
-        fn=lambda dataset, idx: update_trajectory(dataset, idx),
-        inputs=[current_dataset, slider],
+        fn=lambda dataset, idx, dataset_name: update_trajectory(dataset, idx, dataset_name),
+        inputs=[current_dataset, slider, dataset_name_input],
         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,
@@ -214,8 +218,8 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     )
     
     next_btn.click(
-        fn=next_trajectory,
-        inputs=[current_dataset, current_index],
+        fn=lambda dataset, idx, dataset_name: next_trajectory(dataset, idx, dataset_name),
+        inputs=[current_dataset, current_index, dataset_name_input],
         outputs=[current_index, video_output, metadata_output, title_output, image_output]
     ).then(
         fn=lambda idx: idx,
@@ -224,8 +228,8 @@ with gr.Blocks(title="RFM Dataset Visualizer") as demo:
     )
     
     prev_btn.click(
-        fn=prev_trajectory,
-        inputs=[current_dataset, current_index],
+        fn=lambda dataset, idx, dataset_name: prev_trajectory(dataset, idx, dataset_name),
+        inputs=[current_dataset, current_index, dataset_name_input],
         outputs=[current_index, video_output, metadata_output, title_output, image_output]
     ).then(
         fn=lambda idx: idx,