Import main only when necessary

[bencw99]

Before, we imported main regardless of whether the index was already complete. Only importing after checking for completion can speed up the work for an already completed index significantly when imports are expensive (e.g., importing torch).

Before:

After:

[kristian-georgiev]

tried it and it works great -- much faster skipping