Update app.py

app.py

@@ -30,8 +30,9 @@ import io
 import os
 import tempfile
 import zipfile
+import cv2
+import numpy as np
 
-@st.cache_data
 def zoom_at(img, x, y, zoom):
     '''
     Zoom into an image at a specific location.
@@ -97,7 +98,30 @@ def apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness):
     enhanced_sharpness = ImageEnhance.Sharpness(enhanced_brightness).enhance(sharpness)
     return enhanced_sharpness
 
-@st.cache_data
+def apply_enhancements_cv(img, x, y, zoom, contrast, brightness, sharpness):
+    """
+    Use OpenCV for zoom and enhancements.
+    """
+    # Convert PIL to OpenCV format
+    img_cv = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+    h, w = img_cv.shape[:2]
+
+    # Zoom
+    zoom_half = int(zoom / 2)
+    left = max(x - w * zoom_half, 0)
+    top = max(y - h * zoom_half, 0)
+    right = min(x + w * zoom_half, w)
+    bottom = min(y + h * zoom_half, h)
+    cropped = img_cv[int(top):int(bottom), int(left):int(right)]
+    resized = cv2.resize(cropped, (500, 500), interpolation=cv2.INTER_LANCZOS4)
+
+    # Convert back to PIL for other enhancements
+    pil_img = Image.fromarray(cv2.cvtColor(resized, cv2.COLOR_BGR2RGB))
+    enhanced_contrast = ImageEnhance.Contrast(pil_img).enhance(contrast)
+    enhanced_brightness = ImageEnhance.Brightness(enhanced_contrast).enhance(brightness)
+    enhanced_sharpness = ImageEnhance.Sharpness(enhanced_brightness).enhance(sharpness)
+    return enhanced_sharpness
+
 def create_zip(processed_img, description, params):
     '''
     Create a zip archive containing the processed image and annotations.
@@ -171,9 +195,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")
@@ -192,7 +216,7 @@ if uploaded_files:
 
     # Display Original Image Below
     st.subheader("Original Image")
-    st.image(img, use_container_width=True, caption="Original Image")
+    st.image(img, use_column_width=True, caption="Original Image")
 
     # Save and Export Options
     st.markdown("---")