updates app

app.py

@@ -1,14 +1,9 @@
-import asyncio
-import time
 import nh3
-from auth import APIKeyMiddleware
 from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse
-from starlette.status import HTTP_504_GATEWAY_TIMEOUT
+from fastapi import Request, HTTPException
 from neural_searcher import NeuralSearcher
 from huggingface_hub import login
 from config import HUGGING_FACE_API_KEY, COLLECTION_NAME, API_KEY
-from fastapi.security.api_key import APIKey
 
 login(HUGGING_FACE_API_KEY)
 
@@ -16,10 +11,6 @@ app = FastAPI()
 
 neural_searcher = NeuralSearcher(collection_name=COLLECTION_NAME)
 
-REQUEST_TIMEOUT_ERROR = 30
-
-app.add_middleware(APIKeyMiddleware)
-
 @app.get("/api/search")
 async def search(q: str):
     xss = nh3.clean(q)
@@ -27,18 +18,10 @@ async def search(q: str):
     return data
 
 
-# @app.middleware("http")
-# async def combined_middleware(request: Request, call_next):
-#     start_time = time.time()
-
-#     try:
-#         return await asyncio.wait_for(call_next(request), timeout=REQUEST_TIMEOUT_ERROR)
-#     except asyncio.TimeoutError:
-#         process_time = time.time() - start_time
-#         return JSONResponse(
-#             content={
-#                 "detail": "Request processing time exceeded limit",
-#                 "processing_time": process_time
-#             },
-#             status_code=HTTP_504_GATEWAY_TIMEOUT
-#         )
+@app.middleware("http")
+async def combined_middleware(request: Request, call_next):
+    api_key = request.headers.get("X-API-KEY")
+    if api_key != API_KEY:
+        raise HTTPException(status_code=403, detail="Forbidden")
+    response = await call_next(request)
+    return response