Update app.py

app.py

@@ -84,7 +84,8 @@ processed_mask = None
 def segment_button_click(file):
     global processed_image, processed_mask
     processed_image, processed_mask, output_path = process_file(file)
-    return "Segmentation completed! Use the slider to explore slices.", output_path
+    num_slices = processed_image.shape[2]
+    return "Segmentation completed! Use the slider to explore slices.", output_path, gr.update(visible=True, maximum=num_slices - 1)
 
 def update_visualization(slice_index):
     if processed_image is None or processed_mask is None:
@@ -107,11 +108,11 @@ with gr.Blocks() as iface:
     download_output = gr.File(label="Download Binary Mask (.tif)")
 
     with gr.Row():
-        slice_slider = gr.Slider(minimum=0, maximum=np.shape(processed_image)[2] if processed_image else 100, step=1, label="Slice Index", interactive=True)
+        slice_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Slice Index", interactive=True, visible=False)
         visualization_output = gr.Plot(label="2D Slice Visualization")
 
     # Button click triggers segmentation
-    segment_button.click(segment_button_click, inputs=file_input, outputs=[status_output, download_output])
+    segment_button.click(segment_button_click, inputs=file_input, outputs=[status_output, download_output, slice_slider])
 
     # Slider changes trigger visualization updates
     slice_slider.change(update_visualization, inputs=slice_slider, outputs=visualization_output)