Update app.py

app.py

@@ -167,15 +167,6 @@ def update_dashboard(dataset_idx: int| None = None):
 with gr.Blocks() as demo:
     gr.Markdown("# Argilla Dataset Dashboard")
 
-    datasets_dropdown = gr.Dropdown(label="Select your dataset")
-    datasets_dropdown.choices = [(dataset.name, idx) for idx, dataset in enumerate(datasets)]
-
-    def set_selected_dataset(dataset_idx) -> None:
-        global selected_dataset
-
-        dataset = datasets[dataset_idx]
-        selected_dataset = dataset
-
 
     with gr.Row():
         gauge_output = gr.Plot(label="Overall Progress")
@@ -188,16 +179,11 @@ with gr.Blocks() as demo:
 
     demo.load(
         update_dashboard,
-        inputs=[datasets_dropdown],
+        inputs=[selected_dataset],
         outputs=[gauge_output, treemap_output, leaderboard_output],
         every=5,
     )
 
-    datasets_dropdown.change(
-        update_dashboard,
-        inputs=[datasets_dropdown],
-        outputs=[gauge_output, treemap_output, leaderboard_output],
-    )
 
 
 if __name__ == "__main__":
@@ -343,11 +329,10 @@ def get_datasets(client: rg.Argilla) -> List[rg.Dataset]:
 
 datasets = get_datasets(client)
 
-def update_dashboard(dataset_idx: int| None = None):
+def update_dashboard(dataset: str| None = None):
     if dataset_idx is None:
         return [None, None, None]
 
-    dataset = datasets[dataset_idx]
     progress = get_progress(dataset)
 
     gauge_chart = create_gauge_chart(progress)
@@ -379,17 +364,11 @@ with gr.Blocks() as demo:
 
     demo.load(
         update_dashboard,
-        inputs=[datasets_dropdown],
+        inputs=[selected_dataset],
         outputs=[gauge_output, treemap_output, leaderboard_output],
         every=5,
     )
 
-    datasets_dropdown.change(
-        update_dashboard,
-        inputs=[datasets_dropdown],
-        outputs=[gauge_output, treemap_output, leaderboard_output],
-    )
-
 
 if __name__ == "__main__":
     demo.launch()