send input

app.py

@@ -208,19 +208,14 @@ def encrypt(user_id, input_image):
     return (resize_img(input_image), encrypted_image_short)
 
 def send_input(user_id):
-    """Send the encrypted input image as well as the evaluation key to the server.
-
-    Args:
-        user_id (int): The current user's ID.
-    """
+    """Send the encrypted input image as well as the evaluation key to the server."""
     # Get the evaluation key path
     evaluation_key_path = get_client_file_path("evaluation_key", user_id)
+    encrypted_input_path = get_client_file_path("encrypted_image", user_id)
 
     if user_id == "" or not evaluation_key_path.is_file():
         raise gr.Error("Please generate the private key first.")
 
-    encrypted_input_path = get_client_file_path("encrypted_image", user_id)
-
     if not encrypted_input_path.is_file():
         raise gr.Error("Please generate the private key and then encrypt an image first.")
 
@@ -230,17 +225,16 @@ def send_input(user_id):
     }
 
     files = [
-        ("files", open(encrypted_input_path, "rb")),
-        ("files", open(evaluation_key_path, "rb")),
+        ("files", ("encrypted_image", open(encrypted_input_path, "rb"), "application/octet-stream")),
+        ("files", ("evaluation_key", open(evaluation_key_path, "rb"), "application/octet-stream")),
     ]
 
+    logger.info(f"Sending encrypted_image from: {encrypted_input_path}")
+    logger.info(f"Sending evaluation_key from: {evaluation_key_path}")
+
     # Send the encrypted input image and evaluation key to the server
     url = SERVER_URL + "send_input"
-    with requests.post(
-        url=url,
-        data=data,
-        files=files,
-    ) as response:
+    with requests.post(url=url, data=data, files=files) as response:
         return response.ok
 
 def run_fhe(user_id):

server.py

@@ -40,6 +40,7 @@ async def send_input(user_id: str = Form(), files: List[UploadFile] = File(...))
     try:
         for file in files:
             file_path = get_server_file_path(file.filename.split("_")[0], user_id)
+            logger.info(f"Saving file to: {file_path}")
             with file_path.open("wb") as buffer:
                 content = await file.read()
                 buffer.write(content)
@@ -47,7 +48,7 @@ async def send_input(user_id: str = Form(), files: List[UploadFile] = File(...))
     except Exception as e:
         logger.error(f"Error in send_input: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
-
+    
 @app.post("/run_fhe")
 def run_fhe(user_id: str = Form()):
     """Execute seizure detection on the encrypted input image using FHE."""
@@ -57,9 +58,14 @@ def run_fhe(user_id: str = Form()):
         encrypted_image_path = get_server_file_path("encrypted_image", user_id)
         evaluation_key_path = get_server_file_path("evaluation_key", user_id)
 
+        logger.info(f"Looking for encrypted_image at: {encrypted_image_path}")
+        logger.info(f"Looking for evaluation_key at: {evaluation_key_path}")
+
         # Check if files exist
-        if not encrypted_image_path.exists() or not evaluation_key_path.exists():
-            raise FileNotFoundError("Required files not found")
+        if not encrypted_image_path.exists():
+            raise FileNotFoundError(f"Encrypted image file not found at {encrypted_image_path}")
+        if not evaluation_key_path.exists():
+            raise FileNotFoundError(f"Evaluation key file not found at {evaluation_key_path}")
 
         # Read the files using the above paths
         with encrypted_image_path.open("rb") as encrypted_image_file, evaluation_key_path.open("rb") as evaluation_key_file: