Реализуйте переключение состояния квадрата по клику: при первом клике квадрат становится зелёным, при повторном клике — снова красным.

Храните одно булево состояние и переключайте его при клике, выбирая цвет по текущему значению.

<button id="square" aria-label="Переключить цвет" aria-pressed="false"></button>
<style>
  #square { width: 100px; height: 100px; background: red; }
  #square[aria-pressed="true"] { background: green; }
</style>
<script>
  const square = document.getElementById('square');
  square.addEventListener('click', () => {
    const active = square.getAttribute('aria-pressed') === 'true';
    square.setAttribute('aria-pressed', String(!active));
  });
</script>

Это фрагмент: style размещают в head, разметку и script — в body либо подключают внешний обработчик. Настоящая кнопка поддерживает клавиатурную активацию без отдельного div-click.