На какой стадии такс фри возвращают, до прохода таможки или после?

Конечно до.

После паспортного контроля. Но надо отметить возврат до него.