wip: imagen migration to nanobanana

thatfiredev · thatfiredev · commit eb17b98bb210 · 2026-03-26T18:41:55.000Z
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenGenerationViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenGenerationViewModel.kt
@@ -2,17 +2,24 @@ package com.google.firebase.quickstart.ai.feature.media.imagen
 
 import android.graphics.Bitmap
 import com.google.firebase.Firebase
+import com.google.firebase.ai.GenerativeModel
 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.ImagePart
 import com.google.firebase.ai.type.ImagenGenerationResponse
 import com.google.firebase.ai.type.ImagenImageFormat
 import com.google.firebase.ai.type.ImagenInlineImage
 import com.google.firebase.ai.type.ImagenPersonFilterLevel
 import com.google.firebase.ai.type.ImagenSafetyFilterLevel
 import com.google.firebase.ai.type.ImagenSafetySettings
 import com.google.firebase.ai.type.PublicPreviewAPI
+import com.google.firebase.ai.type.ResponseModality
+import com.google.firebase.ai.type.ThinkingLevel
+import com.google.firebase.ai.type.generationConfig
 import com.google.firebase.ai.type.imagenGenerationConfig
+import com.google.firebase.ai.type.thinkingConfig
+import com.google.firebase.ai.type.toImagenInlineImage
 import com.google.firebase.quickstart.ai.ui.ImagenUiState
 import kotlinx.serialization.Serializable
 
@@ -28,28 +35,37 @@ class ImagenGenerationViewModel : ImagenViewModel() {
     override val additionalImage: Bitmap? = null
     override val imageLabels: List<String> = emptyList()
 
-    private val imagenModel: ImagenModel
+    private val generativeModel: GenerativeModel
 
     init {
-        imagenModel = Firebase.ai(
+        generativeModel = Firebase.ai(
             backend = GenerativeBackend.googleAI()
-        ).imagenModel(
-            modelName = "imagen-4.0-generate-001",
-            generationConfig = imagenGenerationConfig {
-                numberOfImages = 4
-                imageFormat = ImagenImageFormat.png()
-            },
-            safetySettings = ImagenSafetySettings(
-                safetyFilterLevel = ImagenSafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
-                personFilterLevel = ImagenPersonFilterLevel.BLOCK_ALL
-            )
+        ).generativeModel(
+            modelName = "gemini-3.1-flash-image-preview",
+            generationConfig = generationConfig {
+                responseModalities = listOf(ResponseModality.TEXT, ResponseModality.IMAGE)
+//                candidateCount = 4
+                thinkingConfig = thinkingConfig {
+                    thinkingLevel = ThinkingLevel.MINIMAL
+                }
+            }
         )
     }
 
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
-        return imagenModel.generateImages(inputText)
+    ): List<Bitmap> {
+        val bitmaps = arrayListOf<Bitmap>()
+        generativeModel.generateContent(inputText)
+            .candidates.forEach { candidate ->
+                candidate.content.parts.filterIsInstance<ImagePart>()
+                    .forEach {
+                        bitmaps.add(it.image)
+                    }
+            }
+
+        return bitmaps
+//        return imagenModel.generateImages(inputText)
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenInpaintingViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenInpaintingViewModel.kt
@@ -57,7 +57,7 @@ class ImagenInpaintingViewModel : ImagenViewModel() {
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
+    ): List<Bitmap> {
         val bitmap = currentState.attachedImage!!
         val mask = when (currentState.selectedOption) {
             "Foreground" -> ImagenForegroundMask()
@@ -67,6 +67,6 @@ class ImagenInpaintingViewModel : ImagenViewModel() {
             listOfNotNull(ImagenRawImage(bitmap.toImagenInlineImage()), mask),
             inputText,
             ImagenEditingConfig(ImagenEditMode.INPAINT_INSERTION)
-        )
+        ).images.map { it.asBitmap() }
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenOutpaintingViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenOutpaintingViewModel.kt
@@ -58,7 +58,7 @@ class ImagenOutpaintingViewModel : ImagenViewModel() {
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
+    ): List<Bitmap> {
         val bitmap = currentState.attachedImage!!
         val position = when (currentState.selectedOption) {
             "Top" -> ImagenImagePlacement.TOP_CENTER
@@ -77,6 +77,6 @@ class ImagenOutpaintingViewModel : ImagenViewModel() {
             listOf(sourceImage, ImagenRawMask(mask.image!!, 0.05)),
             inputText,
             ImagenEditingConfig(ImagenEditMode.OUTPAINT)
-        )
+        ).images.map { it.asBitmap() }
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenStyleTransferViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenStyleTransferViewModel.kt
@@ -55,14 +55,14 @@ class ImagenStyleTransferViewModel : ImagenViewModel() {
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
+    ): List<Bitmap> {
         val attachedImage = currentState.attachedImage!!
         return imagenModel.editImage(
             listOf(
                 ImagenRawImage(MainActivity.catImage.toImagenInlineImage()),
                 ImagenStyleReference(attachedImage.toImagenInlineImage(), 1, "an art style")
             ),
             "Generate an image in an art style [1] based on the following caption: $inputText",
-        )
+        ).images.map { it.asBitmap() }
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenSubjectReferenceViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenSubjectReferenceViewModel.kt
@@ -54,7 +54,7 @@ class ImagenSubjectReferenceViewModel : ImagenViewModel() {
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
+    ): List<Bitmap> {
         val attachedImage = currentState.attachedImage!!
         return imagenModel.editImage(
             listOf(
@@ -67,6 +67,6 @@ class ImagenSubjectReferenceViewModel : ImagenViewModel() {
             ),
             "Create an image about An animal [1] to match the description: " +
                     inputText.replace("<subject>", "An animal [1]"),
-        )
+        ).images.map { it.asBitmap() }
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenTemplateViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenTemplateViewModel.kt
@@ -34,7 +34,7 @@ class ImagenTemplateViewModel : ImagenViewModel() {
     override suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage> {
+    ): List<Bitmap> {
         return try {
             templateImagenModel.generateImages("imagen-basic", mapOf("prompt" to inputText))
         } catch (e: Exception) {
@@ -47,6 +47,6 @@ class ImagenTemplateViewModel : ImagenViewModel() {
             } else {
                 throw e
             }
-        }
+        }.images.map { it.asBitmap() }
     }
 }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/media/imagen/ImagenViewModel.kt
@@ -30,7 +30,7 @@ abstract class ImagenViewModel : ViewModel() {
     protected abstract suspend fun performGeneration(
         inputText: String,
         currentState: ImagenUiState.Success
-    ): ImagenGenerationResponse<ImagenInlineImage>
+    ): List<Bitmap>
 
     fun generateImages(inputText: String) {
         val currentState = (_uiState.value as? ImagenUiState.Success) ?: ImagenUiState.Success()
@@ -39,7 +39,7 @@ abstract class ImagenViewModel : ViewModel() {
             _uiState.value = ImagenUiState.Loading
             try {
                 val imageResponse = performGeneration(inputText, currentState)
-                _uiState.value = currentState.copy(images = imageResponse.images.map { it.asBitmap() })
+                _uiState.value = currentState.copy(images = imageResponse)
             } catch (e: Exception) {
                 _uiState.value = ImagenUiState.Error(e.localizedMessage ?: "Unknown error")
             }
diff --git a/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/text/ImageGenerationViewModel.kt b/firebase-ai/app/src/main/java/com/google/firebase/quickstart/ai/feature/text/ImageGenerationViewModel.kt
@@ -31,13 +31,15 @@ class ImageGenerationViewModel : ChatViewModel() {
             modelName = "gemini-2.5-flash-image",
             generationConfig = generationConfig {
                 responseModalities = listOf(ResponseModality.TEXT, ResponseModality.IMAGE)
+                candidateCount = 4
             }
         )
         chat = generativeModel.startChat()
     }
 
     override suspend fun performSendMessage(prompt: Content, currentMessages: List<UiChatMessage>) {
         val response = chat.sendMessage(prompt)
+
         validateAndDisplayResponse(response, currentMessages)
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ class ImagenTemplateViewModel : ImagenViewModel() {`
`34`	`34`	`override suspend fun performGeneration(`
`35`	`35`	`inputText: String,`
`36`	`36`	`currentState: ImagenUiState.Success`
`37`		`- ): ImagenGenerationResponse<ImagenInlineImage> {`
	`37`	`+ ): List<Bitmap> {`
`38`	`38`	`return try {`
`39`	`39`	`templateImagenModel.generateImages("imagen-basic", mapOf("prompt" to inputText))`
`40`	`40`	`} catch (e: Exception) {`
`@@ -47,6 +47,6 @@ class ImagenTemplateViewModel : ImagenViewModel() {`
`47`	`47`	`} else {`
`48`	`48`	`throw e`
`49`	`49`	`}`
`50`		`- }`
	`50`	`+ }.images.map { it.asBitmap() }`
`51`	`51`	`}`
`52`	`52`	`}`
Original file line number	Diff line number	Diff line change
`@@ -31,13 +31,15 @@ class ImageGenerationViewModel : ChatViewModel() {`
`31`	`31`	`modelName = "gemini-2.5-flash-image",`
`32`	`32`	`generationConfig = generationConfig {`
`33`	`33`	`responseModalities = listOf(ResponseModality.TEXT, ResponseModality.IMAGE)`
	`34`	`+ candidateCount = 4`
`34`	`35`	`}`
`35`	`36`	`)`
`36`	`37`	`chat = generativeModel.startChat()`
`37`	`38`	`}`
`38`	`39`
`39`	`40`	`override suspend fun performSendMessage(prompt: Content, currentMessages: List<UiChatMessage>) {`
`40`	`41`	`val response = chat.sendMessage(prompt)`
	`42`	`+`
`41`	`43`	`validateAndDisplayResponse(response, currentMessages)`
`42`	`44`	`}`
`43`	`45`	`}`