a-guarino commited on
Commit
ee477df
·
1 Parent(s): ea38d67

added endpoint

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -1,16 +1,30 @@
1
  import os
2
 
3
- from fastapi import FastAPI
4
  from huggingface_hub import HfApi, login
5
 
6
  app = FastAPI()
7
  KEY = os.environ.get("WEBHOOK_SECRET")
8
- HF_TOKEN = os.environ.get("HF_TOKEN")
9
 
10
- api = HfApi(token=HF_TOKEN)
11
- login(HF_TOKEN)
12
 
13
 
14
  @app.get("/")
15
  def greet_json():
16
  return {"Hello": "World!"}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
 
3
+ from fastapi import FastAPI, HTTPException, Request
4
  from huggingface_hub import HfApi, login
5
 
6
  app = FastAPI()
7
  KEY = os.environ.get("WEBHOOK_SECRET")
8
+ HF_READ = os.environ.get("HF_READ")
9
 
10
+ api = HfApi(token=HF_READ)
11
+ login(HF_READ)
12
 
13
 
14
  @app.get("/")
15
  def greet_json():
16
  return {"Hello": "World!"}
17
+
18
+
19
+ @app.post("/benchmark")
20
+ async def handle_webhook(request: Request):
21
+ secret = request.headers.get("X-Webhook-Secret")
22
+ if secret != KEY:
23
+ raise HTTPException(status_code=403, detail="Invalid webhook secret")
24
+
25
+ try:
26
+ payload = await request.json()
27
+ except Exception:
28
+ raise HTTPException(status_code=400, detail="Invalid JSON payload")
29
+
30
+ return {"status": "success", "received": payload}