MainActivity.kt

@Composable
fun TakePhotoScreen(modifier: Modifier = Modifier) {
    // The location on disk where the photo was saved to, or null if no photo
    var imageUri by remember { mutableStateOf<Uri?>(null) }

    Column(modifier = modifier) {
        // We'll define this later...
        TakePhotoButton(
            onPhotoTaken = { imageUri = it }
        )

        if (imageUri != null) {
            // This composable is from the Coil library
            AsyncImage(
                modifier = Modifier
                    .weight(1f)
                    .fillMaxWidth(),
                contentScale = ContentScale.Fit,
                model = imageUri,
                contentDescription = "Photo taken by user"
            )
        }
    }
}