Sounds like you could see a big improvement from doing a bit of everything:
Time in the water, to just get used to swimming the distance at high effort.
Coached swimming, improve your technique and efficiency in the water.
Weights, to improve your strength and power.