Pyspark window lead function with offset value from value of different column

Question

I am trying to derive a new column using window lead function, but my offset value to lead function varies depending on the value of the column, here is my example data