alk

app.py

@@ -123,40 +123,38 @@ def get_output(user_id: str = Form()):
 @app.post("/generate_keys")
 def generate_keys(symptoms: Symptoms):
     """
-    Endpoint to generate keys based on user symptoms.
+    Endpoint pour générer des clés basées sur les symptômes de l'utilisateur.
 
     Args:
-        user_symptoms (List[str]): The list of user symptoms.
+        symptoms (Symptoms): Les symptômes de l'utilisateur.
 
     Returns:
-        JSONResponse: A response containing the generated keys and user ID.
+        JSONResponse: Réponse contenant les clés générées et l'ID utilisateur.
     """
-    def is_none(obj):
-        return obj is None or (obj is not None and len(obj) == 0)
-
-    # Call the key generation function
+    # Appel de la fonction de nettoyage
     clean_directory()
 
-    if is_none(symptoms):
+    # Vérification si la liste des symptômes est vide
+    if not symptoms.user_symptoms:
         return JSONResponse(
-            status_code=400, content={"error": "Please submit your symptoms first."}
+            status_code=400, content={"error": "Veuillez soumettre vos symptômes en premier."}
         )
 
-    # Generate a random user ID
+    # Génération d'un ID utilisateur aléatoire
     user_id = np.random.randint(0, 2**32)
-    print(f"Your user ID is: {user_id}....")
+    print(f"Votre ID utilisateur est : {user_id}....")
 
     client = FHEModelClient(path_dir=DEPLOYMENT_DIR, key_dir=KEYS_DIR / f"{user_id}")
     client.load()
 
-    # Creates the private and evaluation keys on the client side
+    # Création des clés privées et d'évaluation côté client
     client.generate_private_and_evaluation_keys()
 
-    # Get the serialized evaluation keys
+    # Récupération des clés d'évaluation sérialisées
     serialized_evaluation_keys = client.get_serialized_evaluation_keys()
     assert isinstance(serialized_evaluation_keys, bytes)
 
-    # Save the evaluation key
+    # Sauvegarde de la clé d'évaluation
     evaluation_key_path = KEYS_DIR / f"{user_id}/evaluation_key"
     with evaluation_key_path.open("wb") as f:
         f.write(serialized_evaluation_keys)