add logging

main.py

@@ -11,6 +11,7 @@ from steps.preprocess import process_image
 from huggingface_hub import hf_hub_download
 from architecture.resnet import ResNet
 import torch
+import logging
 
 app = FastAPI()
 
@@ -22,8 +23,10 @@ VALID_API_KEYS = os.environ.get("api_key")
 
 @app.middleware("http")
 async def verify_api_key(request, call_next):
+    logging.info(f"Received request: {request.method} {request.url}")
     api_key = request.headers.get("x-api-key")
     if api_key is None or api_key not in VALID_API_KEYS:
+        logging.warning("Unauthorized access attempt.")
         raise HTTPException(status_code=403, detail="Unauthorized")
     response = await call_next(request)
     return response