Shake animation for text field on wrong value submission

TextFieldScreen.kt

@Composable
fun TextFieldScreen() {
    Column(
        horizontalAlignment = Alignment.CenterHorizontally,
        verticalArrangement = Arrangement.spacedBy(8.dp, Alignment.CenterVertically)
    ) {

        val context = LocalContext.current

        val (email, onEmailChange) = remember {
            mutableStateOf("")
        }
        val shakeAnimation = remember { Animatable(0f) }

        var isError by remember {
            mutableStateOf(false)
        }


        val coroutineScope = rememberCoroutineScope()

        Column {
            OutlinedTextField(
                value = email,
                isError = isError,
                onValueChange = onEmailChange,
                label = {
                    Text(text = "Email Address")
                },
                modifier = Modifier
                    .offset { IntOffset(shakeAnimation.value.roundToInt(), y = 0) }
                    .graphicsLayer {
                        translationX = shakeAnimation.value * 2
                    }
            )

            AnimatedVisibility(visible = isError, modifier = Modifier.align(Alignment.Start)) {
                Text(text = "Invalid email address", color = MaterialTheme.colorScheme.error)
            }
        }
        
        Button(onClick = {
            if (email.isEmail()) {
                onEmailChange("")
                Toast.makeText(context, email, Toast.LENGTH_SHORT).show()
            } else {
                isError = true
                coroutineScope.launch {
                    for (i in 0..10) {
                        when (i % 2) {
                            0 -> shakeAnimation.animateTo(5f, spring(stiffness = 100_000f))
                            else -> shakeAnimation.animateTo(-5f, spring(stiffness = 100_000f))
                        }
                    }
                    shakeAnimation.animateTo(0f)
                }
            }
        }) {
            Text(text = "Submit")
        }
    }
}