There are already treatments like this.
When I was in high school I started a minor computer repair business. One of my customers was a treatment service where they would place electrodes on the head of an individual to measure certain brainwave activity. The electrodes were connected through a device back to a laptop which was playing a video.
I'm explaining it very imprecisely (I'm an expert by no means) but the way it worked is that the video would only play when the measured brainwaves met the pattern(s) established for concentration. The idea being that the person would want the video to play continuously (without pausing or chopping) so badly that their brain would train itself to enter into the "concentrating" brainwave pattern.
I'm not sure what the efficacy of the solution is but the idea was interesting at the time.
UPDATE: Just looked it up and the term they use for this is "neurofeedback" training.
Looks like there are significant criticisms to the approach and study of this treatment so maybe it isn't a viable approach after all.
https://en.m.wikipedia.org/wiki/Neurofeedback