Update app.py

app.py

@@ -35,6 +35,7 @@ hemoglobin_model = joblib.load("hemoglobin_model_from_anemia_dataset.pkl")
 hemoglobin_r2 = 0.385
 import joblib
 spo2_model = joblib.load("spo2_model_simulated.pkl")
+hr_model = joblib.load("heart_rate_model.pkl")
 
 models = {
     "Hemoglobin": hemoglobin_model,
@@ -100,7 +101,12 @@ def analyze_face(image):
             value = models[label].predict([[random.uniform(0.2, 0.5) for _ in range(7)]])[0]
             test_values[label] = value
             r2_scores[label] = 0.0  # simulate other 7D inputs
-    heart_rate = int(60 + 30 * np.sin(np.mean(frame_rgb) / 255.0 * np.pi))
+    gray = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2GRAY)
+    green_std = np.std(frame_rgb[:, :, 1]) / 255
+    brightness_std = np.std(gray) / 255
+    tone_index = np.mean(frame_rgb[100:150, 100:150]) / 255 if frame_rgb[100:150, 100:150].size else 0.5
+    hr_features = [brightness_std, green_std, tone_index]
+    heart_rate = hr_model.predict([hr_features])[0]
     skin_patch = frame_rgb[100:150, 100:150]
     skin_tone_index = np.mean(skin_patch) / 255 if skin_patch.size else 0.5
     brightness_variation = np.std(cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2GRAY)) / 255