@@ -396,6 +396,7 @@ func (m *UI) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
396 // Reload prompt history for the new session.
396 // Reload prompt history for the new session.
397 m.historyReset()
397 m.historyReset()
398 cmds = append(cmds, m.loadPromptHistory())
398 cmds = append(cmds, m.loadPromptHistory())
399+ m.updateLayoutAndSize()
400 case sendMessageMsg:
401 case sendMessageMsg:
401 cmds = append(cmds, m.sendMessage(msg.Content, msg.Attachments...))
402 cmds = append(cmds, m.sendMessage(msg.Content, msg.Attachments...))