Add loading indicator

app/src/main/java/com/ti/mobpo/ui/screens/PokeSearch.kt

@@ -19,6 +19,7 @@ import androidx.compose.material.icons.filled.Favorite
 import androidx.compose.material.icons.outlined.FavoriteBorder
 import androidx.compose.material3.Card
 import androidx.compose.material3.CardDefaults
+import androidx.compose.material3.CircularProgressIndicator
 import androidx.compose.material3.Icon
 import androidx.compose.material3.IconButton
 import androidx.compose.material3.MaterialTheme
@@ -53,6 +54,7 @@ fun PokeSearchScreen(viewModel: PokeSearchViewModel) {
 @Composable
 fun PokeSearch(pokeSearchViewModel: PokeSearchViewModel) {
     val searchResults by pokeSearchViewModel.pokemonDetails.collectAsState()
+    val isLoading by pokeSearchViewModel.isLoading.collectAsState()
 
     Column(
         verticalArrangement = Arrangement.Top,
@@ -67,6 +69,9 @@ fun PokeSearch(pokeSearchViewModel: PokeSearchViewModel) {
 
         Spacer(modifier = Modifier.height(16.dp))
 
+        if(isLoading) {
+            CircularProgressIndicator()
+        } else {
             searchResults?.let { results ->
                 if (results.isNotEmpty()) {
                     LazyVerticalGrid(
@@ -86,6 +91,7 @@ fun PokeSearch(pokeSearchViewModel: PokeSearchViewModel) {
             }
         }
     }
+}
 
 
 @Composable
@@ -125,7 +131,8 @@ fun PokemonCard(pokemon: PokemonDetails, toggleFavorite: (Int) -> Unit) {
                 )
                 IconButton(
                     onClick = { toggleFavorite(pokemon.id) },
-                    modifier = Modifier.wrapContentSize()
+                    modifier = Modifier
+                        .wrapContentSize()
                         .layout { measurable, constraints ->
                             if (constraints.maxHeight == Constraints.Infinity) {
                                 layout(0, 0) {}

app/src/main/java/com/ti/mobpo/ui/viewmodels/PokeSearchViewModel.kt

@@ -21,6 +21,9 @@ class PokeSearchViewModel(private val favouritesRepository: FavouritesRepository
     private val _pokemonDetails = MutableStateFlow<List<PokemonDetails>?>(null)
     val pokemonDetails: StateFlow<List<PokemonDetails>?> = _pokemonDetails.asStateFlow()
     private val _initialPokemonList = MutableStateFlow<List<PokemonSpecies>?>(null)
+
+    private val _isLoading = MutableStateFlow(false)
+    val isLoading: StateFlow<Boolean> = _isLoading.asStateFlow()
     init {
         fetchPokemonSpecies()
     }
@@ -37,6 +40,7 @@ class PokeSearchViewModel(private val favouritesRepository: FavouritesRepository
     }
 
     fun search(query: String) {
+        _isLoading.value = true
         viewModelScope.launch {
             try {
                 val filteredList = _initialPokemonList.value?.binarySearch(query)
@@ -52,6 +56,8 @@ class PokeSearchViewModel(private val favouritesRepository: FavouritesRepository
                 _pokemonDetails.value = detailsList
             } catch (e: IOException) {
                 /* Handle error */
+            } finally {
+                _isLoading.value = false
             }
         }
     }