I wonder if you could optimize it a little further to get energy from both directions…

Energy produced in the way up (heated water) and on the way down (gravity-Fed turbines)