Indian states are getting ready to battle a surge in omicron infections