I think don't bring in the visions till the events of New Vegas begin, so it's after what happened at The Divide and after being shot in the head. That way you can still keep the aspect, but drop the supernatural mumbo-jumbo, which i fear could make some readers stop reading.