ボタンごとの処理を変更していましたが、前よりも複雑になっていると思ったので、一旦元の状態に戻しました。

先ほど実機で動かしていると、手書き入力の際のキーボードが閉じれないことが分かったので、閉じる処理を追加しようとしています。後は、画像から入力されたテキストを手書きで修正できるようにしたいと考えています。