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()