Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
 import numpy as np
+import pandas as pd
 import joblib
 from supabase import create_client
 import os
@@ -44,7 +45,7 @@ def transform_features(data: Features):
     peak = data.peak_count
     slope = data.slope
 
-    energy = mean * 40
+    energy = data.power * (40 * 0.04) / 3600
     ratio = mx / (mn + 1)
     cv = std / (mean + 1)
     peak_ratio = peak / 40
@@ -75,7 +76,17 @@ def predict(data: Features):
 
     features = transform_features(data)
 
-    X = scaler.transform([features])
+    columns = [
+    "mean","max","min","std","range",
+    "peak_count","slope",
+    "energy","ratio",
+    "cv","peak_ratio",
+    "delta_mean","power_density"
+    ]
+
+    df = pd.DataFrame([features], columns=columns)
+
+    X = scaler.transform(df)
     probs = model.predict_proba(X)[0]
 
     pred_index = np.argmax(probs)