Update utils.py

utils.py

@@ -9,20 +9,29 @@ from reportlab.lib.units import inch
 
 def load_dicom(filepath):
     """
-    Reads a DICOM file and returns a PIL Image.
+    Accepts a list of DICOM file objects.
+    Returns a list of PIL Images, one per sequence.
     """
-    dicom = pydicom.dcmread(filepath)
+
+    images=[]
+    for file in filepaths:
+        dicom = pydicom.dcmread(filepath)
     
-    # Extract the pixel array
-    pixel_array = dicom.pixel_array.astype(float)
+        # Extract the pixel array
+        pixel_array = dicom.pixel_array.astype(float)
     
-    # Normalize to 0-255 range
-    pixel_min = pixel_array.min()
-    pixel_max = pixel_array.max()
-    normalized = (pixel_array - pixel_min) / (pixel_max - pixel_min) * 255
+        # Normalize to 0-255 range
+        pixel_min = pixel_array.min()
+        pixel_max = pixel_array.max()
+
+        if pixel_max - pixel_min == 0:
+            continue #to handle sequences not added
+            
+        normalized = (pixel_array - pixel_min) / (pixel_max - pixel_min) * 255
     
-    # Convert to uint8 RGB image
-    image = Image.fromarray(normalized.astype(np.uint8)).convert("RGB")
+        # Convert to uint8 RGB image
+        image = Image.fromarray(normalized.astype(np.uint8)).convert("RGB")
+        images.append(image)
     
     return image