Update image_api.py

image_api.py

@@ -1,19 +1,35 @@
-from fastapi import APIRouter, HTTPException, Query
+from fastapi import APIRouter, HTTPException, Query, Security, Depends
+from fastapi.security import APIKeyHeader
 import requests
 import os
 from typing import List, Optional
 from pydantic import BaseModel
+import logging
 
 router = APIRouter()
 
+# Setup logging
+logger = logging.getLogger(__name__)
+
+# Setup API key authentication
+api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)
+CHAT_AUTH_KEY = os.environ.get("CHAT_AUTH_KEY")
+
 class UnsplashResponse(BaseModel):
     images: List[str]
 
+async def verify_api_key(api_key: str = Security(api_key_header)):
+    if api_key != CHAT_AUTH_KEY:
+        logger.warning("Invalid API key used")
+        raise HTTPException(status_code=403, detail="Could not validate credentials")
+    return api_key
+
 @router.get("/unsplash-images/", response_model=UnsplashResponse)
 async def get_unsplash_images(
     query: str,
     images_per_page: Optional[int] = Query(4, alias="per_page", ge=1, le=30),
-    page: Optional[int] = Query(1, ge=1)
+    page: Optional[int] = Query(1, ge=1),
+    api_key: str = Depends(verify_api_key)
 ):
     url = "https://api.unsplash.com/search/photos"
     params = {