Update app.py

app.py

@@ -1,5 +1,3 @@
-# Instale as dependências necessárias
-
 import os
 import tempfile
 import pydicom
@@ -20,6 +18,13 @@ def dicom_to_nifti_and_display(dicom_file):
         dicom_data = pydicom.dcmread(dicom_file.name)
         dicom_image = dicom_data.pixel_array
 
+        # Verificar se a imagem é 3D e extrair o slice central
+        if len(dicom_image.shape) == 3:
+            slice_index = dicom_image.shape[0] // 2  # Slice central ao longo do eixo Z
+            dicom_slice = dicom_image[slice_index, :, :]  # Extrair o slice
+        else:
+            dicom_slice = dicom_image  # Se for 2D, use a imagem diretamente
+
         # Criar um objeto NIfTI
         nifti_image = nib.Nifti1Image(dicom_image, affine=np.eye(4))
 
@@ -33,11 +38,18 @@ def dicom_to_nifti_and_display(dicom_file):
 
         # Exibir as imagens usando Matplotlib
         fig, axes = plt.subplots(1, 2, figsize=(10, 5))
-        axes[0].imshow(dicom_image, cmap="gray")
+        
+        # Imagem DICOM (slice central)
+        axes[0].imshow(dicom_slice, cmap="gray")
         axes[0].set_title("Imagem DICOM Original")
         axes[0].axis("off")
 
-        axes[1].imshow(nifti_loaded[:, :, nifti_loaded.shape[2] // 2], cmap="gray")  # Slice central do NIfTI
+        # Imagem NIfTI (slice central)
+        if len(nifti_loaded.shape) == 3:
+            nifti_slice = nifti_loaded[nifti_loaded.shape[2] // 2, :, :]  # Slice central ao longo do eixo Z
+        else:
+            nifti_slice = nifti_loaded
+        axes[1].imshow(nifti_slice, cmap="gray")
         axes[1].set_title("Imagem NIfTI Convertida")
         axes[1].axis("off")