Update app.py

app.py

@@ -17,7 +17,7 @@ def encrypt_image(image, key):
     h, w, c = image.shape
     size = h * w * c
     chaotic_sequence = logistic_map(size, key).reshape(h, w, c)
-    encrypted_image = (image + chaotic_sequence) % 256  # Element-wise encryption
+    encrypted_image = np.mod(image + chaotic_sequence, 256)  # Element-wise encryption
     return encrypted_image.astype(np.uint8)
 
 # Streamlit app
@@ -30,7 +30,7 @@ def main():
     if uploaded_file is not None:
         # Display the uploaded image
         input_image = Image.open(uploaded_file)
-        st.image(input_image, caption="Uploaded Image", use_column_width=True)
+        st.image(input_image, caption="Uploaded Image", use_container_width=True)
 
         # Convert the image to numpy array
         input_image_np = np.array(input_image)
@@ -43,7 +43,7 @@ def main():
             encrypted_image = Image.fromarray(encrypted_image_np)
 
             # Display the encrypted image
-            st.image(encrypted_image, caption="Encrypted Image", use_column_width=True)
+            st.image(encrypted_image, caption="Encrypted Image", use_container_width=True)
 
             # Save encrypted image to a buffer
             buffer = io.BytesIO()