keygen logging

app.py

@@ -126,35 +126,43 @@ def clean_temporary_files(n_keys=20):
                 file.unlink()
 
 def keygen():
-    """Generate the private key for seizure detection.
+    """Generate the private key for seizure detection."""
+    logger.info("Starting key generation process")
+    try:
+        # Clean temporary files
+        clean_temporary_files()
 
-    Returns:
-        (user_id, True) (Tuple[int, bool]): The current user's ID and a boolean used for visual display.
+        # Create an ID for the current user
+        user_id = numpy.random.randint(0, 2**32)
+        logger.info(f"Generated user_id: {user_id}")
 
-    """
-    # Clean temporary files
-    clean_temporary_files()
+        # Retrieve the client API
+        client = get_client(user_id)
+        logger.info("Retrieved client API")
 
-    # Create an ID for the current user
-    user_id = numpy.random.randint(0, 2**32)
+        # Generate a private key
+        logger.info("Generating private and evaluation keys")
+        client.generate_private_and_evaluation_keys(force=True)
+        logger.info("Private and evaluation keys generated successfully")
 
-    # Retrieve the client API
-    client = get_client(user_id)
+        # Retrieve the serialized evaluation key
+        logger.info("Retrieving serialized evaluation keys")
+        evaluation_key = client.get_serialized_evaluation_keys()
+        logger.info("Serialized evaluation keys retrieved")
 
-    # Generate a private key
-    client.generate_private_and_evaluation_keys(force=True)
+        # 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)
+        logger.info(f"Saving evaluation key to: {evaluation_key_path}")
 
-    # Retrieve the serialized evaluation key
-    evaluation_key = client.get_serialized_evaluation_keys()
+        with evaluation_key_path.open("wb") as evaluation_key_file:
+            evaluation_key_file.write(evaluation_key)
+        logger.info("Evaluation key saved successfully")
 
-    # 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)
-
-    with evaluation_key_path.open("wb") as evaluation_key_file:
-        evaluation_key_file.write(evaluation_key)
-
-    return (user_id, True)
+        return (user_id, True)
+    except Exception as e:
+        logger.error(f"Error during key generation: {str(e)}")
+        raise gr.Error(f"Key generation failed: {str(e)}")
 
 def encrypt(user_id, input_image):
     """Encrypt the given image for seizure detection.