Looks cool! One piece of feedback I have is that I personally wouldn't want to autocomplete prompts as shown in the image you have there. I'd want something that "fixes" my prompts so that they're understood by the model. Like say if I put "Afro-Asian samurai carrying two swords in a cyberpunk environment", for it to say how I could re-word it to generate something closer to what I actually want.