Update app.py

app.py

@@ -13,11 +13,88 @@ client = rg.Argilla(
 )
 
 # List of countries to check
-COUNTRIES = [
-    "MEX", "ARG", "COL", "CHL", "PER", "ESP", "BRA", 
-    "VEN", "ECU", "BOL", "PRY", "URY", "CRI", "PAN", 
-    "DOM", "GTM", "HND", "SLV", "NIC", "CUB"
-]
+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": "🇻🇪"
+  }
+}
 
 def count_answers_per_space(country: str):
     """
@@ -29,19 +106,11 @@ def count_answers_per_space(country: str):
     Returns:
         Dictionary with statistics about answers in the space
     """
-    # Convert country code to full country name
-    country_mapping = {
-        "MEX": "Mexico", "ARG": "Argentina", "COL": "Colombia", 
-        "CHL": "Chile", "PER": "Peru", "ESP": "Spain", 
-        "BRA": "Brazil", "VEN": "Venezuela", "ECU": "Ecuador", 
-        "BOL": "Bolivia", "PRY": "Paraguay", "URY": "Uruguay", 
-        "CRI": "Costa Rica", "PAN": "Panama", "DOM": "Dominican Republic", 
-        "GTM": "Guatemala", "HND": "Honduras", "SLV": "El Salvador", 
-        "NIC": "Nicaragua", "CUB": "Cuba"
-    }
+
+    iso = countries[country]["iso"]
+    emoji = countries[country]["emoji"]
     
-    full_country_name = country_mapping.get(country, country)
-    dataset_name = f"{full_country_name}_responder_preguntas"
+    dataset_name = f"{emoji} {country} - {iso} - Responder"
     
     try:
         dataset = client.datasets(dataset_name)
@@ -74,7 +143,7 @@ def count_answers_per_space(country: str):
         percentage_complete = (answered_questions / total_questions * 100) if total_questions > 0 else 0
         
         return {
-            "name": full_country_name,
+            "name": country,
             "total_questions": total_questions,
             "answered_questions": answered_questions,
             "total_answers": total_answers,
@@ -86,7 +155,7 @@ def count_answers_per_space(country: str):
         # If space doesn't exist, return zero values
         print(f"No dataset found for {dataset_name}: {e}")
         return {
-            "name": full_country_name,
+            "name": country,
             "total_questions": 0,
             "answered_questions": 0,
             "total_answers": 0,
@@ -102,8 +171,8 @@ app = FastAPI()
 async def serve_map():
     # Generate data for each country by querying Argilla
     country_data = {}
-    for country_code in COUNTRIES:
-        country_data[country_code] = count_answers_per_space(country_code)
+    for country in countries.keys():
+        country_data[country] = count_answers_per_space(country)
     
     # Convert to JSON for JavaScript
     country_data_json = json.dumps(country_data)