from dataclasses import dataclass
from enum import Enum

class ERDecision(Enum):
    AUTO_MERGE = "auto_merge"
    REVIEW_QUEUE = "review_queue"
    AUTO_REJECT = "auto_reject"

@dataclass
class ThresholdBands:
    auto_merge_floor: float = 0.95   # above this: auto-merge
    review_floor: float = 0.70       # between review_floor and auto_merge_floor: review queue
    # below review_floor: auto-reject

def classify_match(probability: float, bands: ThresholdBands) -> ERDecision:
    if probability >= bands.auto_merge_floor:
        return ERDecision.AUTO_MERGE
    elif probability >= bands.review_floor:
        return ERDecision.REVIEW_QUEUE
    else:
        return ERDecision.AUTO_REJECT