Update app.py

app.py

@@ -4,19 +4,96 @@ import gradio as gr
 import argilla as rg
 from argilla.webhooks import webhook_listener
 
-# Initialize Argilla client
 client = rg.Argilla(
     api_url=os.getenv("ARGILLA_API_URL"),
     api_key=os.getenv("ARGILLA_API_KEY"),
 )
-
-# Get the webhook server
 server = rg.get_webhook_server()
-
-# Queue to store events for display
 incoming_events = Queue()
 
-# Set up the webhook listener for response creation
+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": "🇻🇪"
+  }
+}
+
 @webhook_listener(events=["response.created"])
 async def update_validation_space_on_answer(response, type, timestamp):
     """
@@ -24,53 +101,50 @@ async def update_validation_space_on_answer(response, type, timestamp):
     It will automatically update the corresponding validation space with the new response.
     """
     try:
-        # Store the event for display in the UI
         incoming_events.put({"event": type, "timestamp": str(timestamp)})
         
-        # Get the record from the response
         record = response.record
         
-        # Check if this is from an answering space
         dataset_name = record.dataset.name
-        if not dataset_name.endswith("_responder_preguntas"):
+        if not dataset_name.endswith("Responder"):
             print(f"Ignoring event from non-answering dataset: {dataset_name}")
             return  # Not an answering space, ignore
         
-        # Extract the country from the dataset name
-        country = dataset_name.replace("_responder_preguntas", "")
+        country = " ".join(dataset_name.split("-").split(" ")[1:])
         print(f"Processing response for country: {country}")
-        
-        # Connect to the validation space
-        validation_dataset_name = f"{country}_validar_respuestas"
+
+        iso = countries[country]["iso"]
+        emoji = countries[country]["emoji"]
+        validation_dataset_name = f"{emoji} {country} - {iso} - Responder"
         try:
             validation_dataset = client.datasets(validation_dataset_name)
             print(f"Found validation dataset: {validation_dataset_name}")
         except Exception as e:
             print(f"Error connecting to validation dataset: {e}")
-            # You would need to import the create_validation_space function
-            from build_space import create_validation_space
-            validation_dataset = create_validation_space(country)
 
         response_dict = response.to_dict()
-        answer = response_dict["values"]['text']['value']
+        answer_1 = response_dict["values"]['answer_1']['value']
+        answer_2 = response_dict["values"]['answer_2']['value']
+        answer_3 = response_dict["values"]['answer_3']['value']
 
-        # Get the user ID of the original responder
         original_user_id = str(response.user_id)
-        
-        # Create a validation record with the correct attribute
-        # Instead of response.value, we need to find the correct attribute
-        # Based on the error, let's try to get the value differently
-        validation_record = {
-            "question": record.fields["question"],
-            "answer": answer,
-            "metadata": {
-                "original_responder_id": original_user_id,
-                "original_dataset": dataset_name
-            }
-        }
+
+        new_records = []
+
+        for answer in [answer_1, answer_2, answer_3]:
+            if answer:
+                validation_record = {
+                    "question": record.fields["question"],
+                    "answer": answer,
+                    "metadata": {
+                        "original_responder_id": original_user_id,
+                        "original_dataset": dataset_name
+                    }
+                }
+                new_records.append(answer)
         
         # Add the record to the validation space
-        validation_dataset.records.log(records=[validation_record])
+        validation_dataset.records.log(records=new_records)
         print(f"Added new response to validation space for {country}")
         
     except Exception as e: