Update app.py

app.py

@@ -3,6 +3,17 @@ import numpy as np
 import streamlit as st
 import io
 
+# Resize image to prevent memory issues with large images
+def resize_image(img, max_size=(3000, 3000)):
+    """
+    Resize image to fit within max_size while maintaining aspect ratio.
+    :param img: PIL Image object
+    :param max_size: Maximum dimensions (width, height)
+    :return: Resized PIL Image object
+    """
+    img.thumbnail(max_size, Image.ANTIALIAS)
+    return img
+
 # Logistic map function
 def logistic_map(r, x):
     return r * x * (1 - x)
@@ -34,11 +45,16 @@ def main():
     if uploaded_file is not None:
         # Load the uploaded image
         input_image = Image.open(uploaded_file)
-        img_array = np.array(input_image)
+
+        # Resize the image if necessary
+        input_image = resize_image(input_image)
 
         # Display original image
         st.image(input_image, caption="Uploaded Image", use_container_width=True)
 
+        # Convert image to numpy array
+        img_array = np.array(input_image)
+
         # Key input
         key_seed = st.slider("Set the encryption key seed (0 < key < 1)", min_value=0.001, max_value=0.999, step=0.001)