You can collect the shipping address on the checkout page then show a loader animation (preloader) while you make use of the wix pay api. Once the user is directed back to the page from the wix pay window, use code to get the payment status value from the Wix Pay’s response object.
If the payment is successful then you can fulfill the order using the shipping address the user has filled on your checkout page .