HomeScreenViewModel.kt

@HiltViewModel
class HomeScreenViewModel @Inject constructor(
    private val imageRepository: ImageRepository
) : ViewModel() {

    var state by mutableStateOf(HomeScreenState())
        private set

    fun onEvent(event: HomeScreenEvents) {
        when (event) {
            is HomeScreenEvents.LoadImages -> {
                loadImages(event.q)
            }
            is HomeScreenEvents.UpdateText -> {
                updateTextField(event.q)
            }
        }
    }

    private fun updateTextField(str: String) {
        state = state.copy(text = str)
    }

    private fun loadImages(q: String) {
        viewModelScope.launch {
            imageRepository.getImages(q).collect { result ->
                when (result) {
                    is Resources.Loading -> {
                        state = state.copy(isLoading = result.isLoading)
                    }
                    is Resources.Success -> {
                        println("Data in View model  ${result.data}")
                        result.data?.let { listings ->
                            state = state.copy(
                                images = listings
                            )
                        }
                    }
                    is Resources.Error -> {
                        state = state.copy(isLoading = false)
                    }
                }
            }
        }
    }
}