A specialist mental health unit has undergone a £2.1 million refurbishment to provide a more modern and therapeutic environment for patients.