def handle_demand():
print('1 Load Demand')
print('2 Generate Demand')
selection = input('Please Select: ')
if selection == '1':
return read_csv_dataset(input('Filename of demand dataset: '))
elif selection == '2':
dg.generate_demand()
print('Demand generated')
return read_csv_dataset('demand_data.csv')
else:
raise # Throw error to main()
