From 4a9c90af3f3b8c9d5d382892b8f3392df0b5d212 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Sat, 5 Apr 2025 21:52:10 +0300
Subject: Fix: esc on edit mode

---
 tui.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tui.go b/tui.go
index cba30ab..91c042a 100644
--- a/tui.go
+++ b/tui.go
@@ -291,11 +291,9 @@ func init() {
 		SetPlaceholder("Replace msg...")
 	editArea.SetBorder(true).SetTitle("input")
 	editArea.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
-		defer func() {
-			editMode = false
-		}()
 		// if event.Key() == tcell.KeyEscape && editMode {
 		if event.Key() == tcell.KeyEscape {
+			logger.Warn("edit debug; esc is pressed")
 			defer colorText()
 			editedMsg := editArea.GetText()
 			if editedMsg == "" {
@@ -309,6 +307,7 @@ func init() {
 			// change textarea
 			textView.SetText(chatToText(cfg.ShowSys))
 			pages.RemovePage(editMsgPage)
+			editMode = false
 			return nil
 		}
 		return event
-- 
cgit v1.2.3