Changing navigation bar color according to screen

AppNavHost.kt

@Composable
fun AppNavHost(navController: NavHostController) {
    val context = LocalContext.current
    val navigationBarColor = animateColorAsState(
        targetValue = when (navController.currentDestination()?.route) {
            Screen.ScreenOne.route -> color_one
            Screen.ScreenTwo.route -> color_two
            else -> default_color
        },
        label = "navigationBarColor",
    )
    context.changeNavigationBarColor(navigationBarColor.value.toArgb())
    
    // Your screens...
}

@Composable
fun NavHostController.currentDestination(): NavDestination? {
    val navBackStackEntry by currentBackStackEntryAsState()
    return navBackStackEntry?.destination
}

fun Context.getActivity(): ComponentActivity? = when (this) {
    is ComponentActivity -> this
    is ContextWrapper -> baseContext.getActivity()
    else -> null
}

fun Context.changeNavigationBarColor(color: Int) {
    getActivity()?.window?.navigationBarColor = color
}