Sure, but your power input should be a USB-C. No reason to use the clunky USB-B for that. Labeling as Power In and leaving a regular USB A for the output seems fine to me. Given that most older keyboards use USB-B for their input, A is a perfectly fine solution and solves the “input vs output”.
replyOP mentioned in another comment that a decent chunk of users is elderly, and C-to-B cables (albeit rare) do exist, so confusion isn't impossible.
replyI personally really don't like it in a world where everyone has USB-C power bricks readily available and USB-B cables are dying out, but considering the user base it is not an entirely unreasonable design decision - for now.