AI Solution for Efficient and Sustainable Water Treatment