Update app.py

app.py

@@ -7,19 +7,93 @@ from collections import defaultdict
 from fastapi import FastAPI
 from functools import lru_cache
 
-# Initialize Argilla client with environment variables
 client = rg.Argilla(
     api_url=os.getenv("ARGILLA_API_URL", ""),
     api_key=os.getenv("ARGILLA_API_KEY", "")
 )
 
-# Dataset information - list all the datasets to track
-DATASETS = [
-    "🇪🇸 España - ESP - Responder",
-    # Add more datasets as needed
-]
-
-# Cache results to avoid frequent API calls
+countries = {
+  "Argentina": {
+    "iso": "ARG",
+    "emoji": "🇦🇷"
+  },
+  "Bolivia": {
+    "iso": "BOL",
+    "emoji": "🇧🇴"
+  },
+  "Chile": {
+    "iso": "CHL",
+    "emoji": "🇨🇱"
+  },
+  "Colombia": {
+    "iso": "COL",
+    "emoji": "🇨🇴"
+  },
+  "Costa Rica": {
+    "iso": "CRI",
+    "emoji": "🇨🇷"
+  },
+  "Cuba": {
+    "iso": "CUB",
+    "emoji": "🇨🇺"
+  },
+  "Ecuador": {
+    "iso": "ECU",
+    "emoji": "🇪🇨"
+  },
+  "El Salvador": {
+    "iso": "SLV",
+    "emoji": "🇸🇻"
+  },
+  "España": {
+    "iso": "ESP",
+    "emoji": "🇪🇸"
+  },
+  "Guatemala": {
+    "iso": "GTM",
+    "emoji": "🇬🇹"
+  },
+  "Honduras": {
+    "iso": "HND",
+    "emoji": "🇭🇳"
+  },
+  "México": {
+    "iso": "MEX",
+    "emoji": "🇲🇽"
+  },
+  "Nicaragua": {
+    "iso": "NIC",
+    "emoji": "🇳🇮"
+  },
+  "Panamá": {
+    "iso": "PAN",
+    "emoji": "🇵🇦"
+  },
+  "Paraguay": {
+    "iso": "PRY",
+    "emoji": "🇵🇾"
+  },
+  "Perú": {
+    "iso": "PER",
+    "emoji": "🇵🇪"
+  },
+  "Puerto Rico": {
+    "iso": "PRI",
+    "emoji": "🇵🇷"
+  },
+  "República Dominicana": {
+    "iso": "DOM",
+    "emoji": "🇩🇴"
+  },
+  "Uruguay": {
+    "iso": "URY",
+    "emoji": "🇺🇾"
+  },
+  "Venezuela": {
+    "iso": "VEN",
+    "emoji": "🇻🇪"
+  }
+}
 @lru_cache(maxsize=32)
 def get_user_contributions_cached(cache_buster: int):
     return get_user_contributions()
@@ -30,13 +104,16 @@ def get_user_contributions():
     user_id_to_username = {}
     
     # Process each dataset
-    for dataset_name in DATASETS:
+    for country in countries.keys():
+        iso = countries[country]["iso"]
+        emoji = countries[country]["emoji"]
+    
+        dataset_name = f"{emoji} {country} - {iso} - Responder"
         try:
             print(f"Processing dataset: {dataset_name}")
             dataset = client.datasets(dataset_name)
             records = list(dataset.records(with_responses=True))
             
-            # Track contributions per user in this dataset
             dataset_contributions = defaultdict(int)
             
             for record in records:
@@ -47,7 +124,6 @@ def get_user_contributions():
                             user_id = answer["user_id"]
                             dataset_contributions[user_id] += 1
                             
-                            # Get username if not already cached
                             if user_id not in user_id_to_username:
                                 try:
                                     user = client.users(id=user_id)
@@ -56,7 +132,6 @@ def get_user_contributions():
                                     print(f"Error getting username for {user_id}: {e}")
                                     user_id_to_username[user_id] = f"User-{user_id[:8]}"
             
-            # Add dataset contributions to overall user stats
             for user_id, count in dataset_contributions.items():
                 username = user_id_to_username.get(user_id, f"User-{user_id[:8]}")
                 user_contributions[user_id]["username"] = username