How to stop the attack function if player leaves the check sphere?

The AI will find the player and once within range, attack. If the player leaves the attack range, the AI continues the shoot animation and just slides along the ground. I would like the AI to stop the attack and go back to chase. hey, I got this far with out asking a question.

 using UnityEngine;
 using UnityEngine.AI;
 
 
 public class EnemyAI : MonoBehaviour
 {
 
     public NavMeshAgent agent;
 
     public Transform player;
 
     public Transform alienBulletSpawn;
 
     private float damage = 10;
 
     public LayerMask whatIsGround, whatIsPlayer;
 
     private Animator anim;
 
     //States
     public float attackRange;
     public bool playerInAttackRange; 
 
     private void Awake()
     {
         player = GameObject.Find("Player").transform;
         agent = GetComponent<NavMeshAgent>();
         anim = GetComponent<Animator>();
         
     }
 
     private void Update()
     {
         anim.SetFloat("Speed", 1f);
         playerInAttackRange = Physics.CheckSphere(transform.position, attackRange, whatIsPlayer);
 
 
         if (!playerInAttackRange) EngagePlayer();
         if (playerInAttackRange) AttackPlayer();
         
     }
 
     void EngagePlayer()
     {
         agent.SetDestination(player.position);
 
     }
         
     void AttackPlayer()
     {
         agent.SetDestination(transform.position);
         transform.LookAt(player);
         anim.SetTrigger("Attack");
         
 
         RaycastHit hit;
         if (Physics.Raycast(alienBulletSpawn.transform.position, alienBulletSpawn.transform.forward, out hit))
         {   
             PlayerHealth playerHealth = hit.transform.GetComponent<PlayerHealth>();
             if (playerHealth != null)
             {
                 playerHealth.PlayerDamage(damage);
             }
         }
     }
 
     private void OnDrawGizmosSelected()
     {
         Gizmos.color = Color.red;
         Gizmos.DrawWireSphere(transform.position, attackRange);   
     }
 
 
 
 }