Energy storage technology is introduced into smart grids as a new means of power regulation, which provides a new way to ensure the smooth operation of the grid. A new nonlinear distributed scheme based on the flocking algorithm is presented to solve the transient stability control problem when a power system fails. In this scheme, by using the relation between the nonlinear term of the motor dynamics and the system frequency, the system state feedback term is introduced to speed up the restoration of transient stability of the system. System failure is controlled by the device of external energy storage. Finally, the proposed controller is applied to the IEEE New England 39-node power system model. Simulation results show that the proposed controller has advantages of higher speed and feasibility. Furthermore, the results verify the effectiveness of the proposed controller with a low capacity energy storage device.