Python Plotly Sankey Diagram: how to calculate 'value' based on 'source' and 'target so all lines are the same height as nodes

Question

So I realise that this is quite a specific question, but it\'s been bugging me for ages and I can\'t seem to find a solution..

I am using Plotly with Python to produc