MainActivity.kt

class MainActivity : AppCompatActivity() {
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContent {
            DialogTestTheme {
                App()
            }
        }
    }
}

@Composable
fun App() {
    var dialogOpen by remember { mutableStateOf(false) }

    Column(
        modifier = Modifier.fillMaxSize(),
        verticalArrangement = Arrangement.Center,
        horizontalAlignment = Alignment.CenterHorizontally
    ) {
        Button(onClick = { dialogOpen = true }) {
            Text("OPEN")
        }
    }

    if (dialogOpen) {
        Dialog { dialogOpen = false }
    }
}

@Composable
fun Dialog(onDismiss: () -> Unit) {
    val (field, setField) = remember { mutableStateOf("") }

    AlertDialog(
        onDismissRequest = onDismiss,
        text = {
            OutlinedTextField(
                value = field,
                onValueChange = setField,
                label = {
                    Text("LABEL")
                },
            )
        },
        confirmButton = {
            var count by remember { mutableStateOf(0) }
            Button(onClick = { count++ }) {
                Text("BUTTON $count")
            }
        }
    )
}