gallery: refresh after admin changes

app.py

@@ -352,8 +352,8 @@ def _admin_promote(
     table_df: pd.DataFrame | None,
     method: str | None,
     profile: gr.OAuthProfile | None,
-) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
-    """Promote every ticked row, then refresh the admin table + both tiers.
+) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, str]:
+    """Promote ticked rows, then refresh admin, leaderboard, and gallery.
 
     Re-checks :func:`admin.is_admin` server-side so a tampered client
     that re-enables the button still can't write.
@@ -372,14 +372,14 @@ def _admin_promote(
     gr.Info(f"Promoted {len(ids)} row(s) to validated ({method}).")
     validated, unvalidated, _ = _safe_load_split()
     admin_df, _ = _safe_load_admin()
-    return admin_df, validated, unvalidated
+    return admin_df, validated, unvalidated, _gallery_iframe_html()
 
 
 def _admin_demote(
     table_df: pd.DataFrame | None,
     profile: gr.OAuthProfile | None,
-) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
-    """Demote every ticked row, then refresh the admin table + both tiers."""
+) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, str]:
+    """Demote ticked rows, then refresh admin, leaderboard, and gallery."""
     if not is_admin(profile):
         raise gr.Error("You are not in the admin set.")
     ids = _selected_ids(table_df)
@@ -392,15 +392,15 @@ def _admin_demote(
     gr.Info(f"Demoted {len(ids)} row(s) to unvalidated.")
     validated, unvalidated, _ = _safe_load_split()
     admin_df, _ = _safe_load_admin()
-    return admin_df, validated, unvalidated
+    return admin_df, validated, unvalidated, _gallery_iframe_html()
 
 
 def _admin_delete(
     table_df: pd.DataFrame | None,
     confirm: bool,
     profile: gr.OAuthProfile | None,
-) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, gr.Checkbox, gr.Button]:
-    """Delete every ticked row (artifacts + row), then refresh and disarm.
+) -> tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, str, gr.Checkbox, gr.Button]:
+    """Delete ticked rows, then refresh admin, leaderboard, gallery, and disarm.
 
     Resets the confirm checkbox and re-disables the delete button on
     the way out so the next deletion needs a fresh, deliberate confirm.
@@ -423,6 +423,7 @@ def _admin_delete(
         admin_df,
         validated,
         unvalidated,
+        _gallery_iframe_html(),
         gr.Checkbox(value=False),
         gr.Button(interactive=False),
     )
@@ -641,8 +642,8 @@ with gr.Blocks(title="CADGenBench Leaderboard", theme=gr.themes.Soft()) as block
         # self-contained HTML doc inlined into an iframe `srcdoc` with
         # base64 thumbnails, so it keeps its own style context and makes
         # no second HTTP request (works on the private Space, where
-        # HF's edge 404s in-browser custom-route fetches). Built at boot;
-        # the Refresh button rebuilds it after a promotion/new result.
+        # HF's edge 404s in-browser custom-route fetches). Built at boot,
+        # rebuilt on page load, and refreshed after admin actions.
         gallery_html = gr.HTML(value=_gallery_iframe_html())
         gallery_refresh_btn = gr.Button("Refresh gallery", size="sm")
         gallery_refresh_btn.click(
@@ -863,12 +864,12 @@ to publish the resulting row on the public leaderboard.
         promote_btn.click(
             fn=_admin_promote,
             inputs=[admin_table, admin_method_radio],
-            outputs=[admin_table, validated_view, unvalidated_view],
+            outputs=[admin_table, validated_view, unvalidated_view, gallery_html],
         )
         demote_btn.click(
             fn=_admin_demote,
             inputs=[admin_table],
-            outputs=[admin_table, validated_view, unvalidated_view],
+            outputs=[admin_table, validated_view, unvalidated_view, gallery_html],
         )
         delete_confirm.change(
             fn=_arm_delete, inputs=[delete_confirm], outputs=delete_btn,
@@ -877,7 +878,7 @@ to publish the resulting row on the public leaderboard.
             fn=_admin_delete,
             inputs=[admin_table, delete_confirm],
             outputs=[
-                admin_table, validated_view, unvalidated_view,
+                admin_table, validated_view, unvalidated_view, gallery_html,
                 delete_confirm, delete_btn,
             ],
         )
@@ -901,6 +902,7 @@ to publish the resulting row on the public leaderboard.
     # per page load; LoginButton clicks also re-trigger this through
     # Gradio's auth-event plumbing.
     blocks.load(fn=_enable_submit_when_logged_in, outputs=submit_btn)
+    blocks.load(fn=_gallery_iframe_html, outputs=gallery_html)
 
     # Same per-load OAuth read, gating the Admin tab's controls on
     # membership in the CADGENBENCH_ADMINS set. Logged-out / non-admin

tests/test_proxy.py

@@ -239,6 +239,41 @@ def test_gate_admin_controls_refreshes_live_table(monkeypatch):
     assert table_update.interactive is True
 
 
+def test_admin_delete_refreshes_gallery(monkeypatch):
+    """Deleting rows also replaces the Gallery iframe srcdoc."""
+    table_df = pd.DataFrame(
+        [
+            {
+                "select": True,
+                "submission_id": "validated-old-row",
+            }
+        ]
+    )
+    empty_admin = pd.DataFrame(columns=leaderboard.ADMIN_COLUMNS)
+    empty_validated = pd.DataFrame(columns=leaderboard.VALIDATED_LEADERBOARD_COLS)
+    empty_unvalidated = pd.DataFrame(columns=leaderboard.LEADERBOARD_COLS)
+    deleted: list[list[str]] = []
+
+    monkeypatch.setattr(app, "is_admin", lambda profile: True)
+    monkeypatch.setattr(app, "delete_rows", lambda ids: deleted.append(ids))
+    monkeypatch.setattr(
+        app, "_safe_load_split", lambda: (empty_validated, empty_unvalidated, None)
+    )
+    monkeypatch.setattr(app, "_safe_load_admin", lambda: (empty_admin, None))
+    monkeypatch.setattr(app, "_gallery_iframe_html", lambda: "<iframe>empty</iframe>")
+    monkeypatch.setattr(app.gr, "Info", lambda *a, **k: None)
+
+    admin_df, validated, unvalidated, gallery_html, _confirm, _delete_btn = (
+        app._admin_delete(table_df, True, types.SimpleNamespace(username="michaelr27"))
+    )
+
+    assert deleted == [["validated-old-row"]]
+    assert admin_df is empty_admin
+    assert validated is empty_validated
+    assert unvalidated is empty_unvalidated
+    assert gallery_html == "<iframe>empty</iframe>"
+
+
 def test_data_error_banner_md_present_on_error_empty_otherwise():
     """Banner markdown is non-empty (and names the cause) only on error."""
     assert app._data_error_banner_md(None) == ""