new2

app.py

@@ -20,8 +20,17 @@ class PredictionInput(BaseModel):
 @app.post("/predict")
 def predict(input_data: PredictionInput):
     X = np.array(input_data.data).reshape(1, -1)
-    prediction = model.predict(X)
-    return {"prediction": prediction.tolist()}
+    prediction = model.predict(X)[0]  # ambil nilai scalar
+    labels = {
+        0: "Bipolar Type 1",
+        1: "Bipolar Type 2",
+        2: "Depression",
+        3: "Normal"
+    }
+    return {
+        "prediction": int(prediction),
+        "label": labels.get(int(prediction), "Unknown")
+    }
 
 
 # Tambahkan baris ini jika kamu menjalankan app.py langsung!