from DistrictHeatingNetworkCreator import DistrictHeatingNetworkCreator

# Initialize the class
network_creator = DistrictHeatingNetworkCreator(
    './input_files/buildings.geojson',
    './input_files/roads/geobase_mtl.shp'
)

# Create the network graph
network_graph = network_creator.run()

# Plot the network graph (optional)
network_creator.plot_network_graph(network_graph)