AI water management