Update app.py

app.py

@@ -32,6 +32,7 @@ class Features(BaseModel):
     range: float
     peak_count: int
     slope: float
+    power: float
 
 # ---------------- FEATURE ENGINEERING ----------------
 def transform_features(data: Features):
@@ -86,8 +87,8 @@ def predict(data: Features):
         appliance = "Unknown"
 
     # ---------------- ENERGY CALC ----------------
-    time_seconds = 40 * 0.04   # 1.6 sec
-    energy = (data.mean * time_seconds) / 3600  # Wh
+    time_seconds = 40 * 0.04   # 40 samples × 40ms = 1.6 sec
+    energy = (data.power * time_seconds) / 3600.0
 
     # ---------------- STORE IN SUPABASE ----------------
     supabase.table("energy_logs").insert({
@@ -102,6 +103,7 @@ def predict(data: Features):
         "peak_count": data.peak_count,
         "slope": data.slope,
         "energy": energy
+        "power": data.power
     }).execute()
 
     return {