Update usage_logging.py

usage_logging.py

@@ -113,14 +113,29 @@ def _get_client_ip(request: gr.Request) -> str:
 
 
 def _country_lookup(ip: str) -> str:
+    token = os.environ.get("IPINFO_TOKEN")
+    if not token:
+        return "Unknown"
+
     try:
-        r = requests.get(f"https://ipapi.co/{ip}/json/", timeout=4)
+        url = f"https://ipinfo.io/{ip}/json?token={token}"
+        r = requests.get(url, timeout=4)
         if r.status_code == 200:
             data = r.json()
-            return (data.get("country_name") or "Unknown").strip()
+
+            # ipinfo returns country code like "US"
+            country_code = (data.get("country") or "").strip()
+            if country_code:
+                return country_code
+
+            # fallback to country name if present
+            country_name = (data.get("country_name") or "").strip()
+            if country_name:
+                return country_name
+
+        return "Unknown"
     except Exception:
-        pass
-    return "Unknown"
+        return "Unknown"
 
 
 def append_visit_to_dataset(country: str, city: str, event_type: str = "usage_start", **extra_fields):
@@ -182,7 +197,7 @@ def record_visit(request: gr.Request):
             country=country,
             city="",
             event_type="usage_start",
-            country_source="ipapi" if country != "Unknown" else "ipapi_unknown"
+            country_source="ipinfo" if country != "Unknown" else "ipinfo_unknown"
         )
         return