Compile deployment files to python 3.10 and improve app display

.gitignore

@@ -1,2 +1,7 @@
 __pycache__/
 .ipynb_checkpoints
+
+.venv
+deployment_files/.*
+deployment_files/client_dir/
+deployment_files/server_dir/

app.py

@@ -171,10 +171,11 @@ def key_gen_fn(user_symptoms: List[str]) -> Dict:
     return {
         error_box2: gr.update(visible=False),
         key_box: gr.update(visible=False, value=serialized_evaluation_keys_shorten_hex),
-        user_id_box: gr.update(visible=True, value=user_id),
+        user_id_box: gr.update(visible=False, value=user_id),
         key_len_box: gr.update(
             visible=False, value=f"{len(serialized_evaluation_keys) / (10**6):.2f} MB"
         ),
+        gen_key_btn: gr.update(value="Keys have been generated ✅")
     }
 
 
@@ -608,9 +609,9 @@ if __name__ == "__main__":
             "The evaluation key will be transmitted to the server for further processing."
         )
 
-        gen_key_btn = gr.Button("Generate the evaluation key")
+        gen_key_btn = gr.Button("Generate the private and evaluation keys.")
         error_box2 = gr.Textbox(label="Error ❌", visible=False)
-        user_id_box = gr.Textbox(label="User ID:", visible=True)
+        user_id_box = gr.Textbox(label="User ID:", visible=False)
         key_len_box = gr.Textbox(label="Evaluation Key Size:", visible=False)
         key_box = gr.Textbox(label="Evaluation key (truncated):", max_lines=3, visible=False)
 
@@ -622,6 +623,7 @@ if __name__ == "__main__":
                 user_id_box,
                 key_len_box,
                 error_box2,
+                gen_key_btn,
             ],
         )
 

{deployment_logit → deployment_files}/versions.json

@@ -1 +1 @@
-{"concrete-python": "2.5", "concrete-ml": "1.4.0", "python": "3.8.16"}
+{"concrete-python": "2.5", "concrete-ml": "1.4.0", "python": "3.10.11"}

dev.py

@@ -42,7 +42,7 @@ fhe_circuit = clf.compile(X_train)
 
 fhe_circuit.client.keygen(force=False)
 
-path_to_model = Path("./deployment_logit/").resolve()
+path_to_model = Path("./deployment_files/").resolve()
 
 if path_to_model.exists():
     shutil.rmtree(path_to_model)

server.py

@@ -9,6 +9,9 @@ from utils import DEPLOYMENT_DIR, SERVER_DIR  # pylint: disable=no-name-in-modul
 
 from concrete.ml.deployment import FHEModelServer
 
+# Load the FHE server
+FHE_SERVER = FHEModelServer(DEPLOYMENT_DIR)
+
 # Initialize an instance of FastAPI
 app = FastAPI()
 
@@ -62,12 +65,9 @@ def run_fhe(
         encrypted_output = encrypted_output_file.read()
         evaluation_key = evaluation_key_file.read()
 
-    # Load the FHE server and the model
-    fhe_server = FHEModelServer(DEPLOYMENT_DIR)
-
     # Run the FHE execution
     start = time.time()
-    encrypted_output = fhe_server.run(encrypted_output, evaluation_key)
+    encrypted_output = FHE_SERVER.run(encrypted_output, evaluation_key)
     assert isinstance(encrypted_output, bytes)
     fhe_execution_time = round(time.time() - start, 2)
 

utils.py

@@ -16,7 +16,7 @@ INPUT_BROWSER_LIMIT = 400
 SERVER_URL = "http://localhost:8000/"
 
 CURRENT_DIR = Path(__file__).parent
-DEPLOYMENT_DIR = CURRENT_DIR / "deployment_logit"
+DEPLOYMENT_DIR = CURRENT_DIR / "deployment_files"
 KEYS_DIR = DEPLOYMENT_DIR / ".fhe_keys"
 CLIENT_DIR = DEPLOYMENT_DIR / "client_dir"
 SERVER_DIR = DEPLOYMENT_DIR / "server_dir"