Update app.py

app.py

@@ -12,7 +12,7 @@ from datasets import load_dataset
 # === Config ===
 VISITS_URL = os.getenv(
     "VISITS_URL",
-    "https://huggingface.co/datasets/19arjun89/ai_recruiting_agent_usage/resolve/main/usage/visits.jsonl",
+    "https://huggingface.co/datasets/19arjun89/ai_recruiting_agent_usage/resolve/main/usage/visits_enriched.jsonl",
 )
 
 # Optional: You can keep this env var, but this version uses Plotly Geo (no Mapbox needed)
@@ -32,11 +32,15 @@ def normalize_country_name(country: str | None) -> str | None:
     return c
 
 
-def country_name_to_iso3(country_name: str) -> str | None:
-    """Convert country name -> ISO3 for mapping."""
+def iso2_to_iso3(country_code: str | None) -> str | None:
+    if not country_code or not isinstance(country_code, str):
+        return None
+    c2 = country_code.strip().upper()
+    if len(c2) != 2:
+        return None
     try:
-        rec = pycountry.countries.search_fuzzy(country_name)[0]
-        return rec.alpha_3
+        rec = pycountry.countries.get(alpha_2=c2)
+        return rec.alpha_3 if rec else None
     except Exception:
         return None
 
@@ -74,14 +78,16 @@ def build_report():
 
     for row in load_rows_streaming():
         scanned += 1
-
-        country = normalize_country_name(row.get("country"))
+        event_type = str(row.get("event", "") or "").strip().lower()
+        if event_type == "session_start":
+            continue
+        country = normalize_country_name(row.get("final_country"))
         if not country:
             continue
 
         country_counts[country] += 1
 
-        iso3 = country_name_to_iso3(country)
+        iso3 = iso2_to_iso3(row.get("final_country_code"))
         if not iso3:
             continue