Speed initial records load and hide Gradio chrome

app.py

@@ -33,6 +33,31 @@ HYDRATE_STATUS = maybe_hydrate_from_dataset(DATA_DIR)
 STORE = SolutionStore(DATA_DIR)
 
 
+@lru_cache(maxsize=1)
+def cached_reference_records() -> list[dict[str, Any]]:
+    return STORE.load_reference_records()
+
+
+@lru_cache(maxsize=1)
+def cached_verified_records() -> list[dict[str, Any]]:
+    return STORE.load_records()
+
+
+@lru_cache(maxsize=64)
+def cached_public_records(setup: str = "", show_all: bool = False) -> list[dict[str, Any]]:
+    return STORE.public_records(setup or None, show_all=show_all)
+
+
+def clear_page_caches() -> None:
+    cached_reference_records.cache_clear()
+    cached_verified_records.cache_clear()
+    cached_public_records.cache_clear()
+    setup_choices.cache_clear()
+    coordinate_records_by_case.cache_clear()
+    family_page_html.cache_clear()
+    directory_html.cache_clear()
+
+
 CSS = """
 :root {
   color-scheme: light;
@@ -188,6 +213,24 @@ gradio-app,
   color: #fff !important;
 }
 
+footer,
+.gradio-container footer,
+.gradio-container .footer,
+.gradio-container .built-with,
+.gradio-container .api-docs,
+.gradio-container a[href*="view=api"],
+.gradio-container a[href*="/api"],
+.gradio-container button[aria-label="Settings"],
+.gradio-container button[title="Settings"],
+.gradio-container button[aria-label="More"],
+.gradio-container button[aria-label="More options"],
+.gradio-container button[title="More"],
+.gradio-container button[title="More options"],
+.gradio-container [data-testid="settings-button"],
+.gradio-container [data-testid="more-button"] {
+  display: none !important;
+}
+
 .section {
   padding: 30px 0 8px;
 }
@@ -1250,14 +1293,14 @@ def esc(value: Any) -> str:
 
 
 def setup_title(setup: str) -> str:
-    for record in STORE.load_reference_records():
+    for record in cached_reference_records():
         if record.get("setup") == setup and record.get("title"):
             return str(record["title"])
     return setup
 
 
 def setup_updated(setup: str) -> str:
-    for record in STORE.load_reference_records():
+    for record in cached_reference_records():
         if record.get("setup") != setup:
             continue
         meta = record.get("friedman_reference")
@@ -1268,8 +1311,18 @@ def setup_updated(setup: str) -> str:
     return ""
 
 
+@lru_cache(maxsize=96)
 def setup_choices() -> list[str]:
-    choices = [setup for setup in STORE.setups() if setup != "All"]
+    choices = {
+        str(record["setup"])
+        for record in cached_reference_records()
+        if record.get("setup")
+    }
+    choices.update(
+        str(record["setup"])
+        for record in cached_verified_records()
+        if record.get("verified") and record.get("setup")
+    )
     return sorted(choices, key=lambda setup: (setup_title(setup).lower(), setup.lower()))
 
 
@@ -1363,7 +1416,7 @@ def author_credit_html(author: str) -> str:
 
 def existing_author_choices() -> list[str]:
     seen: dict[str, str] = {}
-    for record in [*STORE.load_reference_records(), *STORE.load_records()]:
+    for record in [*cached_reference_records(), *cached_verified_records()]:
         name = display_author(record)
         normalized = name.strip()
         if not normalized or normalized.lower() in {"unknown", "anonymous"}:
@@ -1473,9 +1526,10 @@ def source_link(record: dict[str, Any]) -> str:
     return f'<a href="{esc(url)}" target="_blank" rel="noopener noreferrer">page</a>'
 
 
+@lru_cache(maxsize=1)
 def coordinate_records_by_case() -> dict[str, dict[str, Any]]:
     best: dict[str, dict[str, Any]] = {}
-    for record in STORE.load_records():
+    for record in cached_verified_records():
         if not record.get("verified") or not record.get("solution_path"):
             continue
         case = str(record.get("case") or "")
@@ -1690,10 +1744,10 @@ def record_card(record: dict[str, Any], coordinates: dict[str, dict[str, Any]])
 
 
 def setup_summary(setup: str) -> dict[str, Any]:
-    records = STORE.public_records(setup, show_all=False)
+    records = cached_public_records(setup, False)
     coordinate_count = sum(
         1
-        for record in STORE.load_records()
+        for record in cached_verified_records()
         if record.get("verified") and record.get("solution_path") and record.get("setup") == setup
     )
     n_values = [int(r.get("n", 0)) for r in records if r.get("n") is not None]
@@ -1942,7 +1996,7 @@ def browse_page_for_request(request: gr.Request):
     setup = str(query.get("family") or "").strip()
     if setup in setup_choices():
         return family_page_html(setup)
-    return directory_html()
+    return gr.update()
 
 
 def condensed_case_links(cases: list[str]) -> str:
@@ -1970,7 +2024,7 @@ def condensed_case_links(cases: list[str]) -> str:
 def leaderboard_html() -> str:
     rows: dict[str, dict[str, Any]] = {}
     coordinates = coordinate_records_by_case()
-    for record in STORE.public_records(show_all=False):
+    for record in cached_public_records("", False):
         author = display_author(record)
         if author.strip().lower() == "trivial":
             continue
@@ -2095,6 +2149,14 @@ def supported_gradio_kwargs(component: Any, **kwargs: Any) -> dict[str, Any]:
     return {key: value for key, value in kwargs.items() if key in parameters}
 
 
+def supported_call_kwargs(callable_obj: Any, **kwargs: Any) -> dict[str, Any]:
+    try:
+        parameters = inspect.signature(callable_obj).parameters
+    except (TypeError, ValueError):
+        return {}
+    return {key: value for key, value in kwargs.items() if key in parameters}
+
+
 def preview_html(solution: dict[str, Any]) -> str:
     return f"""
     <div class="preview-card">
@@ -2135,8 +2197,7 @@ def submit_solution(
         pretty = json.dumps(stored, indent=2, sort_keys=True)
         sync_status = record.get("sync_status", "dataset sync disabled")
         message = result_markdown(result) + f"\n\nSaved as record `{record['id']}`.\n\nDataset sync: `{sync_status}`."
-        family_page_html.cache_clear()
-        directory_html.cache_clear()
+        clear_page_caches()
         return (
             message,
             preview_html(stored),
@@ -2158,10 +2219,10 @@ def prewarm_pages() -> None:
         family_page_html(setup)
 
 
-with gr.Blocks(title="Packing Benchmark") as demo:
+with gr.Blocks(**supported_gradio_kwargs(gr.Blocks, title="Packing Benchmark", analytics_enabled=False)) as demo:
     with gr.Tabs():
         with gr.Tab("Home"):
-            browse_page = gr.HTML(loading_html())
+            browse_page = gr.HTML(directory_html())
 
         with gr.Tab("Leaderboard"):
             leaderboard_page = gr.HTML(leaderboard_html())
@@ -2245,4 +2306,11 @@ with gr.Blocks(title="Packing Benchmark") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(theme=THEME, css=CSS, js=APP_JS, server_name="0.0.0.0", ssr_mode=False)
+    demo.launch(
+        theme=THEME,
+        css=CSS,
+        js=APP_JS,
+        server_name="0.0.0.0",
+        ssr_mode=False,
+        **supported_call_kwargs(demo.launch, show_api=False),
+    )