diff --git a/src/aare/daq/aaredb.py b/src/aare/daq/aaredb.py index 460ba927..e165f7b1 100644 --- a/src/aare/daq/aaredb.py +++ b/src/aare/daq/aaredb.py @@ -185,7 +185,7 @@ class AareWrapper: except Exception as e: logger.error(f"Error sending message to db: {e}") - def upload_image(self, sample_id: int, filename: str, bgr_image: np.ndarray): + def upload_image(self, sample_id: int, filename: str, bgr_image: np.ndarray, message: str | None = None): _, buffer = cv2.imencode('.jpg', bgr_image) jpeg_bytes = io.BytesIO(buffer) url = f"{self.__host}/protected_router/sample_runner/{sample_id}/upload-images" @@ -193,9 +193,16 @@ class AareWrapper: "accept": "application/json", "X-Shared-Password": os.getenv("AAREDB_SHARED_PASSWORD") } - response = requests.post(url, - files={'uploaded_file': (filename + ".jpg", jpeg_bytes, "image/jpeg")}, - verify=False, headers=headers) + + request_kwargs = { + "files": {'uploaded_file': (filename + ".jpg", jpeg_bytes, "image/jpeg")}, + "verify": False, + "headers": headers, + } + if message is not None: + request_kwargs["data"] = {"comment": message} + + response = requests.post(url, **request_kwargs) logger.debug(f"Response status code: {response.status_code}") def upload_jpg(self, sample_id: int, filename: str, jpg_image):