diff options
| -rw-r--r-- | models/models.go | 2 | ||||
| -rw-r--r-- | session.go | 9 | ||||
| -rw-r--r-- | tui.go | 23 |
3 files changed, 30 insertions, 4 deletions
diff --git a/models/models.go b/models/models.go index 2ac087a..82ff63d 100644 --- a/models/models.go +++ b/models/models.go @@ -69,7 +69,7 @@ func (m RoleMsg) ToText(i int) string { default: icon = fmt.Sprintf("(%d) <%s>: ", i, m.Role) } - textMsg := fmt.Sprintf("%s\n%s\n", icon, m.Content) + textMsg := fmt.Sprintf("[-:-:u]%s[-:-:-]\n%s\n", icon, m.Content) return strings.ReplaceAll(textMsg, "\n\n", "\n") } @@ -5,6 +5,7 @@ import ( "encoding/json" "errors" "fmt" + "os" "os/exec" "strings" "time" @@ -25,6 +26,14 @@ func historyToSJSON(msgs []models.RoleMsg) (string, error) { return string(data), nil } +func exportChat() error { + data, err := json.MarshalIndent(chatBody.Messages, "", " ") + if err != nil { + return err + } + return os.WriteFile(activeChatName+".json", data, 0666) +} + func updateStorageChat(name string, msgs []models.RoleMsg) error { var err error chat, ok := chatMap[name] @@ -37,6 +37,7 @@ var ( [yellow]F7[white]: copy last msg to clipboard (linux xclip) [yellow]F8[white]: copy n msg to clipboard (linux xclip) [yellow]Ctrl+s[white]: choose/replace system prompt +[yellow]Ctrl+e[white]: export chat to json file Press Enter to go back ` @@ -168,6 +169,7 @@ func init() { applyCharCard(cc) // replace textview textView.SetText(chatToText(cfg.ShowSys)) + colorText() sysModal.ClearButtons() pages.RemovePage("sys") app.SetFocus(textArea) @@ -319,10 +321,12 @@ func init() { if strings.HasSuffix(text, cfg.AssistantIcon) { logger.Info("deleting assistant icon", "icon", cfg.AssistantIcon) textView.SetText(strings.TrimSuffix(text, cfg.AssistantIcon)) + colorText() return nil } chatBody.Messages = chatBody.Messages[:len(chatBody.Messages)-1] textView.SetText(chatToText(cfg.ShowSys)) + colorText() return nil } if event.Key() == tcell.KeyF4 { @@ -335,6 +339,7 @@ func init() { // switch cfg.ShowSys cfg.ShowSys = !cfg.ShowSys textView.SetText(chatToText(cfg.ShowSys)) + colorText() } if event.Key() == tcell.KeyF6 { interruptResp = true @@ -370,8 +375,12 @@ func init() { return nil } if event.Key() == tcell.KeyCtrlE { - textArea.SetText("pressed ctrl+e", true) - colorText() + // export loaded chat into json file + if err := exportChat(); err != nil { + logger.Error("failed to export chat;", "error", err, "chat_name", activeChatName) + return nil + } + notifyUser("exported chat", "chat: "+activeChatName+" was exported") return nil } if event.Key() == tcell.KeyCtrlA { @@ -406,8 +415,16 @@ func init() { msgText := textArea.GetText() // TODO: check whose message was latest (user icon / assistant) // in order to decide if assistant new icon is needed + nl := "\n" + prevText := textView.GetText(true) + // strings.LastIndex() + // newline is not needed is prev msg ends with one + if strings.HasSuffix(prevText, nl) { + nl = "" + } if msgText != "" { - fmt.Fprintf(textView, "\n(%d) <user>: \n%s\n", len(chatBody.Messages), msgText) + fmt.Fprintf(textView, "%s[-:-:u](%d) <%s>: [-:-:-]\n%s\n", + nl, len(chatBody.Messages), cfg.UserRole, msgText) textArea.SetText("", true) textView.ScrollToEnd() colorText() |