Spaces:
Runtime error

dzenzzz commited on
Commit
a887dc5
·
1 Parent(s): 579610e

updates app

Browse files
Files changed (1) hide show
  1. app.py +7 -11
app.py CHANGED
@@ -22,15 +22,11 @@ async def search(q: str):
22
  @app.middleware("http")
23
  async def api_key_authentication(request: Request, call_next):
24
  api_key = request.headers.get("X-API-KEY")
 
25
  if api_key != API_KEY:
26
- raise HTTPException(status_code=401, detail="Unauthorized: Invalid API Key")
27
-
28
- response = await call_next(request)
29
- return response
30
-
31
- @app.exception_handler(HTTPException)
32
- async def http_exception_handler(request: Request, exc: HTTPException):
33
- return JSONResponse(
34
- status_code=exc.status_code,
35
- content={"message": exc.detail},
36
- )
 
22
  @app.middleware("http")
23
  async def api_key_authentication(request: Request, call_next):
24
  api_key = request.headers.get("X-API-KEY")
25
+
26
  if api_key != API_KEY:
27
+ return JSONResponse(
28
+ status_code=403,
29
+ content={"message": "Forbidden."}
30
+ )
31
+
32
+ return await call_next(request)