From c6de09562bb1e24ea00acba2c0467940f845b632 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Sat, 5 Apr 2025 21:01:48 +0300
Subject: Bug: two esc presses needed to exit edit page

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

diff --git a/tui.go b/tui.go
index 853dccf..cba30ab 100644
--- a/tui.go
+++ b/tui.go
@@ -294,7 +294,8 @@ func init() {
 		defer func() {
 			editMode = false
 		}()
-		if event.Key() == tcell.KeyEscape && editMode {
+		// if event.Key() == tcell.KeyEscape && editMode {
+		if event.Key() == tcell.KeyEscape {
 			defer colorText()
 			editedMsg := editArea.GetText()
 			if editedMsg == "" {
@@ -319,8 +320,8 @@ func init() {
 		SetDoneFunc(func(key tcell.Key) {
 			defer indexPickWindow.SetText("")
 			pages.RemovePage(indexPage)
-			colorText()
-			updateStatusLine()
+			// colorText()
+			// updateStatusLine()
 		})
 	indexPickWindow.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
 		switch event.Key() {
@@ -349,6 +350,7 @@ func init() {
 			}
 			m := chatBody.Messages[selectedIndex]
 			if editMode && event.Key() == tcell.KeyEnter {
+				pages.RemovePage(indexPage)
 				pages.AddPage(editMsgPage, editArea, true, true)
 				editArea.SetText(m.Content, true)
 			}
-- 
cgit v1.2.3