Suggestion for Contextual Task Bar Behavior with Remove Option in Photoshop

I truly appreciate the recent addition of the "Remove" option in the Contextual Task Bar. However, I would like to suggest a small but meaningful improvement to enhance user control and experience.

Currently, the Remove Option in Contextual Bar works via Generative AI (cloud-based), and the Contextual Task Bar reflects only this mode. It would be much more efficient and user-friendly if the Contextual Task Bar adapts based on the user's chosen mode of the Remove Tool.

Suggestion:

If the user has Generative AI turned ON, then the cloud-based Remove option should appear in the Contextual Task Bar.

If the user has Generative AI turned OFF, then the on-device Remove option should be available in the same Contextual Task Bar instead of removing the feature altogether.

This will provide a seamless workflow and give users better control, especially when they prefer offline or on-device operations.