Update app.py

app.py

@@ -31,6 +31,7 @@ import os
 import tempfile
 import zipfile
 
+@st.cache_data
 def zoom_at(img, x, y, zoom):
     '''
     Zoom into an image at a specific location.
@@ -63,6 +64,7 @@ def zoom_at(img, x, y, zoom):
     img_cropped = img.crop((left, upper, right, lower))
     return img_cropped.resize((500, 500), Image.LANCZOS)
 
+@st.cache_data
 def apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness):
     '''
     Apply zoom and image enhancements to the input image.
@@ -93,8 +95,9 @@ def apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness):
     enhanced_contrast = ImageEnhance.Contrast(zoomed).enhance(contrast)
     enhanced_brightness = ImageEnhance.Brightness(enhanced_contrast).enhance(brightness)
     enhanced_sharpness = ImageEnhance.Sharpness(enhanced_brightness).enhance(sharpness)
-    return enhanced_sharpness.resize((500, 500), Image.LANCZOS)
+    return enhanced_sharpness
 
+@st.cache_data
 def create_zip(processed_img, description, params):
     '''
     Create a zip archive containing the processed image and annotations.
@@ -168,9 +171,9 @@ if uploaded_files:
     with image_col:
         st.subheader("Processed Image")
         if 'processed_img' in st.session_state:
-            st.image(st.session_state.processed_img, caption="Processed Image")
+            st.image(st.session_state.processed_img, use_column_width=True, caption="Processed Image")
         else:
-            st.image(img, caption="Processed Image")
+            st.image(img, use_column_width=True, caption="Processed Image")
 
     with controls_col:
         st.subheader("Image Controls")
@@ -186,9 +189,6 @@ if uploaded_files:
         if st.button("Apply Adjustments"):
             processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
             st.session_state.processed_img = processed_img
-        else:
-            processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
-            st.session_state.processed_img = processed_img
 
     # Display Original Image Below
     st.subheader("Original Image")