Fix add_images endpoint

api/routes/flights.py

@@ -1,6 +1,6 @@
 import logging
 from datetime import datetime
-from typing import Any
+from typing import Any, List
 
 from fastapi import APIRouter, HTTPException, Depends, Form, UploadFile, File
 
@@ -130,8 +130,8 @@ async def add_flight(flight_body: FlightSchema, user: UserDisplaySchema = Depend
     return {"id": str(flight)}
 
 
-@router.post('/{flight_id}/add_images', summary="Add images to a flight log")
-async def add_images(log_id: str, images: list[UploadFile] = File(...),
+@router.post('/{log_id}/add_images', summary="Add images to a flight log")
+async def add_images(log_id: str, images: List[UploadFile] = File(...),
                      user: UserDisplaySchema = Depends(get_current_user)):
     """
     Add images to a flight logbook entry

api/routes/img.py

@@ -24,15 +24,12 @@ async def get_image(user: UserDisplaySchema = Depends(get_current_user),
     :param image_id: ID of image to retrieve
     :return: Stream associated with requested image
     """
-    stream, user_created = await img.retrieve_image(image_id)
+    stream, user_created, media_type = await img.retrieve_image(image_id)
 
     if not user.id == user_created and not user.level == AuthLevel.ADMIN:
         raise HTTPException(403, "Access denied")
 
-    file_extension = os.path.splitext(image_id)[1]
-    media_type = mimetypes.types_map.get(file_extension)
-
-    return StreamingResponse(stream, media_type=media_type)
+    return StreamingResponse(stream, headers={'Content-Type': media_type})
 
 
 @router.post("/upload", description="Upload an image to the database")