app/rules/engine.py

from typing import Any, Dict
import os

from app.rules import ecg_rules

USE_ENHANCED_RULES = os.getenv("USE_ENHANCED_RULES", "true").lower() == "true"

if USE_ENHANCED_RULES:
    try:
        from app.rules.medical_rules import evaluate_medical_rules
        _evaluator = evaluate_medical_rules
    except ImportError:
        _evaluator = None
        USE_ENHANCED_RULES = False
else:
    _evaluator = None


def evaluate_ecg_rules(
    patient_context: Dict[str, Any],
    model_output: Dict[str, Any],
) -> Dict[str, Any]:
    if USE_ENHANCED_RULES and _evaluator:
        return _evaluator(patient_context, model_output)

    result = ecg_rules.apply_rules(patient_context, model_output)
    return {"alert_level": result.get("alert_level", "none"), "explanations": result.get("explanations", [])}