Controlling the power is a bigger job, or rather a bigger board. Suddenly we get into higher power devices, and since there's various schemes for powering the smoke unit, I'd have to accommodate them all. I have considered it, but the board would have to be larger than it is now, and I've run across the same issues you have, trying to fit the existing design into some tight locomotives is a challenge. That's the reason I started looking at what I could do to reduce the size and came up with what I'm going to change.
Truthfully, you don't really want that much more power going into the smoke unit, you'd run into a host of issues with smoke unit maintenance if you put more than 6-7 watts into a smoke unit.
Other than the taking the smoke power in to gate the smoke motor, I'm also changing the TMCC headlight sensing circuit to be bi-polar so the Super-Chuffer will work with the older LCRU boards and R2LC boards earlier than C08. Right now, the headlight circuit only senses negative voltage to know when the headlight is on.