Update app.py

app.py

@@ -1,48 +1,53 @@
-import streamlit as st
-import numpy as np
 from PIL import Image
+import numpy as np
+import streamlit as st
 import io
 
 # Logistic map function
-def logistic_map(size, key):
-    x = key
-    sequence = []
-    for _ in range(size):
-        x = 4 * x * (1 - x)  # Logistic map equation
-        sequence.append(int(x * 255) % 256)  # Map to 0-255
-    return np.array(sequence, dtype=np.uint8)
-
-# Encrypt image using logistic map
-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 = np.mod(image + chaotic_sequence, 256)  # Element-wise encryption
-    return encrypted_image.astype(np.uint8)
-
-# Streamlit app
+def logistic_map(r, x):
+    return r * x * (1 - x)
+
+# Generate chaotic key based on the logistic map
+def generate_key(seed, n):
+    key = []
+    x = seed
+    for _ in range(n):
+        x = logistic_map(3.9, x)
+        key.append(int(x * 255) % 256)  # Map to 0-255
+    return np.array(key, dtype=np.uint8)
+
+# Encrypt the image using chaotic logistic map
+def encrypt_image(img_array, seed):
+    h, w, c = img_array.shape
+    flat_image = img_array.flatten()
+    chaotic_key = generate_key(seed, len(flat_image))
+    encrypted_flat = [pixel ^ chaotic_key[i] for i, pixel in enumerate(flat_image)]
+    encrypted_array = np.array(encrypted_flat, dtype=np.uint8).reshape(h, w, c)
+    return encrypted_array
+
+# Streamlit App
 def main():
     st.title("Chaotic Logistic Map Image Encryption")
-    st.write("Upload an image to encrypt using chaotic logistic map encryption.")
+    st.write("Upload an image to encrypt using a chaotic logistic map.")
 
-    # File uploader
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
     if uploaded_file is not None:
-        # Display the uploaded image
+        # Load the uploaded image
         input_image = Image.open(uploaded_file)
-        st.image(input_image, caption="Uploaded Image", use_container_width=True)
+        img_array = np.array(input_image)
 
-        # Convert the image to numpy array
-        input_image_np = np.array(input_image)
+        # Display original image
+        st.image(input_image, caption="Uploaded Image", use_container_width=True)
 
         # Key input
-        key = st.slider("Set the encryption key (0 < key < 1)", min_value=0.001, max_value=0.999, step=0.001)
+        key_seed = st.slider("Set the encryption key seed (0 < key < 1)", min_value=0.001, max_value=0.999, step=0.001)
+
         if st.button("Encrypt Image"):
             # Encrypt the image
-            encrypted_image_np = encrypt_image(input_image_np, key)
-            encrypted_image = Image.fromarray(encrypted_image_np)
+            encrypted_array = encrypt_image(img_array, key_seed)
+            encrypted_image = Image.fromarray(encrypted_array)
 
-            # Display the encrypted image
+            # Display encrypted image
             st.image(encrypted_image, caption="Encrypted Image", use_container_width=True)
 
             # Save encrypted image to a buffer
@@ -50,7 +55,7 @@ def main():
             encrypted_image.save(buffer, format="PNG")
             buffer.seek(0)
 
-            # Provide download link for the encrypted image
+            # Download button for encrypted image
             st.download_button(
                 label="Download Encrypted Image",
                 data=buffer,