Implement Outpainted / Generative Expand to Imagen Editing Sample

Author: madebymozart

ai-catalog/app/src/main/res/values/strings.xml

@@ -15,7 +15,7 @@
     <string name="open_sample_button">Open sample</string>
     <string name="imagen_sample_title">Image generation with Imagen</string>
     <string name="imagen_sample_description">Generate images with Imagen, Google image generation model</string>
-    <string name="imagen_editing_sample_title">Imagen Editing using Inpainting</string>
+    <string name="imagen_editing_sample_title">Inpainting &amp; Outpainting with Imagen</string>
     <string name="imagen_editing_sample_description">Generate images and edit only specific areas of a generated image with Inpainting</string>
     <string name="magic_selfie_sample_title">Magic Selfie with Imagen and ML Kit</string>
     <string name="magic_selfie_sample_description">Change the background of your selfies with Imagen and the ML Kit Segmentation API</string>

ai-catalog/samples/imagen-editing/src/main/java/com/android/ai/samples/imagenediting/data/ImagenEditingDataSource.kt

@@ -18,14 +18,17 @@ package com.android.ai.samples.imagenediting.data
 import android.graphics.Bitmap
 import com.google.firebase.Firebase
 import com.google.firebase.ai.ai
+import com.google.firebase.ai.type.Dimensions
 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
 import com.google.firebase.ai.type.ImagenGenerationConfig
 import com.google.firebase.ai.type.ImagenImageFormat
+import com.google.firebase.ai.type.ImagenMaskReference
 import com.google.firebase.ai.type.ImagenRawImage
 import com.google.firebase.ai.type.ImagenRawMask
+import com.google.firebase.ai.type.ImagenStyleReference
 import com.google.firebase.ai.type.PublicPreviewAPI
 import com.google.firebase.ai.type.toImagenInlineImage
 import javax.inject.Inject
@@ -120,4 +123,29 @@ class ImagenEditingDataSource @Inject constructor() {
         )
         return imageResponse.images.first().asBitmap()
     }
+
+    /**
+     * Outpaints an image to the target dimensions using the Firebase Imagen API.
+     * This function extends the original image by generating content around it
+     * based on the provided prompt and target dimensions.
+     *
+     * @param sourceImage The original bitmap image to be outpainted.
+     * @param targetDimensions The desired dimensions of the outpainted image.
+     * @param prompt An optional text prompt to guide the outpainting process.
+     * @return The outpainted bitmap image.
+     */
+    @OptIn(PublicPreviewAPI::class)
+    suspend fun outpaintImage(
+        sourceImage: Bitmap,
+        targetDimensions: Dimensions,
+        prompt: String = "",
+    ): Bitmap {
+        val imageResponse = editingModel.outpaintImage(
+            image = sourceImage.toImagenInlineImage(),
+            newDimensions = targetDimensions,
+            prompt = prompt,
+        )
+
+        return imageResponse.images.first().asBitmap()
+    }
 }

ai-catalog/samples/imagen-editing/src/main/java/com/android/ai/samples/imagenediting/ui/ImagenEditingUIState.kt

@@ -16,12 +16,14 @@
 package com.android.ai.samples.imagenediting.ui
 
 import android.graphics.Bitmap
+import com.google.firebase.ai.type.Dimensions
 
 sealed interface ImagenEditingUIState {
     data object Initial : ImagenEditingUIState
     data object Loading : ImagenEditingUIState
     data class ImageGenerated(
         val bitmap: Bitmap,
+        val dimensions: Dimensions,
         val contentDescription: String,
     ) : ImagenEditingUIState