From 989624b319d7f160f6807de881d3b4fe407382da Mon Sep 17 00:00:00 2001
From: appleb_m <martin.appleby@psi.ch>
Date: Mon, 9 Mar 2026 17:14:39 +0100
Subject: [PATCH] AareDB: added comments/messages for upload_image method

---
 src/aare/daq/aaredb.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

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):