Confused on App Notification Actions

Trying to automate my coffee machine.
Essentially got a powerplug turning off and on the machine.

Flow:
Turn on machine
Timer starts imediately countdown 45min
Once countdown finished
send notification with keep on or turn off
if off then turn virtual toggle off and thus machine off
if on then reset timer (flow action 2 here)
and replay automation

currently having trouble with the reset timer and replay automation section

id: '1732972208236'
alias: Coffee Off Test
description: ''
triggers:
  - trigger: state
    entity_id:
      - switch.coffee_power
    from: null
    to: 'on'
conditions: []
actions:
  - sequence:
      - action: timer.change
        metadata: {}
        data:
          duration: '00:45:00'
        target:
          entity_id: timer.coffee_timer
        enabled: false
      - action: timer.start
        metadata: {}
        data:
          duration: '00:45:00'
        target:
          entity_id: timer.coffee_timer
      - choose:
          - conditions:
              - condition: zone
                entity_id: device_tracker.sm_g998b
                zone: zone.home
              - condition: state
                entity_id: timer.coffee_timer
                state: idle
            sequence:
              - action: notify.mobile_app_sm_g998b
                metadata: {}
                data:
                  message: Turn Off Coffee Machine?
                  title: Coffee Machine
                  data:
                    actions:
                      - action: COFFEE_OFF
                        title: 'Yes'
                        icon: sfsymbols:tick
                      - action: COFFEE_WAIT
                        title: 'No'
                        icon: sfsymbols:cross
              - wait_for_trigger:
                  - trigger: event
                    event_type: mobile_app_notification_action
                    event_data:
                      action: '{{ COFFEE_OFF }}'
                  - trigger: event
                    event_type: mobile_app_notification_action
                    event_data:
                      action: '{{ COFFEE_WAIT }}'
                timeout:
                  hours: 0
                  minutes: 2
                  seconds: 0
                  milliseconds: 0
              - choose:
                  - conditions:
                      - condition: template
                        value_template: '{{ wait.trigger.event.data.action == COFFEE_OFF }}'
                    sequence:
                      - type: turn_off
                        device_id: c0f02eb5628071de0d9b5775108479cb
                        entity_id: 37f81f7950b26fc06c0c508665e69313
                        domain: switch
                  - conditions:
                      - condition: template
                        value_template: '{{ wait.trigger.event.data.action == COFFEE_WAIT }}'
                    sequence:
                      - action: automation.trigger
                        metadata: {}
                        data:
                          skip_condition: true
                        target:
                          entity_id: automation.coffee_off_test
mode: single

thanks for any help