Spaces:
Runtime error
Runtime error
Francisco Aranda
commited on
Commit
·
36e4ada
1
Parent(s):
c631621
Add application files
Browse files- app.py +74 -0
- requirements.txt +1 -0
app.py
ADDED
@@ -0,0 +1,74 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from queue import Queue
|
2 |
+
|
3 |
+
import argilla as rg
|
4 |
+
import gradio as gr
|
5 |
+
|
6 |
+
client = rg.Argilla()
|
7 |
+
|
8 |
+
incoming_events = Queue()
|
9 |
+
|
10 |
+
def check_incoming_events():
|
11 |
+
"""
|
12 |
+
This function is called every 5 seconds to check if there are any incoming
|
13 |
+
events and send data to update the JSON component.
|
14 |
+
"""
|
15 |
+
events = []
|
16 |
+
while not incoming_events.empty():
|
17 |
+
events.append(incoming_events.get())
|
18 |
+
|
19 |
+
return {"events": events}
|
20 |
+
|
21 |
+
|
22 |
+
with gr.Blocks() as demo:
|
23 |
+
argilla_server = client.http_client.base_url
|
24 |
+
gr.Markdown("## Argilla Events")
|
25 |
+
gr.Markdown(f"This demo shows the incoming events from the [Argilla Server]({argilla_server}).")
|
26 |
+
json_component = gr.JSON(label="Incoming argilla events:")
|
27 |
+
gr.Timer(5, active=True).tick(check_incoming_events, outputs=json_component)
|
28 |
+
|
29 |
+
|
30 |
+
server, _, _ = demo.launch(prevent_thread_lock=True, app_kwargs={"docs_url": "/docs"})
|
31 |
+
|
32 |
+
# Set up the webhook listeners
|
33 |
+
rg.set_webhook_server(server)
|
34 |
+
|
35 |
+
# Delete all existing webhooks
|
36 |
+
for webhook in client.webhooks:
|
37 |
+
webhook.delete()
|
38 |
+
|
39 |
+
# Create a webhook for record events
|
40 |
+
|
41 |
+
@rg.webhook_listener(
|
42 |
+
events=["record.created", "record.updated", "record.completed"],
|
43 |
+
raw_event=True # Using raw events until PR https://github.com/argilla-io/argilla/pull/5500 is merged
|
44 |
+
)
|
45 |
+
async def record_events(event:dict):
|
46 |
+
print("Received event", event)
|
47 |
+
|
48 |
+
incoming_events.put(event)
|
49 |
+
|
50 |
+
# Create a webhook for dataset events
|
51 |
+
@rg.webhook_listener(events=["dataset.created", "dataset.updated", "dataset.published"])
|
52 |
+
async def dataset_events(type: str, dataset: rg.Dataset | None = None, **kwargs):
|
53 |
+
print(f"Received event {type} for dataset {dataset.id}")
|
54 |
+
|
55 |
+
incoming_events.put((type, dataset))
|
56 |
+
|
57 |
+
|
58 |
+
# Create a webhook for response events
|
59 |
+
@rg.webhook_listener(
|
60 |
+
events=["response.created", "response.updated"],
|
61 |
+
raw_event=True # Using raw events until PR https://github.com/argilla-io/argilla/pull/5500 is merged
|
62 |
+
)
|
63 |
+
async def response_events(event: dict):
|
64 |
+
print("Received event", event)
|
65 |
+
|
66 |
+
incoming_events.put(event)
|
67 |
+
|
68 |
+
@rg.webhook_listener(events=["record.deleted", "dataset.deleted", "response.deleted"])
|
69 |
+
async def deleted_events(type: str, data: dict, **kwargs):
|
70 |
+
print(f"Received event {type} for resource {data}")
|
71 |
+
|
72 |
+
incoming_events.put((type, data))
|
73 |
+
|
74 |
+
demo.block_thread()
|
requirements.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
argilla @ git+https://github.com/argilla-io/argilla.git@feat/argilla/working-with-webhooks#subdirectory=argilla
|