AI powered water management