AI water optimization