chore: add resize_img

app.py

@@ -197,12 +197,9 @@ def encrypt(user_id, input_image, filter_name):
     
     # Resize the image if it hasn't the shape (100, 100, 3)
     if input_image.shape != (100 , 100, 3):
-        print(f"Before: {type(input_image)=}, {input_image.shape=}")
         input_image_pil = Image.fromarray(input_image)
-        # Resize the image
         input_image_pil = input_image_pil.resize((100, 100))
         input_image = numpy.array(input_image_pil)
-        print(f"After: {type(input_image)=}, {input_image.shape=}")
 
     # Retrieve the client API
     client = get_client(user_id, filter_name)
@@ -220,7 +217,7 @@ def encrypt(user_id, input_image, filter_name):
     # Create a truncated version of the encrypted image for display
     encrypted_image_short = shorten_bytes_object(encrypted_image)
 
-    return (input_image, encrypted_image_short)
+    return (resize_img(input_image), encrypted_image_short)
 
 
 def send_input(user_id, filter_name):
@@ -321,7 +318,8 @@ def get_output(user_id, filter_name):
             # Decrypt the image using a different (wrong) key for display
             output_image_representation = decrypt_output_with_wrong_key(encrypted_output, filter_name)
 
-            return output_image_representation
+            return {encrypted_output_representation: gr.update(value=resize_img(output_image_representation))}
+
         else:
             raise gr.Error("Please wait for the FHE execution to be completed.")
 
@@ -338,6 +336,9 @@ def decrypt_output(user_id, filter_name):
             well as two booleans used for resetting Gradio checkboxes
 
     """
+
+    print("Decrypt output ------------------------------------------------")
+
     if user_id == "":
         raise gr.Error("Please generate the private key first.")
 
@@ -355,10 +356,27 @@ def decrypt_output(user_id, filter_name):
     client = get_client(user_id, filter_name)
 
     # Deserialize, decrypt and post-process the encrypted output
-    output_image = client.deserialize_decrypt_post_process(encrypted_output_image)
+    decrypted_ouput = client.deserialize_decrypt_post_process(encrypted_output_image)
 
-    return output_image, False, False
+    print(f"Decrypted output: {decrypted_ouput.shape=}")
+    
+    return {output_image: gr.update(value=resize_img(decrypted_ouput))}
+
+
+def resize_img(img, width=256, height=256):
+    # Convert to PIL Image
+    print("Reshape img before", img.shape, type(img))
+    if img.dtype != numpy.uint8:
+        img = img.astype(numpy.uint8)
+    
+    img_pil = Image.fromarray(img)
+    print(type(img_pil))
+    # Resize the image
+    resized_img_pil = img_pil.resize((width, height))
+    print("Reshape img before", resized_img_pil.size)
 
+    # Convert back to a NumPy array
+    return numpy.array(resized_img_pil)
 
 demo = gr.Blocks()
 
@@ -464,7 +482,7 @@ with demo:
 
     with gr.Row():
         encrypted_output_representation = gr.Image(
-            label=f"Encrypted output representation ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", 
+            label=f"Encrypted output representation ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):",
             interactive=False,
             height=256,
             width=256,