The growing adoption of decentralised renewable energy generation (such as solar photovoltaic panels and wind turbines) and low-carbon technologies will increase the strain experienced by the distribution networks in the near future. In such a scenario, energy storage is becoming a key alternative to traditional expensive reinforcements to network infrastructure, due to its flexibility, decreasing costs and fast deployment capabilities. In this work, an end-to-end data-driven solution to optimally design the control of a battery unit with the aim of reducing the peak electricity demand is presented. The proposed solution uses state-of-the-art machine learning methods for forecasting electricity demand and PV generation, combined with an optimisation strategy to maximise the use of photovoltaic energy to charge the energy storage unit. To this end, historical demand, weather, and solar energy generation data collected at the Stentaway Primary substation near Plymouth, UK, and at other six locations were employed.