add questionDone to hey model (#107)

JunNishimura · Sep 2, 2023 · ad78dbf · ad78dbf
1 parent 4dfe9e8
commit ad78dbf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ui/cmd/hey/model.go b/ui/cmd/hey/model.go
@@ -79,6 +79,7 @@ type Model struct {
 	spotifyClient    *spotify.Client
 	openaiClient     *openai.Client
 	questionIndex    int
+	questionDone     bool
 	functionCall     any
 	chatCompMessages []openai.ChatCompletionMessage
 	functions        []openai.FunctionDefinition

diff --git a/ui/cmd/hey/update.go b/ui/cmd/hey/update.go
@@ -25,6 +25,9 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		case "ctrl+c", "q":
 			return m, tea.Quit
 		case "tab":
+			if m.questionDone {
+				m.state = recommendationView
+			}
 			m.state = m.state.Switch()
 		case "enter":
 			if m.err != nil {
@@ -83,6 +86,10 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.chatCompMessages = append(m.chatCompMessages, msg.msg)
 		m.functionCall = "auto"
 		m.questionIndex++
+		if m.questionIndex == len(functions.List) {
+			m.questionDone = true
+			m.state = recommendationView
+		}
 		return m, tea.Batch(m.generate, m.recommend)
 	case recommendMsg:
 		m.recommendItems = msg.items