Add draggability to plots (#153)

* add draggability

* ui

* update dep

Author: abidlabs

pyproject.toml

@@ -14,7 +14,7 @@ requires-python = ">=3.10"
 dependencies = [
     "pandas<3.0.0",
     "huggingface-hub<1.0.0",
-    "gradio>=5.43.1,<6.0.0",
+    "gradio>=5.44.0,<6.0.0",
     "numpy<3.0.0",
     "pillow<12.0.0",
 ]

trackio/ui.py

@@ -617,7 +617,9 @@ def update_dashboard(
             ):
                 # Render direct metrics at this level
                 if group_data["direct_metrics"]:
-                    with gr.Row(key=f"row-{group_name}-direct"):
+                    with gr.Draggable(
+                        key=f"row-{group_name}-direct", orientation="row"
+                    ):
                         for metric_name in group_data["direct_metrics"]:
                             metric_df = master_df.dropna(subset=[metric_name])
                             color = "run" if "run" in metric_df.columns else None
@@ -665,7 +667,7 @@ def update_dashboard(
                             key=f"accordion-{group_name}-{subgroup_name}",
                             preserved_by_key=["value", "open"],
                         ):
-                            with gr.Row(key=f"row-{group_name}-{subgroup_name}"):
+                            with gr.Draggable(key=f"row-{group_name}-{subgroup_name}"):
                                 for metric_name in subgroup_metrics:
                                     metric_df = master_df.dropna(subset=[metric_name])
                                     color = (