Spaces:
Runtime error
Runtime error
Francisco Aranda
commited on
Commit
·
ea4e048
1
Parent(s):
a7c2cd2
refactor: Show queue elements one by one
Browse files
main.py
CHANGED
|
@@ -9,22 +9,16 @@ client = rg.Argilla()
|
|
| 9 |
server = rg.get_webhook_server()
|
| 10 |
incoming_events = Queue()
|
| 11 |
|
| 12 |
-
# Set up the webhook listeners
|
| 13 |
|
| 14 |
-
#
|
| 15 |
-
for webhook in client.webhooks:
|
| 16 |
-
webhook.delete()
|
| 17 |
|
| 18 |
# Create a webhook for record events
|
| 19 |
|
| 20 |
-
@rg.webhook_listener(
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
)
|
| 24 |
-
async def record_events(event:dict):
|
| 25 |
-
print("Received event", event)
|
| 26 |
|
| 27 |
-
incoming_events.put(
|
| 28 |
|
| 29 |
# Create a webhook for dataset events
|
| 30 |
@rg.webhook_listener(events=["dataset.created", "dataset.updated", "dataset.published"])
|
|
@@ -34,15 +28,14 @@ async def dataset_events(type: str, dataset: rg.Dataset | None = None, **kwargs)
|
|
| 34 |
incoming_events.put((type, dataset))
|
| 35 |
|
| 36 |
|
|
|
|
| 37 |
# Create a webhook for response events
|
| 38 |
-
@rg.webhook_listener(
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
print("Received event", event)
|
| 44 |
|
| 45 |
-
incoming_events.put(event)
|
| 46 |
|
| 47 |
@rg.webhook_listener(events=["record.deleted", "dataset.deleted", "response.deleted"])
|
| 48 |
async def deleted_events(type: str, data: dict, **kwargs):
|
|
@@ -50,24 +43,21 @@ async def deleted_events(type: str, data: dict, **kwargs):
|
|
| 50 |
|
| 51 |
incoming_events.put((type, data))
|
| 52 |
|
|
|
|
| 53 |
def check_incoming_events():
|
| 54 |
"""
|
| 55 |
This function is called every 5 seconds to check if there are any incoming
|
| 56 |
events and send data to update the JSON component.
|
| 57 |
"""
|
| 58 |
-
|
| 59 |
-
while not incoming_events.empty():
|
| 60 |
-
events.append(incoming_events.get())
|
| 61 |
-
|
| 62 |
-
return {"events": events}
|
| 63 |
|
|
|
|
| 64 |
|
| 65 |
with gr.Blocks() as demo:
|
| 66 |
argilla_server = client.http_client.base_url
|
| 67 |
gr.Markdown("## Argilla Events")
|
| 68 |
gr.Markdown(f"This demo shows the incoming events from the [Argilla Server]({argilla_server}).")
|
| 69 |
json_component = gr.JSON(label="Incoming argilla events:")
|
| 70 |
-
gr.Timer(
|
| 71 |
-
|
| 72 |
|
| 73 |
gr.mount_gradio_app(server, demo, path="/")
|
|
|
|
| 9 |
server = rg.get_webhook_server()
|
| 10 |
incoming_events = Queue()
|
| 11 |
|
|
|
|
| 12 |
|
| 13 |
+
# Set up the webhook listeners
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# Create a webhook for record events
|
| 16 |
|
| 17 |
+
@rg.webhook_listener(events=["record.created", "record.updated", "record.completed"])
|
| 18 |
+
async def record_events(record: rg.Record, type: str, **kwargs):
|
| 19 |
+
print("Received event", type)
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
+
incoming_events.put(record)
|
| 22 |
|
| 23 |
# Create a webhook for dataset events
|
| 24 |
@rg.webhook_listener(events=["dataset.created", "dataset.updated", "dataset.published"])
|
|
|
|
| 28 |
incoming_events.put((type, dataset))
|
| 29 |
|
| 30 |
|
| 31 |
+
|
| 32 |
# Create a webhook for response events
|
| 33 |
+
@rg.webhook_listener(events=["response.created", "response.updated"])
|
| 34 |
+
async def response_events(response: rg.UserResponse, type:str, **kwargs):
|
| 35 |
+
print("Received event type", type)
|
| 36 |
+
|
| 37 |
+
incoming_events.put(response)
|
|
|
|
| 38 |
|
|
|
|
| 39 |
|
| 40 |
@rg.webhook_listener(events=["record.deleted", "dataset.deleted", "response.deleted"])
|
| 41 |
async def deleted_events(type: str, data: dict, **kwargs):
|
|
|
|
| 43 |
|
| 44 |
incoming_events.put((type, data))
|
| 45 |
|
| 46 |
+
|
| 47 |
def check_incoming_events():
|
| 48 |
"""
|
| 49 |
This function is called every 5 seconds to check if there are any incoming
|
| 50 |
events and send data to update the JSON component.
|
| 51 |
"""
|
| 52 |
+
event = incoming_events.get()
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
+
return event
|
| 55 |
|
| 56 |
with gr.Blocks() as demo:
|
| 57 |
argilla_server = client.http_client.base_url
|
| 58 |
gr.Markdown("## Argilla Events")
|
| 59 |
gr.Markdown(f"This demo shows the incoming events from the [Argilla Server]({argilla_server}).")
|
| 60 |
json_component = gr.JSON(label="Incoming argilla events:")
|
| 61 |
+
gr.Timer(1, active=True).tick(check_incoming_events, outputs=json_component)
|
|
|
|
| 62 |
|
| 63 |
gr.mount_gradio_app(server, demo, path="/")
|