Merge pull request #2 from greycodee/tui

gen title

Author: greycodee

gemini.go

@@ -77,7 +77,7 @@ func (g *GeminiClient) genTitle() string {
 	return strings.ReplaceAll(strings.Join(modelPart, ""), "*", "")
 }
 
-func (g *GeminiClient) sendMessageToTui(textChan chan string, historyChan chan string, genFlagChan chan bool, db *DB) {
+func (g *GeminiClient) sendMessageToTui(textChan chan string, historyChan chan string, genFlagChan chan bool, titleChan chan string, db *DB) {
 	firstQuestion := true
 	for {
 		text := <-textChan
@@ -141,6 +141,7 @@ func (g *GeminiClient) sendMessageToTui(textChan chan string, historyChan chan s
 			firstQuestion = false
 			go func() {
 				title := g.genTitle()
+				titleChan <- title
 				db.InsertChat(GeminiChatList{
 					ChatID:    int64(g.chatID),
 					ChatTitle: title,

main.go

@@ -14,6 +14,7 @@ import (
 
 var HOME_PATH = os.Getenv("HOME")
 var chatID = 1
+var chatHistoryTitle = "Chat History <Ctrl-H>"
 
 func main() {
 	db := initDB()
@@ -38,6 +39,7 @@ func main() {
 	sendMsgChan := make(chan string)
 	historyChan := make(chan string)
 	genFlagChan := make(chan bool)
+	titleChan := make(chan string)
 
 	app := tview.NewApplication()
 
@@ -55,15 +57,15 @@ func main() {
 	geminiClient.startChat(history)
 	defer geminiClient.client.Close()
 
-	go geminiClient.sendMessageToTui(sendMsgChan, historyChan, genFlagChan, db)
+	go geminiClient.sendMessageToTui(sendMsgChan, historyChan, genFlagChan, titleChan, db)
 
 	chatLog := tview.NewTextView().
 		SetDynamicColors(true).
 		SetRegions(true)
 	chatLog.SetChangedFunc(func() {
 		app.Draw()
 		chatLog.ScrollToEnd()
-	}).SetBorder(true).SetTitle("Chat History <Ctrl-H>")
+	}).SetBorder(true).SetTitle(chatHistoryTitle)
 
 	go func() {
 		for {
@@ -74,6 +76,13 @@ func main() {
 		}
 	}()
 
+	go func() {
+		title := <-titleChan
+		app.QueueUpdateDraw(func() {
+			chatLog.SetTitle(chatHistoryTitle + " [" + title + "]")
+		})
+	}()
+
 	textArea := tview.NewTextArea()
 	textArea.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
 		if event.Key() == tcell.KeyCtrlS {