Base DB Call updates

app.py

@@ -204,6 +204,15 @@ def init_leaderboard(dataframe):
     )
 
 
+def refresh_dashboard():
+    """Re-pulls leaderboard + top-3 data. Bound to demo.load() so that if
+    the module-import-time fetch raced ahead of a still-connecting Mongo
+    client on a cold start, the UI corrects itself once the page mounts."""
+    df = get_leaderboard_df(EVAL_RESULTS_PATH, "", COLS, BENCHMARK_COLS)
+    top3 = get_top_3_eval_cards(EVAL_RESULTS_PATH)
+    return df, build_top_3_cards_html(top3)
+
+
 custom_head_js = """
 <script>
 window.selectTopModel = function(modelName) {
@@ -296,6 +305,13 @@ with demo:
         outputs=[results_display_panel],
     )
 
+    # Page mount -> re-pull leaderboard + top-3 cards.
+    # Self-healing refresh: guards against a cold-start race where the
+    # module-level fetch above ran before the Mongo client finished
+    # connecting, which would otherwise freeze a stale/incomplete
+    # snapshot for every visitor until the container restarts.
+    demo.load(fn=refresh_dashboard, outputs=[leaderboard_table, top_cards_display])
+
     with gr.Row():
         with gr.Accordion("📙 Methodology & Limitations", open=False):
             gr.Textbox(value=CITATION_BUTTON_TEXT, label=CITATION_BUTTON_LABEL, lines=8, show_copy_button=True)
@@ -305,3 +321,5 @@ if __name__ == "__main__":
     demo.queue(default_concurrency_limit=5).launch()
 # if __name__ == "__main__":
 #     demo.launch(server_name="0.0.0.0", server_port=7860)
+# if __name__ == "__main__":
+#     demo.launch(server_name="0.0.0.0", server_port=7860)

src/db.py

@@ -1,5 +1,6 @@
 import os
 import re
+import time
 from typing import Dict, List, Optional
 
 try:
@@ -16,27 +17,37 @@ COLLECTION_NAME = "audit_certificates"
 _mongo_client = None
 
 
-def get_db_collection():
-    """Initializes and returns the MongoDB collection using HF Secrets."""
+def get_db_collection(retries: int = 3, delay: float = 2.0):
+    """Initializes and returns the MongoDB collection using HF Secrets.
+    Retries the initial connection — HF Space cold starts often need a
+    few extra seconds for DNS/TLS to Atlas before serverSelection succeeds.
+    """
     global _mongo_client
     if not HAS_PYMONGO:
         return None
     if not MONGODB_URI:
-        # Space secret MONGODB_URI not yet configured; gracefully fallback to disk cache
-        return None
-    try:
-        if _mongo_client is None:
-            _mongo_client = pymongo.MongoClient(
-                MONGODB_URI,
-                serverSelectionTimeoutMS=5000,
-                connectTimeoutMS=5000,
-            )
-        db = _mongo_client[DB_NAME]
-        return db[COLLECTION_NAME]
-    except Exception as e:
-        print(f"[MongoDB] Connection warning: {e}", flush=True)
         return None
 
+    for attempt in range(1, retries + 1):
+        try:
+            if _mongo_client is None:
+                _mongo_client = pymongo.MongoClient(
+                    MONGODB_URI,
+                    serverSelectionTimeoutMS=8000,
+                    connectTimeoutMS=8000,
+                )
+            # Force a real round-trip here (pymongo is lazy otherwise),
+            # so a slow cold-start connection is retried at this single
+            # choke point rather than surfacing as "0 certificates".
+            _mongo_client.admin.command("ping")
+            return _mongo_client[DB_NAME][COLLECTION_NAME]
+        except Exception as e:
+            print(f"[MongoDB] Connection attempt {attempt}/{retries} warning: {e}", flush=True)
+            _mongo_client = None
+            if attempt < retries:
+                time.sleep(delay)
+    return None
+
 
 def mongo_save_certificate(cert: Dict) -> bool:
     """Persists an audit certificate permanently to MongoDB Atlas."""