updayte keygen

app.py

@@ -7,10 +7,12 @@ import time
 import gradio as gr
 import numpy
 import requests
+import numpy as np
 from itertools import chain
 
 from common import (
     CLIENT_TMP_PATH,
+    SEIZURE_DETECTION_MODEL_PATH,
     SERVER_TMP_PATH,
     EXAMPLES,
     INPUT_SHAPE,
@@ -19,7 +21,7 @@ from common import (
     SERVER_URL,
 )
 from client_server_interface import FHEClient
-
+from concrete.ml.deployment import FHEModelClient
 # Uncomment here to have both the server and client in the same terminal
 subprocess.Popen(["uvicorn", "server:app"], cwd=REPO_DIR)
 time.sleep(3)
@@ -108,24 +110,24 @@ def keygen():
     # Clean temporary files
     clean_temporary_files()
 
-    # Create an ID for the current user
-    user_id = numpy.random.randint(0, 2**32)
+    # Generate a random user ID
+    user_id = np.random.randint(0, 2**32)
+    print(f"Your user ID is: {user_id}....")
 
-    # Retrieve the client API
-    client = get_client(user_id)
+    client = FHEModelClient(path_dir=SEIZURE_DETECTION_MODEL_PATH, key_dir=KEYS_PATH / f"{user_id}")
+    client.load()
 
-    # Generate a private key
-    client.generate_private_and_evaluation_keys(force=True)
+    # Creates the private and evaluation keys on the client side
+    client.generate_private_and_evaluation_keys()
 
-    # Retrieve the serialized evaluation key
-    evaluation_key = client.get_serialized_evaluation_keys()
-
-    # 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)
+    # Get the serialized evaluation keys
+    serialized_evaluation_keys = client.get_serialized_evaluation_keys()
+    assert isinstance(serialized_evaluation_keys, bytes)
 
-    with evaluation_key_path.open("wb") as evaluation_key_file:
-        evaluation_key_file.write(evaluation_key)
+    # Save the evaluation key
+    evaluation_key_path = KEYS_PATH / f"{user_id}/evaluation_key"
+    with evaluation_key_path.open("wb") as f:
+        f.write(serialized_evaluation_keys)
 
     return (user_id, True)
 

client_server_interface.py

@@ -89,6 +89,8 @@ class FHEClient:
         self.model_path = SEIZURE_DETECTION_MODEL_PATH
         self.key_dir = key_dir
 
+        print(self.model_path)
+
         # If model_path does not exist raise
         assert self.model_path.exists(), f"{self.model_path} does not exist. Please specify a valid path."