Merge pull request #131 from NHSDigital/load-env-for-modality-emulator

Load env before starting emulator

Author: steventux

src/modality_emulator.py

@@ -4,6 +4,7 @@
 import time
 
 import numpy as np
+from dotenv import load_dotenv
 from PIL import Image
 from pydicom.dataset import Dataset, FileMetaDataset
 from pydicom.uid import ExplicitVRLittleEndian, generate_uid
@@ -18,6 +19,8 @@
 from services.mwl import MWLStatus
 from services.storage import MWLStorage
 
+load_dotenv()
+
 logging.basicConfig(
     level=os.getenv("LOG_LEVEL", "INFO").upper(),
     format=os.getenv("LOG_FORMAT", "%(asctime)s - %(name)s - %(levelname)s - %(message)s"),
@@ -58,7 +61,7 @@ def generate_dicom(self) -> Dataset:
             logger.error("No dataset provided for DICOM generation")
             return ds
 
-        img_path = f"{SAMPLE_IMAGES_PATH}/L{self.view}.jpg"
+        img_path = f"{SAMPLE_IMAGES_PATH}/L{self.view.replace('ID', '')}.jpg"
         img = Image.open(img_path).convert("L")
         if self.laterality == "R":
             img = img.transpose(Image.Transpose.FLIP_LEFT_RIGHT)