add outpainting example

Author: David Motsonashvili

firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/FirebaseAISamples.kt

@@ -1,9 +1,14 @@
 package com.google.firebase.quickstart.ai
 
 import android.graphics.Bitmap
+import com.google.firebase.ai.ImagenModel
+import com.google.firebase.ai.type.Dimensions
 import com.google.firebase.ai.type.FunctionDeclaration
 import com.google.firebase.ai.type.GenerativeBackend
 import com.google.firebase.ai.type.ImagenBackgroundMask
+import com.google.firebase.ai.type.ImagenEditMode
+import com.google.firebase.ai.type.ImagenEditingConfig
+import com.google.firebase.ai.type.ImagenMaskReference
 import com.google.firebase.ai.type.ImagenRawImage
 import com.google.firebase.ai.type.PublicPreviewAPI
 import com.google.firebase.ai.type.ResponseModality
@@ -152,9 +157,35 @@ val FIREBASE_AI_SAMPLES = listOf(
             )
         },
         includeAttach = true,
-        bundleReferenceImages = {string: String, bitmap: Bitmap? ->
-                listOf(ImagenRawImage(bitmap!!.toImagenInlineImage()), ImagenBackgroundMask())
-            }
+        generateImages = { model: ImagenModel, inputText: String, bitmap: Bitmap? ->
+            model.editImage(
+                listOf(ImagenRawImage(bitmap!!.toImagenInlineImage()), ImagenBackgroundMask()),
+                inputText,
+                ImagenEditingConfig(ImagenEditMode.INPAINT_INSERTION)
+            )
+        }
+    ),
+    Sample(
+        title = "Imagen 3 - Outpainting",
+        description = "Expand an image by drawing in more background",
+        modelName= "imagen-3.0-capability-001",
+        backend = GenerativeBackend.vertexAI(),
+        navRoute = "imagen",
+        categories = listOf(Category.IMAGE),
+        initialPrompt = content {
+            text(
+                ""
+            )
+        },
+        includeAttach = true,
+        generateImages = { model: ImagenModel, inputText: String, bitmap: Bitmap? ->
+            val dimensions = Dimensions(bitmap!!.width * 2, bitmap.height * 2)
+            model.editImage(
+                ImagenMaskReference.generateMaskAndPadForOutpainting(bitmap.toImagenInlineImage(), dimensions),
+                inputText,
+                ImagenEditingConfig(ImagenEditMode.OUTPAINT)
+            )
+        }
     ),
     Sample(
         title = "Gemini 2.0 Flash - image generation",

firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenScreen.kt

@@ -102,9 +102,7 @@ fun ImagenScreen(
             }
             TextButton(
                 onClick = {
-                    if (imagenPrompt.isNotBlank()) {
-                        imagenViewModel.generateImages(imagenPrompt)
-                    }
+                    imagenViewModel.generateImages(imagenPrompt)
                 },
                 modifier = Modifier
                     .padding(end = 16.dp, bottom = 16.dp)

firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenViewModel.kt

@@ -9,7 +9,6 @@ import androidx.navigation.toRoute
 import com.google.firebase.Firebase
 import com.google.firebase.ai.ImagenModel
 import com.google.firebase.ai.ai
-import com.google.firebase.ai.type.GenerativeBackend
 import com.google.firebase.ai.type.ImagenAspectRatio
 import com.google.firebase.ai.type.ImagenEditMode
 import com.google.firebase.ai.type.ImagenEditingConfig
@@ -52,7 +51,6 @@ class ImagenViewModel(
     init {
         val config = imagenGenerationConfig {
             numberOfImages = 4
-            aspectRatio = ImagenAspectRatio.SQUARE_1x1
             imageFormat = ImagenImageFormat.png()
         }
         val settings = ImagenSafetySettings(
@@ -73,17 +71,13 @@ class ImagenViewModel(
         viewModelScope.launch {
             _isLoading.value = true
             try {
-                val bundleReferenceImages = sample.bundleReferenceImages
-                val imageResponse = if (bundleReferenceImages == null) {
+                val generateImages = sample.generateImages
+                val imageResponse = if (generateImages == null) {
                     imagenModel.generateImages(
                         inputText
                     )
                 } else {
-                    imagenModel.editImage(
-                        bundleReferenceImages(inputText, attachedImage),
-                        inputText,
-                        ImagenEditingConfig(ImagenEditMode.INPAINT_INSERTION)
-                    )
+                    generateImages(imagenModel, inputText, attachedImage)
                 }
                 _generatedBitmaps.value = imageResponse.images.map { it.asBitmap() }
                 _errorMessage.value = null // clear error message

firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/ui/navigation/Sample.kt

@@ -1,10 +1,12 @@
 package com.google.firebase.quickstart.ai.ui.navigation
 
 import android.graphics.Bitmap
+import com.google.firebase.ai.ImagenModel
 import com.google.firebase.ai.type.Content
 import com.google.firebase.ai.type.GenerationConfig
 import com.google.firebase.ai.type.GenerativeBackend
-import com.google.firebase.ai.type.ImagenReferenceImage
+import com.google.firebase.ai.type.ImagenGenerationResponse
+import com.google.firebase.ai.type.ImagenInlineImage
 import com.google.firebase.ai.type.PublicPreviewAPI
 import com.google.firebase.ai.type.Tool
 import java.util.UUID
@@ -36,5 +38,5 @@ data class Sample(
     val chatHistory: List<Content> = emptyList(),
     val tools: List<Tool>? = null,
     val includeAttach: Boolean = false,
-    val bundleReferenceImages: ((String, Bitmap?) -> List<ImagenReferenceImage>)? = null
+    val generateImages: (suspend (ImagenModel, String, Bitmap?) -> ImagenGenerationResponse<ImagenInlineImage>)? = null
 )