Spaces:
Running
Running
File size: 1,654 Bytes
a2df67a |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import gradio as gr
import numpy as np
import nrrd
import matplotlib.pyplot as plt
import io
from PIL import Image
# Set Matplotlib to use the 'Agg' backend
plt.switch_backend('Agg')
def load_nrrd(file_path):
data, _ = nrrd.read(file_path.name)
num_slices = data.shape[2]
return data, num_slices
def visualize_slice(file_path, slice_index):
data, num_slices = load_nrrd(file_path)
# Ensure the slice index is within the range of available slices
slice_index = min(max(0, slice_index), num_slices - 1)
slice_image = data[:, :, slice_index]
# Plot the slice
fig, ax = plt.subplots()
ax.imshow(slice_image, cmap='gray')
plt.axis('off')
# Convert matplotlib figure to PIL Image
buf = io.BytesIO()
fig.savefig(buf, format='png')
plt.close(fig)
buf.seek(0)
pil_img = Image.open(buf)
return pil_img
def update_slider(file_path):
_, num_slices = load_nrrd(file_path)
return gr.update(maximum=num_slices-1, value=0)
with gr.Blocks() as app:
gr.Markdown("## NRRD Slice Visualizer")
gr.Markdown("Upload an NRRD file and use the slider to select and visualize slices.")
file_input = gr.File(label="Upload NRRD File")
slider = gr.Slider(minimum=0, maximum=1, step=1, value=0, label="Slice Selector")
image_output = gr.Image(type="pil", label="Selected Slice")
file_input.change(fn=update_slider, inputs=file_input, outputs=slider)
file_input.change(fn=visualize_slice, inputs=[file_input, slider], outputs=image_output)
slider.change(fn=visualize_slice, inputs=[file_input, slider], outputs=image_output)
app.launch()
|