The icon for AI-assisted labeling becomes only available while drawing a polygon. It is also available if a polygon is selected. In that case, the response of the AI would replace the selected polygon. In an error case, the selected polygon is restored.