alk

app.py

@@ -2,12 +2,13 @@ from fastapi import FastAPI, File, Form, UploadFile, Body
 from fastapi.responses import JSONResponse, Response
 from concrete.ml.deployment import FHEModelServer
 import numpy as np
+from pydantic import BaseModel
+
 
 from concrete.ml.deployment import FHEModelClient
 import subprocess
 from pathlib import Path
 
-
 from utils import (
     CLIENT_DIR,
     CURRENT_DIR,
@@ -24,7 +25,6 @@ from utils import (
     pretty_print,
 )
 
-
 import time
 from typing import List
 
@@ -32,13 +32,15 @@ from typing import List
 # FHE_SERVER = FHEModelServer(DEPLOYMENT_DIR)
 
 
+class Symptoms(BaseModel):
+    user_symptoms: List[str]
+
 app = FastAPI()
 
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
 
-
 def root():
     """
     Root endpoint of the health prediction API.
@@ -48,7 +50,6 @@ def root():
     """
     return {"message": "Welcome to your disease prediction with FHE!"}
 
-
 @app.post("/send_input")
 def send_input(
     user_id: str = Form(),
@@ -69,7 +70,6 @@ def send_input(
         encrypted_input.write(files[0].file.read())
         evaluation_key.write(files[1].file.read())
 
-
 @app.post("/run_fhe")
 def run_fhe(
     user_id: str = Form(),
@@ -102,7 +102,6 @@ def run_fhe(
 
     return JSONResponse(content=fhe_execution_time)
 
-
 @app.post("/get_output")
 def get_output(user_id: str = Form()):
     """Retrieve the encrypted output from the server."""
@@ -121,12 +120,11 @@ def get_output(user_id: str = Form()):
     # Send the encrypted output
     return Response(encrypted_output)
 
-
 @app.post("/generate_keys")
-def generate_keys(user_symptoms: List[str] = Body(...)):
+def generate_keys(symptoms: Symptoms):
     """
     Endpoint to generate keys based on user symptoms.
-    
+
     Args:
         user_symptoms (List[str]): The list of user symptoms.
 
@@ -139,7 +137,7 @@ def generate_keys(user_symptoms: List[str] = Body(...)):
     # Call the key generation function
     clean_directory()
 
-    if is_none(user_symptoms):
+    if is_none(symptoms):
         return JSONResponse(
             status_code=400, content={"error": "Please submit your symptoms first."}
         )
@@ -173,12 +171,11 @@ def generate_keys(user_symptoms: List[str] = Body(...)):
         }
     )
 
-
 @app.post("/run_dev")
 def run_dev_script():
     """
     Endpoint to execute the dev.py script to generate deployment files.
-    
+
     Returns:
         JSONResponse: Success message or error details.
     """
@@ -204,4 +201,4 @@ def run_dev_script():
         return JSONResponse(
             status_code=500,
             content={"error": "Failed to execute dev.py", "details": e.stderr}
-        )
+        )