Add an "Undo" button to revert to the state before the AI ​​fix.
🤔
Sometimes the result might not be perfect, and if so, you'd have to edit everything manually.