fix: run visit logging in background thread to prevent worker timeout

bde_prediction/visit_counter.py

@@ -17,6 +17,7 @@ import os
 import csv
 import io
 from datetime import datetime, timezone
+import threading
 
 try:
     from huggingface_hub import HfApi, hf_hub_download
@@ -35,8 +36,12 @@ SPACE_NAME     = "patonlab/alfabet_bde" # label stored in each row
 TOKEN          = os.environ.get("HF_TOKEN")  # set as a Space secret
 # ─────────────────────────────────────────────────────────────────────────────
 
-
 def log_visit(space: str = SPACE_NAME):
+    """Fire-and-forget visit logging in a background thread."""
+    thread = threading.Thread(target=_do_log_visit, args=(space,), daemon=True)
+    thread.start()
+
+def _do_log_visit(space: str):
     """Append one row to visits.csv in the analytics dataset repo."""
     if not TOKEN:
         print("[visit_counter] HF_TOKEN not set — skipping visit log.")
@@ -69,15 +74,17 @@ def log_visit(space: str = SPACE_NAME):
     writer.writerows(existing_rows)
 
     # 4. Push back to dataset repo
-    api.upload_file(
-        path_or_fileobj=buf.getvalue().encode(),
-        path_in_repo=ANALYTICS_FILE,
-        repo_id=ANALYTICS_REPO,
-        repo_type="dataset",
-        commit_message=f"visit: {space} @ {now}",
-    )
-    print(f"[visit_counter] Logged visit for {space} at {now}")
-
+    try:
+        api.upload_file(
+            path_or_fileobj=buf.getvalue().encode(),
+            path_in_repo=ANALYTICS_FILE,
+            repo_id=ANALYTICS_REPO,
+            repo_type="dataset",
+            commit_message=f"visit: {space} @ {now}",
+        )
+        print(f"[visit_counter] Logged visit for {space} at {now}")
+    except Exception as e:
+        print(f"[visit_counter] Warning: could not log visit: {e}")
 
 def get_visit_counts() -> dict:
     """Return a dict of {space_name: visit_count} from the analytics repo."""