361 lines
9.7 KiB
Go
361 lines
9.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
|
|
"elefant/models"
|
|
"elefant/pngmeta"
|
|
"elefant/rag"
|
|
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
)
|
|
|
|
func makeChatTable(chatMap map[string]models.Chat) *tview.Table {
|
|
actions := []string{"load", "rename", "delete", "update card"}
|
|
chatList := make([]string, len(chatMap))
|
|
i := 0
|
|
for name := range chatMap {
|
|
chatList[i] = name
|
|
i++
|
|
}
|
|
rows, cols := len(chatMap), len(actions)+2
|
|
chatActTable := tview.NewTable().
|
|
SetBorders(true)
|
|
// for chatName, chat := range chatMap {
|
|
for r := 0; r < rows; r++ {
|
|
// r := 0
|
|
for c := 0; c < cols; c++ {
|
|
color := tcell.ColorWhite
|
|
switch c {
|
|
case 0:
|
|
chatActTable.SetCell(r, c,
|
|
tview.NewTableCell(chatList[r]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
case 1:
|
|
chatActTable.SetCell(r, c,
|
|
tview.NewTableCell(chatMap[chatList[r]].Msgs[len(chatMap[chatList[r]].Msgs)-30:]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
default:
|
|
chatActTable.SetCell(r, c,
|
|
tview.NewTableCell(actions[c-2]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
}
|
|
}
|
|
// r++
|
|
}
|
|
chatActTable.Select(0, 0).SetFixed(1, 1).SetDoneFunc(func(key tcell.Key) {
|
|
if key == tcell.KeyEsc || key == tcell.KeyF1 {
|
|
pages.RemovePage(historyPage)
|
|
return
|
|
}
|
|
if key == tcell.KeyEnter {
|
|
chatActTable.SetSelectable(true, true)
|
|
}
|
|
}).SetSelectedFunc(func(row int, column int) {
|
|
tc := chatActTable.GetCell(row, column)
|
|
tc.SetTextColor(tcell.ColorRed)
|
|
chatActTable.SetSelectable(false, false)
|
|
selectedChat := chatList[row]
|
|
defer pages.RemovePage(historyPage)
|
|
// notification := fmt.Sprintf("chat: %s; action: %s", selectedChat, tc.Text)
|
|
switch tc.Text {
|
|
case "load":
|
|
history, err := loadHistoryChat(selectedChat)
|
|
if err != nil {
|
|
logger.Error("failed to read history file", "chat", selectedChat)
|
|
pages.RemovePage(historyPage)
|
|
return
|
|
}
|
|
chatBody.Messages = history
|
|
textView.SetText(chatToText(cfg.ShowSys))
|
|
activeChatName = selectedChat
|
|
pages.RemovePage(historyPage)
|
|
return
|
|
case "rename":
|
|
pages.RemovePage(historyPage)
|
|
pages.AddPage(renamePage, renameWindow, true, true)
|
|
return
|
|
case "delete":
|
|
sc, ok := chatMap[selectedChat]
|
|
if !ok {
|
|
// no chat found
|
|
pages.RemovePage(historyPage)
|
|
return
|
|
}
|
|
if err := store.RemoveChat(sc.ID); err != nil {
|
|
logger.Error("failed to remove chat from db", "chat_id", sc.ID, "chat_name", sc.Name)
|
|
}
|
|
if err := notifyUser("chat deleted", selectedChat+" was deleted"); err != nil {
|
|
logger.Error("failed to send notification", "error", err)
|
|
}
|
|
// load last chat
|
|
chatBody.Messages = loadOldChatOrGetNew()
|
|
textView.SetText(chatToText(cfg.ShowSys))
|
|
pages.RemovePage(historyPage)
|
|
return
|
|
case "update card":
|
|
// save updated card
|
|
fi := strings.Index(selectedChat, "_")
|
|
agentName := selectedChat[fi+1:]
|
|
cc, ok := sysMap[agentName]
|
|
if !ok {
|
|
logger.Warn("no such card", "agent", agentName)
|
|
//no:lint
|
|
notifyUser("error", "no such card: "+agentName)
|
|
}
|
|
if err := pngmeta.WriteToPng(cc.ToSpec(cfg.UserRole), cc.FilePath, cc.FilePath); err != nil {
|
|
logger.Error("failed to write charcard",
|
|
"error", err)
|
|
}
|
|
// pages.RemovePage(historyPage)
|
|
return
|
|
default:
|
|
// pages.RemovePage(historyPage)
|
|
return
|
|
}
|
|
})
|
|
return chatActTable
|
|
}
|
|
|
|
// func makeRAGTable(fileList []string) *tview.Table {
|
|
func makeRAGTable(fileList []string) *tview.Flex {
|
|
actions := []string{"load", "delete"}
|
|
rows, cols := len(fileList), len(actions)+1
|
|
fileTable := tview.NewTable().
|
|
SetBorders(true)
|
|
longStatusView := tview.NewTextView()
|
|
longStatusView.SetText("status text")
|
|
longStatusView.SetBorder(true).SetTitle("status")
|
|
longStatusView.SetChangedFunc(func() {
|
|
app.Draw()
|
|
})
|
|
ragflex := tview.NewFlex().SetDirection(tview.FlexRow).
|
|
AddItem(longStatusView, 0, 10, false).
|
|
AddItem(fileTable, 0, 60, true)
|
|
for r := 0; r < rows; r++ {
|
|
for c := 0; c < cols; c++ {
|
|
color := tcell.ColorWhite
|
|
if c < 1 {
|
|
fileTable.SetCell(r, c,
|
|
tview.NewTableCell(fileList[r]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
} else {
|
|
fileTable.SetCell(r, c,
|
|
tview.NewTableCell(actions[c-1]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
}
|
|
}
|
|
}
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
defer pages.RemovePage(RAGPage)
|
|
for {
|
|
select {
|
|
case err := <-errCh:
|
|
if err == nil {
|
|
logger.Error("somehow got a nil err", "error", err)
|
|
continue
|
|
}
|
|
logger.Error("got an err in rag status", "error", err, "textview", longStatusView)
|
|
longStatusView.SetText(fmt.Sprintf("%v", err))
|
|
close(errCh)
|
|
return
|
|
case status := <-rag.LongJobStatusCh:
|
|
logger.Info("reading status channel", "status", status)
|
|
longStatusView.SetText(status)
|
|
// fmt.Fprintln(longStatusView, status)
|
|
// app.Sync()
|
|
if status == rag.FinishedRAGStatus {
|
|
close(errCh)
|
|
time.Sleep(2 * time.Second)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
fileTable.Select(0, 0).SetFixed(1, 1).SetDoneFunc(func(key tcell.Key) {
|
|
if key == tcell.KeyEsc || key == tcell.KeyF1 {
|
|
pages.RemovePage(RAGPage)
|
|
return
|
|
}
|
|
if key == tcell.KeyEnter {
|
|
fileTable.SetSelectable(true, true)
|
|
}
|
|
}).SetSelectedFunc(func(row int, column int) {
|
|
// defer pages.RemovePage(RAGPage)
|
|
tc := fileTable.GetCell(row, column)
|
|
tc.SetTextColor(tcell.ColorRed)
|
|
fileTable.SetSelectable(false, false)
|
|
fpath := fileList[row]
|
|
// notification := fmt.Sprintf("chat: %s; action: %s", fpath, tc.Text)
|
|
switch tc.Text {
|
|
case "load":
|
|
fpath = path.Join(cfg.RAGDir, fpath)
|
|
longStatusView.SetText("clicked load")
|
|
go func() {
|
|
if err := ragger.LoadRAG(fpath); err != nil {
|
|
logger.Error("failed to embed file", "chat", fpath, "error", err)
|
|
errCh <- err
|
|
// pages.RemovePage(RAGPage)
|
|
return
|
|
}
|
|
}()
|
|
return
|
|
case "delete":
|
|
fpath = path.Join(cfg.RAGDir, fpath)
|
|
if err := os.Remove(fpath); err != nil {
|
|
logger.Error("failed to delete file", "filename", fpath, "error", err)
|
|
return
|
|
}
|
|
if err := notifyUser("chat deleted", fpath+" was deleted"); err != nil {
|
|
logger.Error("failed to send notification", "error", err)
|
|
}
|
|
return
|
|
default:
|
|
return
|
|
}
|
|
})
|
|
return ragflex
|
|
}
|
|
|
|
func makeLoadedRAGTable(fileList []string) *tview.Table {
|
|
actions := []string{"delete"}
|
|
rows, cols := len(fileList), len(actions)+1
|
|
fileTable := tview.NewTable().
|
|
SetBorders(true)
|
|
for r := 0; r < rows; r++ {
|
|
for c := 0; c < cols; c++ {
|
|
color := tcell.ColorWhite
|
|
if c < 1 {
|
|
fileTable.SetCell(r, c,
|
|
tview.NewTableCell(fileList[r]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
} else {
|
|
fileTable.SetCell(r, c,
|
|
tview.NewTableCell(actions[c-1]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
}
|
|
}
|
|
}
|
|
fileTable.Select(0, 0).SetFixed(1, 1).SetDoneFunc(func(key tcell.Key) {
|
|
if key == tcell.KeyEsc || key == tcell.KeyF1 {
|
|
pages.RemovePage(RAGPage)
|
|
return
|
|
}
|
|
if key == tcell.KeyEnter {
|
|
fileTable.SetSelectable(true, true)
|
|
}
|
|
}).SetSelectedFunc(func(row int, column int) {
|
|
defer pages.RemovePage(RAGPage)
|
|
tc := fileTable.GetCell(row, column)
|
|
tc.SetTextColor(tcell.ColorRed)
|
|
fileTable.SetSelectable(false, false)
|
|
fpath := fileList[row]
|
|
// notification := fmt.Sprintf("chat: %s; action: %s", fpath, tc.Text)
|
|
switch tc.Text {
|
|
case "delete":
|
|
if err := ragger.RemoveFile(fpath); err != nil {
|
|
logger.Error("failed to delete file", "filename", fpath, "error", err)
|
|
return
|
|
}
|
|
if err := notifyUser("chat deleted", fpath+" was deleted"); err != nil {
|
|
logger.Error("failed to send notification", "error", err)
|
|
}
|
|
return
|
|
default:
|
|
// pages.RemovePage(RAGPage)
|
|
return
|
|
}
|
|
})
|
|
return fileTable
|
|
}
|
|
|
|
func makeAgentTable(agentList []string) *tview.Table {
|
|
actions := []string{"load"}
|
|
rows, cols := len(agentList), len(actions)+1
|
|
chatActTable := tview.NewTable().
|
|
SetBorders(true)
|
|
for r := 0; r < rows; r++ {
|
|
for c := 0; c < cols; c++ {
|
|
color := tcell.ColorWhite
|
|
if c < 1 {
|
|
chatActTable.SetCell(r, c,
|
|
tview.NewTableCell(agentList[r]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
} else {
|
|
chatActTable.SetCell(r, c,
|
|
tview.NewTableCell(actions[c-1]).
|
|
SetTextColor(color).
|
|
SetAlign(tview.AlignCenter))
|
|
}
|
|
}
|
|
}
|
|
chatActTable.Select(0, 0).SetFixed(1, 1).SetDoneFunc(func(key tcell.Key) {
|
|
if key == tcell.KeyEsc || key == tcell.KeyF1 {
|
|
pages.RemovePage(agentPage)
|
|
return
|
|
}
|
|
if key == tcell.KeyEnter {
|
|
chatActTable.SetSelectable(true, true)
|
|
}
|
|
}).SetSelectedFunc(func(row int, column int) {
|
|
tc := chatActTable.GetCell(row, column)
|
|
tc.SetTextColor(tcell.ColorRed)
|
|
chatActTable.SetSelectable(false, false)
|
|
selected := agentList[row]
|
|
// notification := fmt.Sprintf("chat: %s; action: %s", selectedChat, tc.Text)
|
|
switch tc.Text {
|
|
case "load":
|
|
if ok := charToStart(selected); !ok {
|
|
logger.Warn("no such sys msg", "name", selected)
|
|
pages.RemovePage(agentPage)
|
|
return
|
|
}
|
|
// replace textview
|
|
textView.SetText(chatToText(cfg.ShowSys))
|
|
colorText()
|
|
updateStatusLine()
|
|
// sysModal.ClearButtons()
|
|
pages.RemovePage(agentPage)
|
|
app.SetFocus(textArea)
|
|
return
|
|
case "rename":
|
|
pages.RemovePage(agentPage)
|
|
pages.AddPage(renamePage, renameWindow, true, true)
|
|
return
|
|
case "delete":
|
|
sc, ok := chatMap[selected]
|
|
if !ok {
|
|
// no chat found
|
|
pages.RemovePage(agentPage)
|
|
return
|
|
}
|
|
if err := store.RemoveChat(sc.ID); err != nil {
|
|
logger.Error("failed to remove chat from db", "chat_id", sc.ID, "chat_name", sc.Name)
|
|
}
|
|
if err := notifyUser("chat deleted", selected+" was deleted"); err != nil {
|
|
logger.Error("failed to send notification", "error", err)
|
|
}
|
|
pages.RemovePage(agentPage)
|
|
return
|
|
default:
|
|
pages.RemovePage(agentPage)
|
|
return
|
|
}
|
|
})
|
|
return chatActTable
|
|
}
|