image shape transform

app.py

@@ -147,20 +147,26 @@ def encrypt(user_id, input_image):
     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 (224, 224, 3)
     if input_image.shape != (224, 224, 3):
         input_image_pil = Image.fromarray(input_image)
         input_image_pil = input_image_pil.resize((224, 224))
         input_image = numpy.array(input_image_pil)
 
+    # Convert RGB to grayscale
+    input_image_gray = numpy.mean(input_image, axis=2).astype(numpy.uint8)
+
+    # Reshape to (1, 1, 224, 224)
+    input_image_reshaped = input_image_gray.reshape(1, 1, 224, 224)
+
+    # Convert to int12 (assuming the range is 0-255, we can simply cast to int16)
+    input_image_int12 = input_image_reshaped.astype(numpy.int16)
+
     # Retrieve the client API
     client = get_client(user_id)
 
     # Pre-process, encrypt and serialize the image
-    encrypted_image = client.encrypt_serialize(input_image)
+    encrypted_image = client.encrypt_serialize(input_image_int12)
 
     # 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