Spaces:
Sleeping
Sleeping
File size: 905 Bytes
b40ec46 298a539 b40ec46 298a539 b40ec46 8ba74bb 298a539 d44ceac 298a539 18290d5 d44ceac a805477 18290d5 298a539 b40ec46 1b05017 b40ec46 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
import nibabel as nib
import pydicom
import os
def convert(input):
# if linux
path = input.name
print("Zip file path:", path)
assert path.endswith('.zip'), 'File must be a zip file'
os.system(f'unzip -q "{path}" -d "{path[:-4]}"')
path = path[:-4]
files = os.listdir(path)
if len(files) == 1 and os.path.isdir(os.path.join(path, files[0])):
path = os.path.join(path, files[0])
print("Dicom folder path:", path)
dicom_files = os.listdir(path)
assert [f.endswith('.dcm') for f in dicom_files], 'Dicom folder must contain only .dcm files'
print("This is the list of files in the folder:", os.listdir(path))
gr.Interface(
fn=convert,
inputs=gr.File(type="file", label="Dicom folder (.zip)", file_types=["zip"]),
outputs="file",
title="Convert a Dicom folder to a Nifti (.nii.gz) file",
).launch() |