Continue the discussion with the chatbot after generating the text

Hi. Can you add these features to the AI templates? (Continue the discussion with the chatbot after generating the text, see the screenshot taken on Magicschool AI)