Spaces:
Sleeping
Sleeping
added endpoint
Browse files
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 |
-
|
9 |
|
10 |
-
api = HfApi(token=
|
11 |
-
login(
|
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}
|