Create App.py

App.py

@@ -0,0 +1,77 @@
+# Instale as dependências necessárias
+!pip install pydicom nibabel numpy gradio matplotlib
+
+import os
+import tempfile
+import pydicom
+import nibabel as nib
+import numpy as np
+import matplotlib.pyplot as plt
+import gradio as gr
+
+def dicom_to_nifti_and_display(dicom_file):
+    """
+    Converte um arquivo DICOM para NIfTI e exibe as imagens antes e depois da conversão.
+    
+    :param dicom_file: Arquivo DICOM carregado pelo usuário.
+    :return: Tupla contendo a imagem DICOM original e a imagem NIfTI convertida.
+    """
+    try:
+        # Ler o arquivo DICOM
+        dicom_data = pydicom.dcmread(dicom_file.name)
+        dicom_image = dicom_data.pixel_array
+
+        # Criar um objeto NIfTI
+        nifti_image = nib.Nifti1Image(dicom_image, affine=np.eye(4))
+
+        # Salvar temporariamente o arquivo NIfTI
+        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as temp_nifti:
+            nib.save(nifti_image, temp_nifti.name)
+            nifti_path = temp_nifti.name
+
+        # Carregar o arquivo NIfTI para exibição
+        nifti_loaded = nib.load(nifti_path).get_fdata()
+
+        # Exibir as imagens usando Matplotlib
+        fig, axes = plt.subplots(1, 2, figsize=(10, 5))
+        axes[0].imshow(dicom_image, 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
+        axes[1].set_title("Imagem NIfTI Convertida")
+        axes[1].axis("off")
+
+        # Salvar a figura em um buffer temporário
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_image:
+            plt.savefig(temp_image.name, bbox_inches="tight")
+            plt.close(fig)
+            return temp_image.name, nifti_path
+
+    except Exception as e:
+        raise gr.Error(f"Erro ao processar o arquivo: {e}")
+
+# Interface Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Conversor DICOM para NIfTI com Visualização")
+    gr.Markdown("Faça upload de um arquivo DICOM, visualize a imagem original, converta para NIfTI e veja o resultado.")
+
+    with gr.Row():
+        input_dicom = gr.File(label="Upload de Arquivo DICOM")
+        output_images = gr.Image(label="Imagens DICOM e NIfTI")
+        output_nifti = gr.File(label="Download do Arquivo NIfTI")
+
+    convert_button = gr.Button("Converter e Visualizar")
+
+    def process_and_display(dicom_file):
+        image_path, nifti_path = dicom_to_nifti_and_display(dicom_file)
+        return image_path, nifti_path
+
+    convert_button.click(
+        process_and_display,
+        inputs=input_dicom,
+        outputs=[output_images, output_nifti]
+    )
+
+# Executar a interface
+demo.launch()