Fix comments and imlearn version

app.py

@@ -19,8 +19,6 @@ from settings import (
     FAMILY_STATUS,
 )
 from backend import (
-    shorten_bytes_object,
-    clean_temporary_files,
     pre_process_keygen_encrypt_send_user,
     pre_process_keygen_encrypt_send_bank,
     pre_process_keygen_encrypt_send_third_party,

backend.py

@@ -53,7 +53,7 @@ def shorten_bytes_object(bytes_object, limit=500):
 
 
 def clean_temporary_files(n_keys=20):
-    """Clean keys and encrypted images.
+    """Clean keys and encrypted inputs.
 
     A maximum of n_keys keys and associated temporary files are allowed to be stored. Once this 
     limit is reached, the oldest files are deleted.
@@ -100,7 +100,7 @@ def _get_client(client_id, client_type):
 
 
 def _keygen(client_id, client_type):
-    """Generate the private key associated to a filter.
+    """Generate the private key associated to a client.
 
     Args:
         client_id (int): The client ID to consider.
@@ -129,7 +129,7 @@ def _keygen(client_id, client_type):
 
 
 def _send_input(client_id, client_type):
-    """Send the encrypted input image as well as the evaluation key to the server.
+    """Send the encrypted inputs as well as the evaluation key to the server.
 
     Args:
         client_id (int): The client ID to consider.
@@ -150,7 +150,7 @@ def _send_input(client_id, client_type):
         ("files", open(evaluation_key_path, "rb")),
     ]
 
-    # Send the encrypted input image and evaluation key to the server
+    # Send the encrypted inputs and evaluation key to the server
     url = SERVER_URL + "send_input"
     with requests.post(
         url=url,
@@ -210,7 +210,7 @@ def _keygen_encrypt_send(inputs, client_type):
     with encrypted_inputs_path.open("wb") as encrypted_inputs_file:
         encrypted_inputs_file.write(encrypted_inputs)
 
-    # Create a truncated version of the encrypted image for display
+    # Create a truncated version of the encrypted inputs for display
     encrypted_inputs_short = shorten_bytes_object(encrypted_inputs)
 
     _send_input(client_id, client_type)
@@ -349,7 +349,7 @@ def get_output(user_id, bank_id, third_party_id):
         "third_party_id": third_party_id,
     }
 
-    # Retrieve the encrypted output image
+    # Retrieve the encrypted output
     url = SERVER_URL + "get_output"
     with requests.post(
         url=url,

deployment_files/client.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:06c7bd8264089eb169342aa5c3f638b11d894c54d054511a91523bfdfab69487
-size 76130
+oid sha256:1661827a0038bff99f6c880f3528409d0468e4a0466fb63d2967736599157a74
+size 76152

deployment_files/server.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:04c3f1de7261abe6ad075f6cc13885677ddf4ca0b03d6a31f26a60f94d5aa2ae
-size 10975
+oid sha256:2c2167b544eeea289a485070216ee01694f3f15df856067f7458878b78452836
+size 11049

requirements.txt

@@ -1,4 +1,3 @@
-# concrete-ml==1.3.0
 gradio==3.40.1
 concrete-ml==1.3.0
-imblearn==0.0
+imbalanced-learn==0.11.0

server.py

@@ -5,10 +5,10 @@ from typing import List
 from fastapi import FastAPI, File, Form, UploadFile
 from fastapi.responses import JSONResponse, Response
 
-from settings import DEPLOYMENT_PATH, SERVER_FILES, CLIENT_TYPES
+from settings import DEPLOYMENT_PATH, SERVER_FILES
 from utils.client_server_interface import MultiInputsFHEModelServer
 
-# Load the server objects related to all currently available filters once and for all
+# Load the server
 FHE_SERVER = MultiInputsFHEModelServer(DEPLOYMENT_PATH)