Fix IndexError by clamping slice indices to valid range

app.py

@@ -151,12 +151,15 @@ def create_slice_visualization(ct_data, seg_data, axis, slice_idx, alpha=0.5, sh
     
     # Get the slice based on axis
     if axis == "Axial":
+        slice_idx = max(0, min(slice_idx, ct_data.shape[2] - 1))
         ct_slice = ct_data[:, :, slice_idx]
         seg_slice = seg_data[:, :, slice_idx] if seg_data is not None else None
     elif axis == "Coronal":
+        slice_idx = max(0, min(slice_idx, ct_data.shape[1] - 1))
         ct_slice = ct_data[:, slice_idx, :]
         seg_slice = seg_data[:, slice_idx, :] if seg_data is not None else None
     else:  # Sagittal
+        slice_idx = max(0, min(slice_idx, ct_data.shape[0] - 1))
         ct_slice = ct_data[slice_idx, :, :]
         seg_slice = seg_data[slice_idx, :, :] if seg_data is not None else None