The image that was generated wasn't based off the last message in the chat. I want it to literally use the last message in the users chat as the prompt and generate an image based off that prompt using an AI image generator model that is fully built in and integrated into the website.

Add an ai model that is hosted locally on the website to generate images based off the chat. Add a button in the users chat to generate an actual image of the current scene using the local image model that's hosted on the website.