Update usage_logging.py

usage_logging.py

@@ -278,7 +278,7 @@ def rebuild_visits_rollup_from_event_files() -> str:
     Rebuilds usage/visits.jsonl from immutable per-event JSON files in usage/events/.
     ALSO writes an enriched rollup usage/visits_enriched.jsonl where:
       - legacy rows (no session_id) keep their original country as final_country
-      - new click rows (with session_id) get final_country from the session's usage_start row when available
+      - new click rows (with session_id) get final_country from the session's session_start row when available
     """
     api = _hf_api()
     if not api:
@@ -321,11 +321,10 @@ def rebuild_visits_rollup_from_event_files() -> str:
     # 3) Sort by ts_utc
     events.sort(key=lambda e: (e.get("ts_utc") or ""))
 
-    # 4) Build session → geo map from usage_start events (new schema)
-    # NOTE: you said you are using "usage_start" (not "session_start")
+    # 4) Build session → geo map from session_start events (new schema)
     session_geo = {}
     for e in events:
-        if e.get("event") == "usage_start":
+        if e.get("event") == "session_start":
             sid = (e.get("session_id") or "").strip()
             if not sid:
                 continue  # legacy usage_start rows won’t have session_id; ignore for mapping