import nibabel as nib

import matplotlib.pyplot as plt

def load_nifti(file_path):

        '''

        Load the NIfTI image and access the image data as a Numpy array.

        Args:

            file_path ('str'): Path to the NIfTI file.

        Returns:

            data_array ('np.array'): Numpy array representing the image data.

            nii_image: Loaded NIfTI image object.

        '''

        nii_image = nib.load(file_path)

        data_array = nii_image.get_fdata()

        return data_array, nii_image

def show_nifti(file_data, title, slice=25):

    '''

    Display a single slice from the NIfTI volume.

    Args:

        file_data ('np.array'): Numpy array representing the image data.

        title ('str'): Title for the plot.

        slice ('int'): Slice index to display.

    '''        

    plt.imshow(file_data[slice, :, :], cmap='gray')

    plt.title(title)

    # plt.colorbar()

    plt.axis('off')

    plt.show()