[AnimatedContent] JetSurvey initial wrap

JetSurvey.kt

/* Copyright 2022 Google LLC.	
   SPDX-License-Identifier: Apache-2.0 */
val state = viewModel.uiState.observeAsState().value ?: return@JetsurveyTheme
AnimatedContent(targetState = state) { targetState ->
   // It's important to use targetState and not state, as its critical to ensure
   // a successful lookup of all the incoming and outgoing content during
   // content transform.
   when (targetState) {
       is SurveyState.Questions -> SurveyQuestionsScreen(
           questions = targetState, /* other parameters */)
       is SurveyState.Result -> SurveyResultScreen(
           result = targetState, /* other parameters */)
   }
}