Bring back Improve with AI!

Please bring back the improve with AI functionality to the inbox. Prior to the latest update if I use the suggest with AI option and edited the message the option to improve with AI would appear. Now, we don’t have the improve with AI option and only have a clear all option. In order to improve the suggested with AI message I now have to copy the message, clear it, start a new message, paste the message and then click improve with AI.