aled

app.py

@@ -159,7 +159,7 @@ def encrypt(user_id, input_image):
 
     import numpy as np
     # Resize image if necessary
-    if input_image.shape != (32, 32, 3):
+    if input_image.shape != (32, 32, 1):
         input_image_pil = Image.fromarray(input_image)
         input_image_pil = input_image_pil.resize((32, 32))
         input_image = np.array(input_image_pil)
@@ -172,19 +172,28 @@ def encrypt(user_id, input_image):
     input_image = (input_image / 255.0 * 4095 - 2048).astype(np.int16)
     input_image = np.clip(input_image, -2048, 2047)
 
+    print("Processing the image finished")
     # Retrieve the client API
     client = get_client(user_id)
 
+    print("Client retrieved")
+
     # Pre-process, encrypt and serialize the image
     encrypted_image = client.encrypt_serialize(input_image)
 
+    print("Encrypted image retrieved")
+
     # Save encrypted_image to bytes in a file, since too large to pass through regular Gradio
     # buttons, https://github.com/gradio-app/gradio/issues/1877
     encrypted_image_path = get_client_file_path("encrypted_image", user_id)
 
+    print("Encrypted image path retrieved")
+
     with encrypted_image_path.open("wb") as encrypted_image_file:
         encrypted_image_file.write(encrypted_image)
 
+    print("Encrypted image file retrieved")
+
     # Get a short representation of the encrypted image for display purposes
     encrypted_image_short = encrypted_image[:100]  # Take first 100 bytes
 
@@ -339,7 +348,7 @@ with demo:
     gr.Markdown("## Client side")
     gr.Markdown("### Step 1: Upload an EEG image. ")
     gr.Markdown(
-        f"The image will automatically be resized to shape (224x224). "
+        f"The image will automatically be resized to shape (32, 32). "
         "The image here, however, is displayed in its original resolution."
     )
     with gr.Row():