Update app.py

app.py

@@ -18,10 +18,17 @@ 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
+        # Extrair o slice central da imagem DICOM
         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
+            # Identificar o maior eixo e extrair o slice central
+            largest_axis = np.argmax(dicom_image.shape)
+            slice_index = dicom_image.shape[largest_axis] // 2
+            if largest_axis == 0:
+                dicom_slice = dicom_image[slice_index, :, :]
+            elif largest_axis == 1:
+                dicom_slice = dicom_image[:, slice_index, :]
+            else:
+                dicom_slice = dicom_image[:, :, slice_index]
         else:
             dicom_slice = dicom_image  # Se for 2D, use a imagem diretamente
 
@@ -36,6 +43,19 @@ def dicom_to_nifti_and_display(dicom_file):
         # Carregar o arquivo NIfTI para exibição
         nifti_loaded = nib.load(nifti_path).get_fdata()
 
+        # Extrair o slice central da imagem NIfTI
+        if len(nifti_loaded.shape) == 3:
+            largest_axis = np.argmax(nifti_loaded.shape)
+            slice_index = nifti_loaded.shape[largest_axis] // 2
+            if largest_axis == 0:
+                nifti_slice = nifti_loaded[slice_index, :, :]
+            elif largest_axis == 1:
+                nifti_slice = nifti_loaded[:, slice_index, :]
+            else:
+                nifti_slice = nifti_loaded[:, :, slice_index]
+        else:
+            nifti_slice = nifti_loaded  # Se for 2D, use a imagem diretamente
+
         # Exibir as imagens usando Matplotlib
         fig, axes = plt.subplots(1, 2, figsize=(10, 5))
         
@@ -45,10 +65,6 @@ def dicom_to_nifti_and_display(dicom_file):
         axes[0].axis("off")
 
         # 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")