add response media type information for openapi schema for image endpoints

Author: d-j-hatton

.pre-commit-config.yaml

@@ -26,9 +26,7 @@ repos:
         ]
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.8.6
+    rev: v0.14.14
     hooks:
-      - id: ruff
-        args: [--force-exclude]
+      - id: ruff-check
       - id: ruff-format
-        args: [--force-exclude]

src/smartem_backend/api_server.py

@@ -2366,7 +2366,7 @@ def complete(self):
     return [LatentRepresentationResponse(foilhole_uuid=k, x=v.x, y=v.y, index=v.index) for k, v in rep.items() if v]
 
 
-@app.get("/grids/{grid_uuid}/atlas_image")
+@app.get("/grids/{grid_uuid}/atlas_image", responses={200: {"content": {"image/png": {}}}})
 def get_grid_atlas_image(
     grid_uuid: str,
     x: int | None = None,
@@ -2393,7 +2393,7 @@ def get_grid_atlas_image(
     return Response(im_bytes, media_type="image/png")
 
 
-@app.get("/gridsquares/{gridsquare_uuid}/gridsquare_image")
+@app.get("/gridsquares/{gridsquare_uuid}/gridsquare_image", responses={200: {"content": {"image/png": {}}}})
 def get_gridsquare_image(
     gridsquare_uuid: str,
     db: SqlAlchemySession = DB_DEPENDENCY,