Update app.py

app.py

@@ -4,7 +4,7 @@ import joblib
 import numpy as np
 from fastapi.middleware.cors import CORSMiddleware
 
-app = FastAPI(title="Web Attack Detection API")
+app = FastAPI(title="Bot Detection API")
 
 # Allow browser requests
 app.add_middleware(
@@ -24,38 +24,52 @@ class InputData(BaseModel):
 
 # Load model
 try:
-    model = joblib.load("web_attack_detection_model.pkl")
+    model = joblib.load("bot_detection_model.pkl")
 except Exception as e:
     raise RuntimeError(f"Model failed to load: {e}")
 
+# Root endpoint
 @app.get("/")
 def home():
-    return {"message": "Web attack detection model running"}
+    return {
+        "message": "Bot detection model running",
+        "expected_features": EXPECTED_FEATURES
+    }
 
+# Health check
 @app.get("/health")
 def health():
     return {"status": "ok"}
 
+# Prediction endpoint
 @app.post("/predict")
 def predict(data: InputData):
 
+    # Validate feature count
     if len(data.features) != EXPECTED_FEATURES:
         raise HTTPException(
             status_code=400,
             detail=f"Expected {EXPECTED_FEATURES} features"
         )
 
-    input_data = np.array(data.features).reshape(1, -1)
+    try:
+        input_data = np.array(data.features).reshape(1, -1)
 
-    prediction = model.predict(input_data)[0]
+        prediction = model.predict(input_data)[0]
 
-    # Convert prediction to readable output
-    if prediction == -1:
-        result = "attack detected"
-    else:
-        result = "normal request"
+        # Convert prediction to readable output
+        if prediction == 1:
+            result = "bot_detected"
+        else:
+            result = "normal_traffic"
 
-    return {
-        "prediction": result,
-        "raw_prediction": int(prediction)
-    }
+        return {
+            "prediction": result,
+            "raw_prediction": int(prediction)
+        }
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Prediction failed: {str(e)}"
+        )