Check the code

(editMode = true)} >

 codeElement && codeElement.focus()}
    on:keypress={() => codeElement && codeElement.focus()}>
  {#if editMode}
      
    {:else}
      {code}
    {/if}