osanseviero HF staff commited on
Commit
c4c93d2
1 Parent(s): 83f048b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +10 -8
main.py CHANGED
@@ -103,12 +103,14 @@ def read_root():
103
 
104
  @app.post("/webhook")
105
  async def webhook(request):
106
- if request.method == "POST":
107
- if request.headers.get("X-Webhook-Secret") != KEY:
108
- return Response("Invalid secret", status_code=401)
109
- data = await request.json()
110
- print("Webhook received!")
111
- train()
112
- find_samples_to_label()
113
- return "Webhook received!" if result else result
 
 
114
 
 
103
 
104
  @app.post("/webhook")
105
  async def webhook(request):
106
+ print("Received request")
107
+ if request.headers.get("X-Webhook-Secret") is None:
108
+ return Response("No secret", status_code=401)
109
+ if request.headers.get("X-Webhook-Secret") != KEY:
110
+ return Response("Invalid secret", status_code=401)
111
+ data = await request.json()
112
+ print("Webhook received!")
113
+ train()
114
+ find_samples_to_label()
115
+ return "Webhook received!" if result else result
116