naming fix

server.py

@@ -32,6 +32,10 @@ FHE_SERVER = FHEModelServer(DEPLOYMENT_DIR)
 
 def get_server_file_path(file_type: str, user_id: str) -> Path:
     """Get the path to a file on the server."""
+    if file_type == "encrypted_image":
+        file_type = "encrypted"
+    elif file_type == "evaluation_key":
+        file_type = "evaluation"
     return Path(__file__).parent / "server_tmp" / f"{file_type}_{user_id}"
 
 @app.post("/send_input")
@@ -41,9 +45,17 @@ async def send_input(user_id: str = Form(), files: List[UploadFile] = File(...))
         for file in files:
             file_path = get_server_file_path(file.filename.split("_")[0], user_id)
             logger.info(f"Saving file to: {file_path}")
+            file_path.parent.mkdir(parents=True, exist_ok=True)  # Ensure the directory exists
             with file_path.open("wb") as buffer:
                 content = await file.read()
                 buffer.write(content)
+            
+            # Check if the file was saved successfully
+            if not file_path.exists():
+                raise IOError(f"Failed to save file at {file_path}")
+            else:
+                logger.info(f"File saved successfully at {file_path}")
+
         return JSONResponse(content={"message": "Files received successfully"})
     except Exception as e:
         logger.error(f"Error in send_input: {str(e)}")
@@ -55,8 +67,8 @@ def run_fhe(user_id: str = Form()):
     logger.info(f"Starting FHE execution for user {user_id}")
     try:
         # Retrieve the encrypted input image and the evaluation key paths
-        encrypted_image_path = get_server_file_path("encrypted_image", user_id)
-        evaluation_key_path = get_server_file_path("evaluation_key", user_id)
+        encrypted_image_path = get_server_file_path("encrypted", user_id)
+        evaluation_key_path = get_server_file_path("evaluation", user_id)
 
         logger.info(f"Looking for encrypted_image at: {encrypted_image_path}")
         logger.info(f"Looking for evaluation_key at: {evaluation_key_path}")