updates app

app.py

@@ -5,7 +5,7 @@ from fastapi.responses import JSONResponse
 from starlette.status import HTTP_504_GATEWAY_TIMEOUT
 from neural_searcher import NeuralSearcher
 from huggingface_hub import login
-from config import HUGGING_FACE_API_KEY,COLLECTION_NAME, API_KEY
+from config import HUGGING_FACE_API_KEY, COLLECTION_NAME, API_KEY
 
 login(HUGGING_FACE_API_KEY)
 
@@ -14,32 +14,31 @@ app = FastAPI()
 neural_searcher = NeuralSearcher(collection_name=COLLECTION_NAME)
 
 REQUEST_TIMEOUT_ERROR = 30
-
-ALLOWED_API_KEY = API_KEY 
+ALLOWED_API_KEY = API_KEY
 
 
 @app.get("/api/search")
 async def search(q: str):
-    data =  await neural_searcher.search(text=q)
+    data = await neural_searcher.search(text=q)
     return data
 
+
 @app.middleware("http")
-async def security_middleware(request: Request, call_next):
-    api_key = request.headers.get("X-API-KEY", "")
+async def combined_middleware(request: Request, call_next):
+    start_time = time.time()
 
+    api_key = request.headers.get("X-API-KEY", "")
     if api_key != ALLOWED_API_KEY:
-        raise HTTPException(status_code=403, detail="Access denied.")
+        return JSONResponse(status_code=403, content={"detail": "Access denied."})
 
-    return await call_next(request)
-
-@app.middleware("http")
-async def timeout_middleware(request: Request, call_next):
     try:
-        start_time = time.time()
         return await asyncio.wait_for(call_next(request), timeout=REQUEST_TIMEOUT_ERROR)
-
     except asyncio.TimeoutError:
         process_time = time.time() - start_time
-        return JSONResponse({'detail': 'Request processing time excedeed limit',
-                             'processing_time': process_time},
-                            status_code=HTTP_504_GATEWAY_TIMEOUT)
+        return JSONResponse(
+            content={
+                "detail": "Request processing time exceeded limit",
+                "processing_time": process_time
+            },
+            status_code=HTTP_504_GATEWAY_TIMEOUT
+        )