diff --git a/src/modality_emulator.py b/src/modality_emulator.py index 3d6cd678..1517a1fa 100644 --- a/src/modality_emulator.py +++ b/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)