@Composable
@OptIn(ExperimentalFoundationApi::class)
fun SaveVariable() {
    val viewModel = VariableViewModel()
    val pagerState = rememberPagerState()
    LaunchedEffect(key1 = pagerState) {
        snapshotFlow { pagerState.currentPage }.collect { page ->
            viewModel.currentIndex = page
        }
    }
//    this code is causing issue
//    LaunchedEffect(key1 = viewModel.setLastCurrentPage) {
//        Log.e(">> LaunchedEffect", "${viewModel.currentIndex}")
//    }
    LgcTheme {
        SaveVariableItem(viewModel.currentIndex, viewModel.uiState, pagerState) {
            viewModel.changeUIi()
            viewModel.setLastCurrentPage = true
        }
    }
}

class VariableViewModel : ViewModel() {
    var uiState by mutableStateOf(ScreenName.FIRST)
    var setLastCurrentPage by mutableStateOf(false)
    var currentIndex by mutableStateOf(0)
    fun changeUIi() {
        uiState = if (uiState == ScreenName.FIRST) {
            ScreenName.SECOND
        } else {
            ScreenName.FIRST
        }
    }
}

enum class ScreenName {
    FIRST,
    SECOND
}

@OptIn(ExperimentalFoundationApi::class)
@Composable
fun SaveVariableItem(
    currentIndex: Int,
    uiState: ScreenName,
    pagerState: PagerState,
    changeUiState: () -> Unit,
) {
    Column {
        when (uiState) {
            ScreenName.FIRST -> {
                ScreenFirstView(pagerState)
            }

            ScreenName.SECOND -> {
                ScreenSecondView(currentIndex, "SECOND")
            }
        }
        Button(onClick = { changeUiState() }) {
            Text(text = "Add")
        }
    }
}

@OptIn(ExperimentalFoundationApi::class)
@Composable
fun ScreenFirstView(pagerState: PagerState) {
    Column(
        modifier = Modifier
            .wrapContentSize()
            .background(Color.Black)
    ) {
        HorizontalPager(
            modifier = Modifier.padding(20.dp),
            pageCount = 10,
            state = pagerState
        ) { page ->
            Text(
                text = "Page: $page",
                modifier = Modifier.fillMaxWidth(),
                color = Color.White
            )
        }
    }
}

@Composable
fun ScreenSecondView(currentIndex: Int, screenImage: String) {
    Column(
        modifier = Modifier
            .wrapContentSize()
            .background(Color.Black)
    ) {
        Text(text = "$currentIndex ==== $screenImage", color = Color.White)
    }
}

LaunchedEffect(key1 = viewModel.setLastCurrentPage) {
    Log.e(">> LaunchedEffect", "${viewModel.currentIndex}")
}