STATE_RULES = {
    "WAITING_USER_TYPE": {
        "expected": ["new_student", "current_student"],
        "supports_topic_switch": True,
    },
    "WAITING_AUDIENCE": {
        "expected": ["adults", "children"],
        "supports_topic_switch": True,
    },
    "WAITING_PRIOR_STUDY": {
        "expected": ["prior_study_yes", "prior_study_no"],
        "supports_topic_switch": True,
    },
    "WAITING_BEGINNER_SCHEDULE_CHOICE": {
        "expected": ["confirm_schedule_reviewed", "proceed_booking"],
        "state_switches": ["switch_to_prior_study_true", "switch_to_prior_study_false", "support_needed"],
        "supports_topic_switch": True,
    },
    "WAITING_PDF_102_CONFIRMATION": {
        "expected": ["confirm_pdf_reviewed"],
        "state_switches": ["switch_to_prior_study_true", "switch_to_prior_study_false", "support_needed"],
        "supports_topic_switch": True,
    },
    "WAITING_PLACEMENT_TEST_CONFIRMATION": {
        "expected": ["confirm_placement_test_reviewed"],
        "state_switches": ["switch_to_prior_study_true", "switch_to_prior_study_false", "support_needed"],
        "supports_topic_switch": True,
    },
    "WAITING_CURRENT_STUDENT_ACTION": {
        "expected": ["current_student_support", "current_student_next_level"],
        "supports_topic_switch": True,
    },
    "WAITING_SUPPORT_QUESTION": {
        "expected": ["support_question_text"],
        "supports_topic_switch": True,
    },
    "WAITING_LEVEL_SELECTION": {
        "expected": ["level_selected"],
        "state_switches": ["support_needed"],
        "supports_topic_switch": True,
    },
    "WAITING_PAYMENT_METHOD": {
        "expected": ["payment_method_selected"],
        "state_switches": ["support_needed"],
        "supports_topic_switch": True,
    },
    "WAITING_COMPLAINT_FORM": {
        "expected": ["complaint_form_submitted"],
        "supports_topic_switch": True,
    },
    "HANDOFF_DONE": {
        "expected": ["thanks"],
        "supports_topic_switch": True,
    },
}