Automating Water Management Systems Using AI