Pick/Reject Shortcuts/Button

Would it be possible for the Pick, Unpick, and Reject keyboard shortcuts to still work when in the NLP dialogue?

Alternatively, there could be a button for flagging. Something like this?

I’m pretty sure there is a way in the Sdk for me to flag photos, but not for registering shortcuts, unfortunately. I also try to not overlap too much with things you can do directly in Lightroom. But I understand that it would be great to have those shortcuts still working!