Spaces:
Sleeping
Sleeping
File size: 2,268 Bytes
f6634f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
# rules_engine.py
def compute_bmi(height_cm, weight_kg):
"""Compute BMI = weight (kg) / height (m^2)."""
if height_cm <= 0:
return 0.0
h_m = height_cm / 100
return round(weight_kg / (h_m * h_m), 2)
def evaluate_risks(age, bmi, glucose, sbp, dbp, hr=70, spo2=98):
"""Simple numeric risk scoring for conditions."""
risks = {}
# Type 2 Diabetes
risks["Type 2 Diabetes Risk"] = min(1.0, (glucose - 90) / 100 + bmi / 40)
# Hypertension
risks["Hypertension Risk"] = min(1.0, (sbp - 120) / 60 + (dbp - 80) / 40)
# Depression / Mood
risks["Depression/Mood Concern"] = 0.3 if age > 40 else 0.2
# Migraine
risks["Migraine Risk"] = 0.4 if sbp > 135 else 0.2
# Sleep Apnea
risks["Sleep Apnea Risk"] = 0.5 if bmi > 30 else 0.2
# Anemia
risks["Anemia Risk"] = 0.3 if spo2 < 92 else 0.1
return risks
def rules_risk_assessment(age, sex, sbp, dbp, hr, spo2, glucose, bmi, symptoms):
"""
Return (scores, explanations).
scores: dict of condition → 0–1 float
explanations: dict of condition → reason string
"""
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)
|