wip

app.py

@@ -157,6 +157,17 @@ def add_solution_cbk(
     )
 
 
+def check_valid_token(token: gr.OAuthToken | None):
+    logger.info("TOKEN %s", token)
+    if token is None:
+        return ""
+    try:
+        whoami(token.token)
+        return ""
+    except Exception:
+        return '<script>window.location.href = "/logout";</script>'
+
+
 def gate_submission(token: gr.OAuthToken | None):
     """
     @brief Toggles the visibility of the login box and submission panel based on the user's login status.
@@ -164,14 +175,14 @@ def gate_submission(token: gr.OAuthToken | None):
     logger.info("TOKEN %s", token)
     if token is None:
         logger.info("NO TOKEN")
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value="")
+        return gr.update(visible=True), gr.update(visible=False)
     try:
         whoami(token.token)
         logger.info("TOKEN IS VALID")
-        return gr.update(visible=False), gr.update(visible=True), gr.update(value="")
+        return gr.update(visible=False), gr.update(visible=True)
     except Exception:
         logger.info("TOKEN HAS EXPIRED")
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value='<script>window.location.href = "/logout";</script>')
+        return gr.update(visible=True), gr.update(visible=False)
 
 
 def get_theme():
@@ -295,12 +306,15 @@ with blocks:
                 elem_id="citation-block",
             )
 
+    # Trigger reload if token is expired
+    blocks.load(check_valid_token, outputs=reload_html)
+
     # UI refresh triggers latest data swap.
     # The work already happened in the background - refresh_leaderboard_data().
     blocks.load(lambda: leaderboard_df, inputs=[], outputs=[leaderboard_component])
 
     # On initial load (and after OAuth redirect), toggle the UI based on login status.
-    blocks.load(gate_submission, inputs=None, outputs=[login_box, submit_panel, reload_html])
+    blocks.load(gate_submission, inputs=None, outputs=[login_box, submit_panel])
 
 
 logger.info("Scheduler")