Update app.py

app.py

@@ -27,6 +27,13 @@ st.markdown("""
     html, body, .stApp {
         overflow-x: hidden !important;
     }
+    /* Force scrollbar always visible — this is the primary fix for page jitter */
+    html {
+        overflow-y: scroll !important;
+    }
+    body {
+        overflow-y: scroll !important;
+    }
     img {
         max-width: 100% !important;
         height: auto !important;
@@ -121,6 +128,7 @@ def get_image_from_plot(fig):
     buf.seek(0)
     return Image.open(buf)
 
+@st.cache_data
 def draw_kspace_diagram():
     fig, ax = plt.subplots(figsize=(6, 6))
     
@@ -156,6 +164,7 @@ def draw_kspace_diagram():
     fig.subplots_adjust(left=0.1, right=0.9, top=0.9, bottom=0.1)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def draw_kspace_point(kx, ky, bg_image):
     fig, ax = plt.subplots(figsize=(4, 4))
     ax.imshow(bg_image, cmap='gray', extent=[-112, 112, -112, 112], vmin=0, vmax=1)
@@ -169,6 +178,7 @@ def draw_kspace_point(kx, ky, bg_image):
     fig.subplots_adjust(left=0, right=1, top=1, bottom=0)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def draw_wave(kx, ky):
     fig, ax = plt.subplots(figsize=(4, 4))
     x = np.linspace(-112, 112, 224)
@@ -182,6 +192,7 @@ def draw_wave(kx, ky):
     fig.subplots_adjust(left=0, right=1, top=1, bottom=0)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def apply_filter(k_data, mode, radius):
     Y, X = np.ogrid[:224, :224]
     dist = np.sqrt((X - 112)**2 + (Y - 112)**2)
@@ -194,6 +205,7 @@ def apply_filter(k_data, mode, radius):
     img = np.fft.fftshift(np.fft.ifft2(np.fft.ifftshift(filtered_k)))
     return filtered_k, np.abs(img)
 
+@st.cache_data
 def draw_filtered_kspace(filtered_k):
     fig, ax = plt.subplots(figsize=(4, 4))
     ax.imshow(format_kspace_display(filtered_k), cmap='gray', vmin=0, vmax=1)
@@ -201,6 +213,7 @@ def draw_filtered_kspace(filtered_k):
     fig.subplots_adjust(left=0, right=1, top=1, bottom=0)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def draw_mri(mri_result):
     fig, ax = plt.subplots(figsize=(4, 4))
     if np.max(mri_result) > 0:
@@ -213,6 +226,7 @@ def draw_mri(mri_result):
     fig.subplots_adjust(left=0, right=1, top=1, bottom=0)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def draw_pulse_sequence(current_step, total_steps):
     fig, axes = plt.subplots(4, 1, figsize=(5, 6), sharex=True, gridspec_kw={'height_ratios': [1, 1.5, 1, 1]})
     t = np.linspace(0, 10, 1000)
@@ -262,6 +276,7 @@ def draw_pulse_sequence(current_step, total_steps):
     fig.subplots_adjust(left=0.2, right=0.95, top=0.95, bottom=0.05, hspace=0.2)
     return get_image_from_plot(fig)
 
+@st.cache_data
 def draw_kspace_filling(current_step, total_steps):
     fig, ax = plt.subplots(figsize=(5, 6))