Add a feature toggle for background vibes

Author: riggaroo

core/network/src/main/java/com/android/developers/androidify/RemoteConfigDataSource.kt

@@ -24,6 +24,7 @@ interface RemoteConfigDataSource {
     fun isAppInactive(): Boolean
     fun textModelName(): String
     fun imageModelName(): String
+    fun isBackgroundVibesFeatureEnabled(): Boolean
     fun promptTextVerify(): String
     fun promptImageValidation(): String
     fun promptImageDescription(): String
@@ -60,6 +61,10 @@ class RemoteConfigDataSourceImpl @Inject constructor() : RemoteConfigDataSource
         return remoteConfig.getString("image_model_name")
     }
 
+    override fun isBackgroundVibesFeatureEnabled(): Boolean {
+        return remoteConfig.getBoolean("background_vibes_feature_enabled")
+    }
+
     override fun promptTextVerify(): String {
         return remoteConfig.getString("prompt_text_verify")
     }

core/network/src/main/res/xml/remote_config_defaults.xml

@@ -15,6 +15,10 @@
      limitations under the License.
 -->
 <defaults>
+    <entry>
+        <key>background_vibes_feature_enabled</key>
+        <value>false</value>
+    </entry>
     <entry>
         <key>bot_background_instruction_prompt</key>
         <value>Add the input image android bot as the main subject to the result,

core/testing/src/main/java/com/android/developers/testing/network/TestRemoteConfigDataSource.kt

@@ -30,6 +30,10 @@ class TestRemoteConfigDataSource(private val useGeminiNano: Boolean) : RemoteCon
         TODO("Not yet implemented")
     }
 
+    override fun isBackgroundVibesFeatureEnabled(): Boolean {
+        return true
+    }
+
     override fun promptTextVerify(): String {
         TODO("Not yet implemented")
     }

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt

@@ -24,6 +24,7 @@ import androidx.compose.material3.SnackbarHostState
 import androidx.compose.ui.Modifier
 import androidx.lifecycle.AndroidViewModel
 import androidx.lifecycle.viewModelScope
+import com.android.developers.androidify.RemoteConfigDataSource
 import com.android.developers.androidify.data.ImageGenerationRepository
 import com.android.developers.androidify.util.LocalFileProvider
 import dagger.assisted.Assisted
@@ -43,6 +44,7 @@ class CustomizeExportViewModel @AssistedInject constructor(
     val imageGenerationRepository: ImageGenerationRepository,
     val composableBitmapRenderer: ComposableBitmapRenderer,
     val localFileProvider: LocalFileProvider,
+    val remoteConfigDataSource: RemoteConfigDataSource,
     application: Application,
 ) : AndroidViewModel(application) {
 
@@ -63,10 +65,39 @@ class CustomizeExportViewModel @AssistedInject constructor(
         get() = _snackbarHostState
 
     init {
+        val enableBackgroundVibes = remoteConfigDataSource.isBackgroundVibesFeatureEnabled()
+        var backgrounds = mutableListOf(
+            BackgroundOption.None,
+            BackgroundOption.Plain,
+            BackgroundOption.Lightspeed,
+            BackgroundOption.IO,
+        )
+        if (enableBackgroundVibes) {
+            val backgroundVibes = listOf(
+                BackgroundOption.MusicLover,
+                BackgroundOption.PoolMaven,
+                BackgroundOption.SoccerFanatic,
+                BackgroundOption.StarGazer,
+                BackgroundOption.FitnessBuff,
+                BackgroundOption.Fandroid,
+                BackgroundOption.GreenThumb,
+                BackgroundOption.Gamer,
+                BackgroundOption.Jetsetter,
+                BackgroundOption.Chef,
+            )
+            backgrounds.addAll(backgroundVibes)
+        }
+
         _state.update {
             it.copy(
                 originalImageUrl = originalImageUrl,
                 exportImageCanvas = it.exportImageCanvas.copy(imageUri = resultImageUrl),
+                toolState = mapOf(
+                    CustomizeTool.Size to AspectRatioToolState(),
+                    CustomizeTool.Background to BackgroundToolState(
+                        options = backgrounds
+                    ),
+                )
             )
         }
         loadInitialBitmap(resultImageUrl)

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt

@@ -60,17 +60,7 @@ data class BackgroundToolState(
         BackgroundOption.None,
         BackgroundOption.Plain,
         BackgroundOption.Lightspeed,
-        BackgroundOption.IO,
-        BackgroundOption.MusicLover,
-        BackgroundOption.PoolMaven,
-        BackgroundOption.SoccerFanatic,
-        BackgroundOption.StarGazer,
-        BackgroundOption.FitnessBuff,
-        BackgroundOption.Fandroid,
-        BackgroundOption.GreenThumb,
-        BackgroundOption.Gamer,
-        BackgroundOption.Jetsetter,
-        BackgroundOption.Chef,
+        BackgroundOption.IO
     ),
 ) : ToolState