The main thing I would love to see is a dedicated "Image Generation" section right inside the Kagi Assistant model selector. Currently, it is hard to know which models actually support image generation because the UI doesn't tell you at all.
By adding a specific image generation category directly to the model selector, it would clearly show users exactly which models can generate images. Additionally, it would be great to add a dedicated "Images" section to the main navigation bar. This could serve as a specific workspace for viewing and managing those generated images, making the workflow much smoother and keeping image prompts completely separated from our standard text chats.