Update app.py

app.py

@@ -2,6 +2,7 @@ from PIL import Image
 import numpy as np
 import streamlit as st
 import io
+import time
 import random
 
 # Resize image to prevent memory issues with large images
@@ -38,13 +39,6 @@ def shuffle_pixels(img_array, seed):
     shuffled = flattened[indices]
     return shuffled.reshape(h, w, c), indices
 
-# Unshuffle pixels (for decryption, if needed)
-def unshuffle_pixels(shuffled_array, indices, original_shape):
-    flattened = shuffled_array.reshape(-1, shuffled_array.shape[-1])
-    unshuffled = np.zeros_like(flattened)
-    unshuffled[indices] = flattened
-    return unshuffled.reshape(original_shape)
-
 # Multi-layer encryption using logistic map and pixel shuffling
 def encrypt_image(img_array, seed):
     h, w, c = img_array.shape
@@ -75,6 +69,13 @@ def main():
 
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
     if uploaded_file is not None:
+        with st.spinner("Processing upload..."):
+            # Simulate a loading delay for upload progress
+            progress_bar = st.progress(0)
+            for percent_complete in range(0, 101, 20):
+                time.sleep(0.1)  # Simulate progress
+                progress_bar.progress(percent_complete)
+
         # Load the uploaded image
         input_image = Image.open(uploaded_file)
 
@@ -91,25 +92,32 @@ def main():
         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_array = encrypt_image(img_array, key_seed)
-            encrypted_image = Image.fromarray(encrypted_array)
-
-            # Display encrypted image
-            st.image(encrypted_image, caption="Encrypted Image", use_container_width=True)
-
-            # Save encrypted image to a buffer
-            buffer = io.BytesIO()
-            encrypted_image.save(buffer, format="PNG")
-            buffer.seek(0)
-
-            # Download button for encrypted image
-            st.download_button(
-                label="Download Encrypted Image",
-                data=buffer,
-                file_name="encrypted_image.png",
-                mime="image/png"
-            )
+            with st.spinner("Encrypting image..."):
+                # Simulate a loading delay for encryption progress
+                progress_bar = st.progress(0)
+                for percent_complete in range(0, 101, 20):
+                    time.sleep(0.1)  # Simulate progress
+                    progress_bar.progress(percent_complete)
+
+                # Encrypt the image
+                encrypted_array = encrypt_image(img_array, key_seed)
+                encrypted_image = Image.fromarray(encrypted_array)
+
+                # Display encrypted image
+                st.image(encrypted_image, caption="Encrypted Image", use_container_width=True)
+
+                # Save encrypted image to a buffer
+                buffer = io.BytesIO()
+                encrypted_image.save(buffer, format="PNG")
+                buffer.seek(0)
+
+                # Download button for encrypted image
+                st.download_button(
+                    label="Download Encrypted Image",
+                    data=buffer,
+                    file_name="encrypted_image.png",
+                    mime="image/png"
+                )
 
 if __name__ == "__main__":
     main()