chore: reshape the image to (100, 100, 3) if not and add a check for rgb format

app.py

@@ -1,5 +1,5 @@
 """A local gradio app that filters images using FHE."""
-
+from PIL import Image
 import os
 import shutil
 import subprocess
@@ -191,6 +191,18 @@ def encrypt(user_id, input_image, filter_name):
 
     if input_image is None:
         raise gr.Error("Please choose an image first.")
+    
+    if input_image.shape[-1] != 3:
+        raise ValueError(f"Input image must have 3 channels (RGB). Current shape: {input_image.shape}")
+    
+    # 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)
@@ -482,7 +494,7 @@ with demo:
         )
 
         output_image = gr.Image(
-            label=f"Output image ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", 
+            label=f"Output image ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):",
             interactive=False,
             height=256,
             width=256,
@@ -513,7 +525,7 @@ with demo:
     # Button to send the encodings to the server using post method
     get_output_button.click(
         get_output, 
-        inputs=[user_id, filter_name], 
+        inputs=[user_id, filter_name],
         outputs=[encrypted_output_representation]
     )