Citation: 1C_294/2018 E. D

A.________ forme un recours en matière de droit public au Tribunal fédéral par lequel il demande la réforme de l'arrêt cantonal en ce sens que le permis de conduire de catégorie B délivré le 14 mars 2017 lui est restitué, les frais de l'expertise ainsi que les frais de la cause étant mis à la charge du SAN. Il réclame l'allocation d'une indemnité équitable. Subsidiairement, il conclut à la réforme de l'arrêt cantonal en ce sens que le permis de conduire de catégorie B lui est restitué moyennant son engagement à poursuivre des consultations thérapeutiques de soutien auprès d'un centre de psychothérapie reconnu pendant une année au moins à compter de la restitution. Plus subsidiairement encore, il réclame le renvoi de la cause au SAN pour instruction, notamment en ce qui concerne le rapport d'expertise de l'UMPT. Il requiert également l'octroi de l'assistance judiciaire. Le Tribunal cantonal renvoie aux considérants de son arrêt. Le SAN et l'Office fédéral des routes se réfèrent également à l'arrêt attaqué et concluent au rejet du recours, sans autres observations.