chore: remove all size displays

app.py

@@ -163,10 +163,6 @@ def keygen(filter_name):
     # execution
     evaluation_key = client.get_serialized_evaluation_keys()
 
-    # Compute the private key's size in Kilobytes
-    private_key_path = next(client.key_dir.iterdir()) / "0_0/secretKey_big" 
-    private_key_size = private_key_path.stat().st_size / 1000
-
     # Save evaluation_key as bytes in a file as it is too large to pass through regular Gradio
     # buttons (see https://github.com/gradio-app/gradio/issues/1877)
     evaluation_key_path = get_client_file_path("evaluation_key", user_id, filter_name)
@@ -174,7 +170,7 @@ def keygen(filter_name):
     with evaluation_key_path.open("wb") as evaluation_key_file:
         evaluation_key_file.write(evaluation_key)
 
-    return (user_id, True, private_key_size)
+    return (user_id, True)
 
 
 def encrypt(user_id, input_image, filter_name):
@@ -202,9 +198,6 @@ def encrypt(user_id, input_image, filter_name):
     # Pre-process, encrypt and serialize the image
     encrypted_image = client.encrypt_serialize(input_image)
 
-    # Compute the input's size in Megabytes
-    encrypted_input_size = len(encrypted_image) / 1000000
-
     # 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, filter_name)
@@ -215,7 +208,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, encrypted_input_size)
+    return (input_image, encrypted_image_short)
 
 
 def send_input(user_id, filter_name):
@@ -306,9 +299,6 @@ def get_output(user_id, filter_name):
         if response.ok:
             encrypted_output = response.content
 
-            # Compute the output's size in Megabytes
-            encrypted_output_size = len(encrypted_output) / 1000000
-
             # Save the encrypted output to bytes in a file as it is too large to pass through regular
             # Gradio buttons (see https://github.com/gradio-app/gradio/issues/1877)
             encrypted_output_path = get_client_file_path("encrypted_output", user_id, filter_name)
@@ -319,7 +309,7 @@ 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, encrypted_output_size
+            return output_image_representation
         else:
             raise gr.Error("Please wait for the FHE execution to be completed.")
 
@@ -398,7 +388,7 @@ with demo:
     )
     with gr.Row():
         input_image = gr.Image(
-            label="Upload an image here.", shape=INPUT_SHAPE, source="upload", interactive=True
+            value=None, label="Upload an image here.", shape=INPUT_SHAPE, source="upload", interactive=True,
         )
 
         examples = gr.Examples(
@@ -424,10 +414,6 @@ with demo:
     with gr.Row():
         keygen_checkbox = gr.Checkbox(label="Private key generated:", interactive=False)
 
-        private_key_size = gr.Number(
-            label="Private key size (in kB):", value=0, precision=1, interactive=False
-        )
-
     user_id = gr.Textbox(label="", max_lines=2, interactive=False, visible=False)
     
     gr.Markdown("### Step 4: Encrypt the image using FHE.")
@@ -438,10 +424,6 @@ with demo:
             label="Encrypted input representation:", max_lines=2, interactive=False
         )
 
-        encrypted_input_size = gr.Number(
-            label="Encrypted input size (in MB):", value=0, precision=1, interactive=False
-        )
-
     gr.Markdown("## Server side")
     gr.Markdown(
         "The encrypted value is received by the server. The server can then compute the filter "
@@ -473,10 +455,6 @@ with demo:
         )
         encrypted_output_representation.style(height=256, width=256)
 
-        encrypted_output_size = gr.Number(
-            label="Encrypted output size (in MB):", value=0, precision=1, interactive=False
-        )
-
     gr.Markdown("## Client side")
     gr.Markdown(
         "The encrypted output is sent back to the client, who can finally decrypt it with the "
@@ -508,14 +486,14 @@ with demo:
     keygen_button.click(
         keygen,
         inputs=[filter_name],
-        outputs=[user_id, keygen_checkbox, private_key_size],
+        outputs=[user_id, keygen_checkbox],
     )
 
     # Button to encrypt inputs on the client side
     encrypt_button.click(
         encrypt,
         inputs=[user_id, input_image, filter_name],
-        outputs=[original_image, encrypted_input, encrypted_input_size],
+        outputs=[original_image, encrypted_input],
     )
 
     # Button to send the encodings to the server using post method
@@ -530,7 +508,7 @@ with demo:
     get_output_button.click(
         get_output, 
         inputs=[user_id, filter_name], 
-        outputs=[encrypted_output_representation, encrypted_output_size]
+        outputs=[encrypted_output_representation]
     )
 
     # Button to decrypt the output on the client side