feat: Improve URL selection handling in DataUI; clear stale selections on load

Author: dwr-psandhu

dvue/dataui.py

@@ -1538,7 +1538,11 @@ def _setup_selection_url_sync(self):
         if not loc:
             return
 
-        # --- Restore on load ---
+        # --- Restore on load (one-shot) ---
+        # Apply any selection encoded in the URL, then immediately clear it
+        # so that the next server startup begins with no stale selection.
+        # The watcher below is registered *after* this block, so the clear
+        # is not overwritten by the change event from the initial restore.
         query_params = loc.query_params or {}
         sel_str = query_params.get("sel", "")
         if sel_str:
@@ -1547,6 +1551,9 @@ def _setup_selection_url_sync(self):
                 indices = self._dfcat.index[self._dfcat["name"].isin(names)].tolist()
                 if indices and hasattr(self, "display_table"):
                     self.display_table.selection = indices
+            # Clear from URL before the watcher is registered so the next
+            # startup does not auto-select rows from a stale browser URL.
+            loc.update_query(sel="")
 
         # --- Watch for changes and update URL ---
         def _on_selection_change(event):

dvue/session_persistence.py

@@ -319,30 +319,7 @@ def make_app():
         if user_id:
             _registry[user_id] = {"mgr": mgr, "ui": ui, "template": tmpl}
 
-        sel = saved.get("selection", [])
-
         def _on_load():
-            # Replay saved selection → re-trigger Plot action.  Only
-            # meaningful after a server restart when diskcache had a saved
-            # selection; fresh users have no saved state.
-            if sel and hasattr(ui, "display_table") and hasattr(ui, "_registered_actions"):
-                plot_cb = next(
-                    (
-                        a["callback"]
-                        for a in ui._registered_actions
-                        if a.get("name") == "Plot"
-                    ),
-                    None,
-                )
-                if plot_cb:
-                    n_rows = len(ui.display_table.value) if ui.display_table.value is not None else 0
-                    valid_sel = [i for i in sel if i < n_rows]
-                    if valid_sel:
-                        ui.display_table.selection = valid_sel
-                    pn.state.curdoc.add_next_tick_callback(
-                        lambda: plot_cb(None, ui)
-                    )
-
             # Wire live-persistence watchers so any param change is
             # immediately flushed to diskcache (only when persist=True).
             if persist:
@@ -690,31 +667,11 @@ def make_app():
         if user_id:
             _registry[user_id] = {"mgr": mgr, "ui": ui, "template": tmpl}
 
-        sel = saved.get("selection", [])
-
         def _on_load():
             # Capture the running Tornado IOLoop so the pywebview GUI thread can
             # schedule confirmation callbacks via IOLoop.add_callback (thread-safe).
             _app_state["ioloop"] = _tornado_ioloop.IOLoop.current()
 
-            if sel and hasattr(ui, "display_table") and hasattr(ui, "_registered_actions"):
-                plot_cb = next(
-                    (
-                        a["callback"]
-                        for a in ui._registered_actions
-                        if a.get("name") == "Plot"
-                    ),
-                    None,
-                )
-                if plot_cb:
-                    n_rows = len(ui.display_table.value) if ui.display_table.value is not None else 0
-                    valid_sel = [i for i in sel if i < n_rows]
-                    if valid_sel:
-                        ui.display_table.selection = valid_sel
-                    pn.state.curdoc.add_next_tick_callback(
-                        lambda: plot_cb(None, ui)
-                    )
-
             if persist:
                 def _do_save(event=None):
                     if user_id: