chore: prepare space example

main.py

@@ -1,4 +1,5 @@
 import os
+from datetime import datetime
 from queue import Queue
 
 import argilla as rg
@@ -9,55 +10,53 @@ client = rg.Argilla()
 server = rg.get_webhook_server()
 incoming_events = Queue()
 
+# The default webhook server URL is set to http://127.0.0.1:8000
+# This value is important since the webhook server URL is used to create the webhook URLs
+# When the SPACE_HOST environment variable is set, the default webhook server URL is set to https://SPACE_HOST value
+# If the FastAPI app is running on a different URL, the WEBHOOK_SERVER_URL environment variable should be set
+default_webhook_server_url = os.getenv("WEBHOOK_SERVER_URL")
+if default_webhook_server_url is None and "SPACE_HOST" in os.environ:
+    # We set it to the default gradio server URL
+    os.environ["WEBHOOK_SERVER_URL"] = "http://127.0.0.1:7860"
+
 
 # Set up the webhook listeners
 
 # Create a webhook for record events
-
-@rg.webhook_listener(events=["record.created", "record.updated", "record.completed"])
-async def record_events(record: rg.Record, type: str, **kwargs):
-    print("Received event", type, record)
+@rg.webhook_listener(events=["record.created", "record.updated", "record.deleted", "record.completed"])
+async def record_events(record: rg.Record, type: str, timestamp: datetime):
+    print(f"Received event type {type} at {timestamp}: ", record)
 
     incoming_events.put(record)
 
 
 # Create a webhook for dataset events
-@rg.webhook_listener(events=["dataset.created", "dataset.updated", "dataset.published"])
-async def dataset_events(type: str, dataset: rg.Dataset, **kwargs):
-    print(f"Received event", type, dataset)
+@rg.webhook_listener(events=["dataset.created", "dataset.updated", "dataset.deleted", "dataset.published"])
+async def dataset_events(dataset: rg.Dataset, type: str, timestamp: datetime):
+    print(f"Received event type {type} at {timestamp}: ", dataset)
 
     incoming_events.put((type, dataset))
 
 
 # Create a webhook for response events
 @rg.webhook_listener(events=["response.created", "response.updated"])
-async def response_events(type:str, response: rg.UserResponse,  **kwargs):
-    print(f"Received event", type, response)
+async def response_events(response: rg.UserResponse, type: str, timestamp: datetime):
+    print(f"Received event type {type} at {timestamp}: ", response)
 
     incoming_events.put(response)
 
 
-@rg.webhook_listener(events=["record.deleted", "dataset.deleted", "response.deleted"])
-async def deleted_events(type: str, data: dict, **kwargs):
-    print(f"Received event {type} for resource {data}")
-
-    incoming_events.put((type, data))
-
-
-def check_incoming_events():
-    """
-    This function is called every 5 seconds to check if there are any incoming
-    events and send data to update the JSON component.
-    """
+def read_next_event():
     event = incoming_events.get()
 
     return event
 
+
 with gr.Blocks() as demo:
     argilla_server = client.http_client.base_url
     gr.Markdown("## Argilla Events")
     gr.Markdown(f"This demo shows the incoming events from the [Argilla Server]({argilla_server}).")
-    json_component = gr.JSON(label="Incoming argilla events:")
-    gr.Timer(1, active=True).tick(check_incoming_events, outputs=json_component)
+    json_component = gr.JSON(label="Incoming argilla events:", value={})
+    gr.Timer(1, active=True).tick(read_next_event, outputs=json_component)
 
 gr.mount_gradio_app(server, demo, path="/")

start.sh

@@ -1 +1,2 @@
-WEBHOOK_SERVER_URL="https://$SPACE_HOST" uvicorn main:server --host 0.0.0.0 --port 7860
+uvicorn main:server --host 0.0.0.0 --port 7860
+