Spaces:
Sleeping
Sleeping
| import pandas as pd | |
| import joblib | |
| import gradio as gr | |
| import json | |
| import requests | |
| MODEL_PATH = "risk_model.joblib" | |
| ZAPIER_WEBHOOK_URL = "https://hooks.zapier.com/hooks/catch/27083565/uvtxiub/" | |
| DISCLAIMER = ( | |
| "Educational demo only. This tool does not provide diagnosis, " | |
| "prescriptions, or treatment plans. If symptoms are severe or urgent, seek emergency care." | |
| ) | |
| RED_FLAGS = [ | |
| "chest pain", | |
| "chest tightness", | |
| "severe shortness of breath", | |
| "fainting", | |
| "loss of consciousness", | |
| "stroke" | |
| ] | |
| risk_model = joblib.load(MODEL_PATH) | |
| def send_urgent_alert(symptom_text, risk_score): | |
| try: | |
| requests.post(ZAPIER_WEBHOOK_URL, json={ | |
| "triage_level": "urgent", | |
| "symptom": symptom_text, | |
| "risk_score": risk_score, | |
| "message": "URGENT triage case detected by HealthBridge AI" | |
| }, timeout=5) | |
| except Exception: | |
| pass | |
| def assess(sexvar, educa, genhlth, menthlth, stress, sleep_hours, sleep_quality, duration_days, adherence_prob, symptom_text): | |
| text_lower = str(symptom_text).lower() | |
| red_flag_hit = any(flag in text_lower for flag in RED_FLAGS) | |
| input_df = pd.DataFrame([{ | |
| "user_id": 0, | |
| "SEXVAR": sexvar, | |
| "EDUCA": educa, | |
| "GENHLTH": genhlth, | |
| "MENTHLTH": menthlth, | |
| "stress_score_0_10": stress, | |
| "sleep_hours": sleep_hours, | |
| "sleep_quality_0_10": sleep_quality, | |
| "symptom_duration_days": duration_days, | |
| "adherence_prob": adherence_prob, | |
| "symptom_text": symptom_text, | |
| "triage_label": "medium" | |
| }]) | |
| risk_score = float(risk_model.predict_proba(input_df)[0][1]) | |
| if red_flag_hit: | |
| triage_level = "urgent" | |
| elif risk_score >= 0.7: | |
| triage_level = "high" | |
| elif risk_score >= 0.4: | |
| triage_level = "medium" | |
| else: | |
| triage_level = "low" | |
| if triage_level == "urgent": | |
| send_urgent_alert(symptom_text, round(risk_score, 3)) | |
| if triage_level == "urgent": | |
| next_steps = [ | |
| "🚨 Seek emergency care immediately or call emergency services.", | |
| "Do not drive yourself — ask someone nearby to help.", | |
| "Do not wait to see if symptoms improve." | |
| ] | |
| elif triage_level == "high": | |
| next_steps = [ | |
| "Contact a healthcare provider as soon as possible.", | |
| "Monitor symptoms closely and note any changes.", | |
| "Seek emergency care if symptoms worsen rapidly." | |
| ] | |
| elif triage_level == "medium": | |
| next_steps = [ | |
| "Schedule an appointment with your doctor within the next few days.", | |
| "Rest and monitor your symptoms.", | |
| "Seek urgent care if symptoms escalate." | |
| ] | |
| else: | |
| next_steps = [ | |
| "Monitor symptoms and maintain healthy routines.", | |
| "Seek professional care if symptoms persist beyond a week.", | |
| "Stay hydrated and rest as needed." | |
| ] | |
| LEVEL_COLORS = { | |
| "urgent": "#FF2D2D", | |
| "high": "#FF8C00", | |
| "medium": "#F5C518", | |
| "low": "#2ECC71" | |
| } | |
| LEVEL_BG = { | |
| "urgent": "#FFF0F0", | |
| "high": "#FFF5E6", | |
| "medium": "#FFFBEA", | |
| "low": "#F0FFF6" | |
| } | |
| LEVEL_ICONS = { | |
| "urgent": "🚨", | |
| "high": "⚠️", | |
| "medium": "🔶", | |
| "low": "✅" | |
| } | |
| color = LEVEL_COLORS[triage_level] | |
| bg = LEVEL_BG[triage_level] | |
| icon = LEVEL_ICONS[triage_level] | |
| risk_pct = int(risk_score * 100) | |
| bar_color = color | |
| steps_html = "".join(f"<li style='margin-bottom:6px'>{s}</li>" for s in next_steps) | |
| alert_banner = "" | |
| if triage_level == "urgent": | |
| alert_banner = f""" | |
| <div style='background:#FF2D2D;color:white;padding:14px 18px;border-radius:8px; | |
| font-weight:700;font-size:15px;letter-spacing:0.5px;margin-bottom:16px; | |
| animation: pulse 1.5s infinite;'> | |
| 🚨 URGENT — An automated alert email has been sent. | |
| </div> | |
| """ | |
| html = f""" | |
| <style> | |
| @import url('https://fonts.googleapis.com/css2?family=DM+Sans:wght@400;500;600;700&family=DM+Mono:wght@400;500&display=swap'); | |
| .hb-card * {{ font-family: 'DM Sans', sans-serif; box-sizing: border-box; }} | |
| @keyframes pulse {{ | |
| 0%, 100% {{ opacity: 1; }} | |
| 50% {{ opacity: 0.75; }} | |
| }} | |
| @keyframes fadeIn {{ | |
| from {{ opacity: 0; transform: translateY(8px); }} | |
| to {{ opacity: 1; transform: translateY(0); }} | |
| }} | |
| .hb-card {{ animation: fadeIn 0.4s ease; }} | |
| </style> | |
| <div class='hb-card' style='background:{bg};border:2px solid {color};border-radius:12px; | |
| padding:24px;max-width:600px;'> | |
| {alert_banner} | |
| <!-- Triage Level Badge --> | |
| <div style='display:flex;align-items:center;gap:12px;margin-bottom:20px;'> | |
| <span style='font-size:32px;'>{icon}</span> | |
| <div> | |
| <div style='font-size:11px;font-weight:600;color:#888;letter-spacing:1.5px; | |
| text-transform:uppercase;'>Triage Level</div> | |
| <div style='font-size:28px;font-weight:700;color:{color};line-height:1.1;'> | |
| {triage_level.upper()} | |
| </div> | |
| </div> | |
| </div> | |
| <!-- Risk Score Bar --> | |
| <div style='margin-bottom:20px;'> | |
| <div style='display:flex;justify-content:space-between;margin-bottom:6px;'> | |
| <span style='font-size:12px;font-weight:600;color:#555;text-transform:uppercase; | |
| letter-spacing:1px;'>Risk Score</span> | |
| <span style='font-size:14px;font-weight:700;color:{color}; | |
| font-family:"DM Mono",monospace;'>{risk_pct}%</span> | |
| </div> | |
| <div style='background:#E8E8E8;border-radius:99px;height:10px;overflow:hidden;'> | |
| <div style='background:{bar_color};width:{risk_pct}%;height:100%; | |
| border-radius:99px;transition:width 0.6s ease;'></div> | |
| </div> | |
| </div> | |
| <!-- Red Flag --> | |
| <div style='display:flex;align-items:center;gap:8px;margin-bottom:20px; | |
| padding:10px 14px;background:white;border-radius:8px; | |
| border:1px solid #E0E0E0;'> | |
| <span style='font-size:16px;'>{"🚩" if red_flag_hit else "✔️"}</span> | |
| <span style='font-size:13px;font-weight:600;color:{"#FF2D2D" if red_flag_hit else "#2ECC71"};'> | |
| Red Flag: {"DETECTED" if red_flag_hit else "None detected"} | |
| </span> | |
| </div> | |
| <!-- Next Steps --> | |
| <div style='margin-bottom:20px;'> | |
| <div style='font-size:12px;font-weight:600;color:#555;letter-spacing:1px; | |
| text-transform:uppercase;margin-bottom:10px;'>Recommended Next Steps</div> | |
| <ul style='margin:0;padding-left:20px;color:#333;font-size:14px;line-height:1.6;'> | |
| {steps_html} | |
| </ul> | |
| </div> | |
| <!-- Disclaimer --> | |
| <div style='font-size:11px;color:#999;border-top:1px solid #E0E0E0; | |
| padding-top:12px;line-height:1.5;'> | |
| ⚠️ {DISCLAIMER} | |
| </div> | |
| </div> | |
| """ | |
| return html | |
| demo = gr.Interface( | |
| fn=assess, | |
| inputs=[ | |
| gr.Dropdown([1, 2], value=1, label="Sex (BRFSS-coded)"), | |
| gr.Slider(1, 6, value=4, step=1, label="Education level"), | |
| gr.Slider(1, 5, value=3, step=1, label="General health"), | |
| gr.Slider(0, 30, value=5, step=1, label="Mental health bad days"), | |
| gr.Slider(0, 10, value=5, step=0.1, label="Stress score"), | |
| gr.Slider(3, 10, value=7, step=0.1, label="Sleep hours"), | |
| gr.Slider(0, 10, value=7, step=0.1, label="Sleep quality"), | |
| gr.Slider(1, 30, value=5, step=1, label="Symptom duration (days)"), | |
| gr.Slider(0.05, 0.95, value=0.65, step=0.01, label="Adherence probability"), | |
| gr.Textbox(lines=4, placeholder="e.g. chest pain for 2 days, high stress, poor sleep", label="Symptom description") | |
| ], | |
| outputs=gr.HTML(label="Triage Result"), | |
| title="HealthBridge AI — Non-diagnostic Triage Demo", | |
| description="Educational demo only. Not a substitute for professional medical advice.", | |
| theme=gr.themes.Soft() | |
| ) | |
| demo.launch() |