The introduction of electric mobility solutions mitigates transportion-related greenhouse gas emissions. Transit buses are considered a promising candidate for electrification. This study contributes to the growing literature on battery-electric buses (BEBs) and aims to quantify the optimal allocation of BEB infrastructure and charging schedules. A generic model for the charging capacity and scheduling of the BEB network is developed. The proposed model proposes an algorithm for the calculation of trip-level BEB energy consumption based on a surrogate model-based space mapping algorithm. Instead of using vehicle simulators or constant values for the energy consumption rate for each trip, the input space mapping has been applied to a simple coarse model to build an accurate surrogate model. The proposed algorithm is tested on the bus transit network in Belleville City in Canada considering BEBs using both Flash and Opportunity charging. The results show the efficiency of the proposed model and highlight the impact on the optimization results of calculating the trip-level energy consumption compared with the traditional methods.