Update app.py

app.py

@@ -11,12 +11,14 @@ app = FastAPI(title="EEG Epilepsy Prediction API")
 
 @app.get("/", tags=["Introduction Endpoints"])
 async def index():
-    return {
-        "message": (
-            "Hello world! Welcome to the EEG Epilepsy Prediction API. "
-            "Submit an EEG recording EDF file to the `/predict` endpoint to receive a prediction."
-        )
-    }
+    """
+    Simply returns a welcome message!
+    """
+    message = (
+        "Hello world! Welcome to the EEG Epilepsy Prediction API. "
+        "Submit an EEG recording EDF file to the `/predict` endpoint to receive a prediction."
+    )
+    return message
 
 @app.post("/predict", tags=["Machine Learning"])
 async def predict_endpoint(
@@ -24,6 +26,12 @@ async def predict_endpoint(
     model_choice: str = "2DCNN",
     ensemble_method: str = None
 ):
+    """
+    Query parameters:
+      - model_choice: Choose one model among "2DCNN", "EEGNet", "EpilepsyNet", or "ensemble".
+      - ensemble_method: (Optional, required if model_choice is "ensemble") 
+                           The ensemble method to use ("average" or "voting").
+    """
     print("Saving uploaded file as temporary file...")
     try:
         suffix = os.path.splitext(file.filename)[1]
@@ -31,8 +39,8 @@ async def predict_endpoint(
             tmp.write(await file.read())
             tmp_path = tmp.name
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error saving temporary file: {e}")
-    
+        raise HTTPException(status_code=500, detail="Error saving temporary file")
+
     print("Performing prediction using model_choice =", model_choice)
     try:
         if model_choice.lower() == "ensemble":
@@ -47,8 +55,8 @@ async def predict_endpoint(
             )
     except Exception as e:
         os.remove(tmp_path)
-        raise HTTPException(status_code=500, detail=f"Prediction failed: {e}")
-    
+        raise HTTPException(status_code=400, detail=f"Prediction failed: {e}")
+
     os.remove(tmp_path)
 
     response = {