tmpm323uwl2.kotlin

sealed class Route {
    data object ImageList : Route()
    data object ImagePicker : Route()
}

@Composable
fun ImageListScreen(navController: NavController) {
    var selectedImage by remember { mutableStateOf<String?>(null) }
    
    val pickedImage = navController.currentBackStackEntry
        ?.savedStateHandle
        ?.getStateFlow<String?>("picked_image", null)
        ?.collectAsState()
    
    LaunchedEffect(pickedImage?.value) {
        pickedImage?.value?.let { image ->
            selectedImage = image
            Log.d("ImageList", "Received image: $image")
        }
    }
    
    Column {
        if (selectedImage != null) {
            AsyncImage(model = selectedImage, contentDescription = null)
        }
        
        Button(onClick = { navController.navigate(Route.ImagePicker) }) {
            Text("Pick Image")
        }
    }
}

@Composable
fun ImagePickerScreen(navController: NavController) {
    val images = listOf(
        "https://example.com/image1.jpg",
        "https://example.com/image2.jpg",
        "https://example.com/image3.jpg"
    )
    
    LazyVerticalGrid(
        columns = GridCells.Fixed(3),
        modifier = Modifier.fillMaxSize()
    ) {
        items(images) { imageUrl ->
            AsyncImage(
                model = imageUrl,
                contentDescription = null,
                modifier = Modifier
                    .fillMaxWidth()
                    .aspectRatio(1f)
                    .clickable {
                        navController.previousBackStackEntry
                            ?.savedStateHandle
                            ?.set("picked_image", imageUrl)
                        
                        navController.popBackStack()
                    }
            )
        }
    }
}