That's actually something that occurs anyway at this point. Right now, if there are vulnerable witnesses, be they adults or children, they can testify in another room. There are mechanisms in the current Criminal Code that are essential almost by default. The defence would have to argue why such mechanisms should not be used, such as having a child testify in another room or testify behind a screen or by video link. In addition, some courthouses, at least in Toronto, do have support animals for victims as aides. There are things currently in place.
I think that's already in the Criminal Code but I defer to anyone else if there is any other improvements they can see to address that.