My team selection would be

1. Parthiv Patel (always delivered as opener, good against fast bowling)
2. L Rahul/ M. Agarwal
3. Che Pu
4. V.Kohli
5. A. Rahane
6. R. Pant
7. H. Pandya
8. R. Jadeja/ Ashwin (if fit)
9. I. Sharma
10. M. Shami
11. Boom Bumrah

Solves the tail issue, also have an additional bowler in Pandya, spin option available and no.8 contributes to score. Only issue would be between M.Agarwal and Rahul, I would still go with Rahul if he can replicate the first test second innings. Vijay should be dropped, not contributing and no point in investing in him. Parthiv knows aussie conditions and every time he has been asked to bat, he has come good (not at keeping though) so like in S. Africa.