import pandas as pd
import holoviews as hv

hv.extension('bokeh')

edges = pd.read_csv('../../../assets/energy.csv')
edges.head(5)

sankey = hv.Sankey(edges, label='Energy Diagram')
sankey.opts(label_position='left', edge_color='target', node_color='index', cmap='tab20')