Update app.py

app.py

@@ -20,7 +20,6 @@ def dicom_to_nifti_and_display(dicom_file):
 
         # Extrair o slice central da imagem DICOM
         if len(dicom_image.shape) == 3:
-            # 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:
@@ -57,21 +56,21 @@ def dicom_to_nifti_and_display(dicom_file):
             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))
+        fig, axes = plt.subplots(1, 2, figsize=(12, 6), constrained_layout=True)
         
         # Imagem DICOM (slice central)
-        axes[0].imshow(dicom_slice, cmap="gray")
-        axes[0].set_title("Imagem DICOM Original")
+        axes[0].imshow(dicom_slice, cmap="gray", aspect="auto")
+        axes[0].set_title("Imagem DICOM Original", fontsize=14, color="blue")
         axes[0].axis("off")
 
         # Imagem NIfTI (slice central)
-        axes[1].imshow(nifti_slice, cmap="gray")
-        axes[1].set_title("Imagem NIfTI Convertida")
+        axes[1].imshow(nifti_slice, cmap="gray", aspect="auto")
+        axes[1].set_title("Imagem NIfTI Convertida", fontsize=14, color="green")
         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.savefig(temp_image.name, bbox_inches="tight", dpi=300)  # Alta resolução
             plt.close(fig)
             return temp_image.name, nifti_path
 
@@ -85,7 +84,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         input_dicom = gr.File(label="Upload de Arquivo DICOM")
-        output_images = gr.Image(label="Imagens DICOM e NIfTI")
+        output_images = gr.Image(label="Visualização das Imagens", interactive=False)
         output_nifti = gr.File(label="Download do Arquivo NIfTI")
 
     convert_button = gr.Button("Converter e Visualizar")