Like the new ask lex box have this copy button bellow the box but maybe also make it so its can just automatically apply the ai edit or sugestion directly to the document so you don't have to always copy paste it like by just use apply button or something