Fix: streaming to tui

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).