Correct display after switching series

Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>

Author: jantonguirao

monailabel/datastore/utils/convert.py

@@ -92,6 +92,20 @@ def send_response(datastore, result, output, background_tasks):
         return res_json
 
     if output == "image":
+        # Log NRRD metadata before sending response
+        try:
+            import nrrd
+            if res_img and os.path.exists(res_img) and (res_img.endswith('.nrrd') or res_img.endswith('.nrrd.gz')):
+                _, header = nrrd.read(res_img, index_order='C')
+                logger.info(f"[NRRD Geometry] File: {os.path.basename(res_img)}")
+                logger.info(f"[NRRD Geometry] Dimensions: {header.get('sizes')}")
+                logger.info(f"[NRRD Geometry] Space Origin: {header.get('space origin')}")
+                logger.info(f"[NRRD Geometry] Space Directions: {header.get('space directions')}")
+                logger.info(f"[NRRD Geometry] Space: {header.get('space')}")
+                logger.info(f"[NRRD Geometry] Type: {header.get('type')}")
+                logger.info(f"[NRRD Geometry] Encoding: {header.get('encoding')}")
+        except Exception as e:
+            logger.warning(f"Failed to read NRRD metadata: {e}")
         return FileResponse(res_img, media_type=get_mime_type(res_img), filename=os.path.basename(res_img))
 
     if output == "dicom_seg":

monailabel/endpoints/infer.py

@@ -141,6 +141,15 @@ def write_seg_nrrd(
         ]
     )
 
+    # Log NRRD geometry being written
+    logger.info(f"[NRRD Writer] Writing segmentation to: {output_file}")
+    logger.info(f"[NRRD Writer] Image shape: {image_np.shape}")
+    logger.info(f"[NRRD Writer] Affine matrix:\n{affine}")
+    logger.info(f"[NRRD Writer] Space origin: {origin}")
+    logger.info(f"[NRRD Writer] Space directions:\n{space_directions}")
+    logger.info(f"[NRRD Writer] Space: {space}")
+    logger.info(f"[NRRD Writer] Index order: {index_order}")
+
     header.update(
         {
             "kinds": kinds,

monailabel/transform/reader.py

@@ -14,23 +14,6 @@
 curr_dir="$(pwd)"
 my_dir="$(dirname "$(readlink -f "$0")")"
 
-# Load nvm and ensure Node.js 18 is available
-export NVM_DIR="$HOME/.nvm"
-if [ -s "$NVM_DIR/nvm.sh" ]; then
-  echo "Loading nvm..."
-  . "$NVM_DIR/nvm.sh"
-  nvm use 18 2>/dev/null || nvm install 18
-  echo "Using Node.js $(node --version)"
-else
-  echo "WARNING: nvm not found. Checking Node.js version..."
-  NODE_VERSION=$(node --version 2>/dev/null | cut -d'v' -f2 | cut -d'.' -f1)
-  if [ -z "$NODE_VERSION" ] || [ "$NODE_VERSION" -lt 18 ]; then
-    echo "ERROR: Node.js >= 18 is required. Current version: $(node --version 2>/dev/null || echo 'not installed')"
-    echo "Please install Node.js 18 or higher, or install nvm."
-    exit 1
-  fi
-fi
-
 echo "Installing requirements..."
 sh $my_dir/requirements.sh