回複：You must apply for Canadian visa if...

you're not Canadian permanent resident. But you don't need to apply U.S. visa in order to come back. You only need to show your I797 (probably with recent payroll stubs) and valid passport when re-entering U.S.