Update predict_impulse.py

predict_impulse.py

@@ -1,9 +1,13 @@
 import joblib
 import pandas as pd
 from huggingface_hub import hf_hub_download
-import opik
 
-# Download model from HF Model repo
+# Opik (correct usage)
+from opik import Opik
+
+# -----------------------------
+# Load model from HF Model repo
+# -----------------------------
 MODEL_REPO = "obx0x3/sensei-model"
 MODEL_FILE = "impulse_model.pkl"
 
@@ -14,6 +18,15 @@ model_path = hf_hub_download(
 
 impulse_model = joblib.load(model_path)
 
+# -----------------------------
+# Initialize Opik client
+# -----------------------------
+try:
+    opik_client = Opik(project_name="budgetbuddy-hackathon")
+except Exception:
+    opik_client = None
+
+
 def predict_impulse(category, amount, payment_method, day):
     df = pd.DataFrame([{
         "category": category,
@@ -30,12 +43,22 @@ def predict_impulse(category, amount, payment_method, day):
         "confidence": round(float(prob), 3)
     }
 
-    # ✅ Correct Opik usage (NO init)
-    opik.log(
-        name="Impulse Prediction",
-        input=df.to_dict(orient="records")[0],
-        output=result,
-        model="sensei-impulse-model"
-    )
+    # -----------------------------
+    # Opik logging (SAFE)
+    # -----------------------------
+    if opik_client:
+        try:
+            opik_client.log_event(
+                name="Impulse Prediction",
+                input=df.to_dict(orient="records")[0],
+                output=result,
+                model="sensei-impulse-model",
+                metadata={
+                    "source": "hf-space",
+                    "task": "impulse-detection"
+                }
+            )
+        except Exception:
+            pass  # Never crash inference
 
     return result