Update rules_engine.py

rules_engine.py

@@ -1,91 +1,69 @@
 # rules_engine.py
 
-def compute_bmi(weight_kg, height_cm):
-    """Compute Body Mass Index (BMI)."""
+def compute_bmi(height_cm, weight_kg):
+    """Compute BMI = weight (kg) / height (m^2)."""
     if height_cm <= 0:
-        return 0
-    height_m = height_cm / 100
-    return round(weight_kg / (height_m ** 2), 2)
+        return 0.0
+    h_m = height_cm / 100
+    return round(weight_kg / (h_m * h_m), 2)
 
 
-def evaluate_risks(age, bmi, glucose, systolic_bp, diastolic_bp, cholesterol, hemoglobin):
-    """Return a dictionary of risk scores (0–100)."""
+def evaluate_risks(age, bmi, glucose, sbp, dbp, hr=70, spo2=98):
+    """Simple numeric risk scoring for conditions."""
     risks = {}
 
-    # Type 2 Diabetes Risk
-    risks["Type 2 Diabetes Risk"] = min(100, 0.2 * glucose + 0.3 * bmi + 0.1 * age)
+    # Type 2 Diabetes
+    risks["Type 2 Diabetes Risk"] = min(1.0, (glucose - 90) / 100 + bmi / 40)
 
-    # Hypertension Risk
-    risks["Hypertension Risk"] = min(100, 0.25 * systolic_bp + 0.15 * diastolic_bp + 0.1 * age)
+    # Hypertension
+    risks["Hypertension Risk"] = min(1.0, (sbp - 120) / 60 + (dbp - 80) / 40)
 
-    # Depression/Mood Concern
-    risks["Depression/Mood Concern"] = min(100, 0.05 * age + 0.05 * bmi)
+    # Depression / Mood
+    risks["Depression/Mood Concern"] = 0.3 if age > 40 else 0.2
 
-    # Migraine Risk
-    risks["Migraine Risk"] = min(100, 0.05 * age + 0.1 * systolic_bp)
+    # Migraine
+    risks["Migraine Risk"] = 0.4 if sbp > 135 else 0.2
 
-    # Sleep Apnea Risk
-    risks["Sleep Apnea Risk"] = min(100, 0.3 * bmi + 0.05 * age)
+    # Sleep Apnea
+    risks["Sleep Apnea Risk"] = 0.5 if bmi > 30 else 0.2
 
-    # Anemia Risk
-    if hemoglobin < 13.5:
-        risks["Anemia Risk"] = min(100, (13.5 - hemoglobin) * 15 + 0.1 * age)
-    else:
-        risks["Anemia Risk"] = 0
+    # Anemia
+    risks["Anemia Risk"] = 0.3 if spo2 < 92 else 0.1
 
     return risks
 
-# rules_engine.py
-
-# rules_engine.py
-
-def compute_bmi(weight, height):
-    """Compute BMI given weight (kg) and height (cm)."""
-    if height <= 0:
-        return None
-    return round(weight / ((height / 100) ** 2), 2)
-
 
-def rules_risk_assessment(age, sex, smoking, diabetes, cholesterol, sbp, bmi):
+def rules_risk_assessment(age, sex, sbp, dbp, hr, spo2, glucose, bmi, symptoms):
     """
-    Simple risk scoring system.
+    Return (scores, explanations).
+    scores: dict of condition → 0–1 float
+    explanations: dict of condition → reason string
     """
-    risk_score = 0
-
-    # Age factor
-    if age > 50:
-        risk_score += 2
-    elif age > 30:
-        risk_score += 1
-
-    # Sex factor
-    if sex.lower() == "male":
-        risk_score += 1
-
-    # Smoking
-    if smoking:
-        risk_score += 2
-
-    # Diabetes
-    if diabetes:
-        risk_score += 2
-
-    # Cholesterol
-    if cholesterol > 200:
-        risk_score += 1
-
-    # Systolic BP
-    if sbp > 140:
-        risk_score += 2
-
-    # BMI
-    if bmi and bmi > 30:
-        risk_score += 1
-
-    # Categorize
-    if risk_score <= 2:
-        return "Low Risk"
-    elif risk_score <= 5:
-        return "Moderate Risk"
-    else:
-        return "High Risk"
+    scores = {}
+    explain = {}
+
+    risks = evaluate_risks(age, bmi, glucose, sbp, dbp, hr, spo2)
+    for cond, val in risks.items():
+        scores[cond] = round(val, 3)
+        if cond == "Type 2 Diabetes Risk":
+            explain[cond] = f"Glucose={glucose}, BMI={bmi}"
+        elif cond == "Hypertension Risk":
+            explain[cond] = f"SBP={sbp}, DBP={dbp}"
+        elif cond == "Depression/Mood Concern":
+            explain[cond] = f"Age={age}, Sex={sex}"
+        elif cond == "Migraine Risk":
+            explain[cond] = "Headache symptom" if "Headache" in symptoms else "Vitals"
+        elif cond == "Sleep Apnea Risk":
+            explain[cond] = "Snoring / daytime sleepiness" if any(s in symptoms for s in ["Loud snoring", "Daytime sleepiness"]) else "BMI factor"
+        elif cond == "Anemia Risk":
+            explain[cond] = f"SpO₂={spo2}"
+        else:
+            explain[cond] = "Rule-based heuristic"
+
+    return scores, explain
+
+
+def explain_rules(explain_dict):
+    """Format explanation dict into readable text."""
+    lines = [f"- **{cond}**: {reason}" for cond, reason in explain_dict.items()]
+    return "\n".join(lines)