Update heart_utils.py

heart_utils.py

@@ -1,10 +1,21 @@
 import joblib
 import numpy as np
 
-model = joblib.load("heart_model.joblib")
+# Load the trained heart disease model
+heart_model = joblib.load("heart_model.joblib")
 
-def predict_heart(age, sex, cp, trestbps, chol):
-    sex = 1 if sex == "Male" else 0
-    data = np.array([[age, sex, cp, trestbps, chol]])
-    prediction = model.predict(data)
-    return "High Risk" if prediction[0] == 1 else "Low Risk"
+def encode_sex(sex):
+    return 1 if sex == "Male" else 0
+
+def predict_heart(age, sex, cp, trestbps, chol, fbs, restecg,
+                  thalach, exang, oldpeak, slope, ca, thal):
+    # Encode 'sex' from string to numeric
+    sex_encoded = encode_sex(sex)
+
+    # Create feature array (13 features)
+    features = np.array([[age, sex_encoded, cp, trestbps, chol, fbs, restecg,
+                          thalach, exang, oldpeak, slope, ca, thal]])
+
+    # Predict using the loaded model
+    prediction = heart_model.predict(features)[0]
+    return "Heart Disease Detected" if prediction == 1 else "No Heart Disease Detected"