Enha: close rag on success

1 files changed, 7 insertions, 1 deletions

diff --git a/tables.go b/tables.go
index 23000f4..5f01fe3 100644
--- a/tables.go
+++ b/tables.go
@@ -387,9 +387,15 @@ func makeRAGTable(fileList []string) *tview.Flex {
 				if err := ragger.LoadRAG(fpath); err != nil {
 					logger.Error("failed to embed file", "chat", fpath, "error", err)
 					_ = notifyUser("RAG", "failed to embed file; error: "+err.Error())
-					pages.RemovePage(RAGPage)
+					app.QueueUpdate(func() {
+						pages.RemovePage(RAGPage)
+					})
 					return
 				}
+				_ = notifyUser("RAG", "file loaded successfully")
+				app.QueueUpdate(func() {
+					pages.RemovePage(RAGPage)
+				})
 			}()
 			return
 		case "delete":