Fix export PNG to use current sort state instead of default order

The response_data path was unreliable — it returned None mid-interaction,
causing fallback to the old sort. Now we re-apply the session-state sort
model (always in sync with AgGrid) to the original data before returning,
so the exported PNG always matches what the user sees in the grid.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

visualization/betting_page.py

@@ -348,6 +348,8 @@ def _build_cell_style_js(style_field: str, align_right: bool, emphasize: bool):
 def _render_sortable_grid(display_df: pd.DataFrame, numeric_df: pd.DataFrame, css_df: pd.DataFrame, *, family: str, side: str) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
     sort_key = _get_sort_state_key(family, side)
     sort_model = st.session_state.get(sort_key) or []
+    # Keep originals so we can re-sort after the response updates the sort model
+    orig_display, orig_numeric, orig_css = display_df.copy(), numeric_df.copy(), css_df.copy()
     display_df, numeric_df, css_df = _apply_sort_model(display_df, numeric_df, css_df, sort_model)
     if AgGrid is None or GridOptionsBuilder is None:
         styler = display_df.style.apply(lambda _: css_df, axis=None)
@@ -404,20 +406,10 @@ def _render_sortable_grid(display_df: pd.DataFrame, numeric_df: pd.DataFrame, cs
     new_sort_model = _extract_sort_model(response)
     if new_sort_model:
         st.session_state[sort_key] = new_sort_model
-    response_data = response.get("data") if isinstance(response, dict) else None
-    if response_data is None:
-        return display_df, numeric_df, css_df
-    response_df = pd.DataFrame(response_data)
-    if response_df.empty or "__row_key" not in response_df.columns:
-        return display_df, numeric_df, css_df
-    order = response_df["__row_key"].astype(str).tolist()
-    keyed_display = display_df.assign(__row_key=numeric_df["player_event_market_key"].astype(str)).set_index("__row_key")
-    keyed_numeric = numeric_df.assign(__row_key=numeric_df["player_event_market_key"].astype(str)).set_index("__row_key")
-    keyed_css = css_df.assign(__row_key=numeric_df["player_event_market_key"].astype(str)).set_index("__row_key")
-    order = [key for key in order if key in keyed_display.index]
-    if not order:
-        return display_df, numeric_df, css_df
-    return keyed_display.loc[order].reset_index(drop=True), keyed_numeric.loc[order].reset_index(drop=True), keyed_css.loc[order].reset_index(drop=True)
+    # Re-apply the now-current sort model to the original data so the export
+    # PNG always reflects exactly what the grid is showing the user.
+    current_sort = st.session_state.get(sort_key) or []
+    return _apply_sort_model(orig_display, orig_numeric, orig_css, current_sort)
 
 
 def _build_side_table(df: pd.DataFrame, book_cols: list[str], prev_snap: dict[str, float], steam_set: set[str], model_probs: pd.DataFrame | None) -> tuple[pd.DataFrame, pd.DataFrame]: