Fix: streaming to tui

bot.go

@@ -48,7 +48,6 @@ var (
 	chunkParser    ChunkParser
 	lastToolCall   *models.FuncCall
 	lastRespStats  *models.ResponseStats
-
 	outputHandler  OutputHandler
 	cliPrevOutput  string
 	cliRespDone    chan bool
@@ -65,30 +64,15 @@ type TUIOutputHandler struct {
 }
 
 func (h *TUIOutputHandler) Write(p string) {
-	if h.tv != nil {
 	fmt.Fprint(h.tv, p)
-	}
-	if cfg != nil && cfg.CLIMode {
-		fmt.Print(p)
-		cliPrevOutput = p
-	}
 }
 
 func (h *TUIOutputHandler) Writef(format string, args ...interface{}) {
-	s := fmt.Sprintf(format, args...)
+	fmt.Fprintf(h.tv, format, args...)
-	if h.tv != nil {
-		fmt.Fprint(h.tv, s)
-	}
-	if cfg != nil && cfg.CLIMode {
-		fmt.Print(s)
-		cliPrevOutput = s
-	}
 }
 
 func (h *TUIOutputHandler) ScrollToEnd() {
-	if h.tv != nil {
 	h.tv.ScrollToEnd()
-	}
 }
 
 type CLIOutputHandler struct{}

tui.go

@@ -230,7 +230,6 @@ func initTUI() {
 	tview.Styles = colorschemes["default"]
 	app = tview.NewApplication()
 	pages = tview.NewPages()
-	outputHandler = &TUIOutputHandler{tv: textView}
 	shellInput = tview.NewInputField().
 		SetLabel(fmt.Sprintf("[%s]$ ", cfg.FilePickerDir)). // dynamic prompt
 		SetFieldWidth(0).
@@ -349,6 +348,7 @@ func initTUI() {
 			// calling it explicitly makes text streaming to look more smooth
 			app.Draw()
 		})
+	outputHandler = &TUIOutputHandler{tv: textView}
 	notificationWidget = tview.NewTextView().
 		SetTextAlign(tview.AlignCenter).
 		SetDynamicColors(true).