The World No. 2 has rediscovered his best form following a serious ankle injury in 2022