removed references to ControlNet and added an integration test

Author: David Motsonashvili

firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ImagenTests.kt

@@ -0,0 +1,46 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.google.firebase.ai
+
+import com.google.firebase.ai.AIModels.Companion.app
+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.ImagenRawImage
+import com.google.firebase.ai.type.PublicPreviewAPI
+import kotlinx.coroutines.runBlocking
+import org.junit.Test
+
+@OptIn(PublicPreviewAPI::class)
+class ImagenTests {
+    @Test
+    fun testGenerateAndEditImage() {
+        val imageGenerationModel = FirebaseAI.getInstance(app()).imagenModel("imagen-3.0-generate-002")
+        val imageEditingModel = FirebaseAI.getInstance(app()).imagenModel("imagen-3.0-capability-001")
+
+        runBlocking {
+            val catImage = imageGenerationModel.generateImages("A cat").images.first()
+            val editedCatImage = imageEditingModel.editImage(
+                listOf(ImagenRawImage(catImage), ImagenBackgroundMask()),
+                "A cat flying through space",
+                ImagenEditingConfig(
+                    ImagenEditMode.INPAINT_INSERTION
+                )
+            )
+            assert(editedCatImage.images.size == 1)
+        }
+    }
+}

firebase-ai/src/main/kotlin/com/google/firebase/ai/type/ImagenReferenceImage.kt

@@ -73,15 +73,16 @@ internal constructor(
 }
 
 /**
- * Represents a reference image (provided or generated) to bound the created image via ControlNet
+ * Represents a reference image (provided or generated) to bound the created image via controlled
+ * generation.
  * @param image the image provided, required if [enableComputation] is false
- * @param type the type of ControlNet reference image
+ * @param type the type of control reference image
  * @param referenceId the reference ID for this image, to be referenced in the prompt
- * @param enableComputation requests that the reference image be generated serverside instead of
+ * @param [enableComputation] requests that the reference image be generated serverside instead of
  * provided
- * @param superpixelRegionSize if type is COLOR_SUPERPIXEL and enableComputation is true, this will
+ * @param superpixelRegionSize if type is COLOR_SUPERPIXEL and [enableComputation] is true, this will
  * control the size of each superpixel region in pixels for the generated referenced image
- * @param superpixelRuler if type is COLOR_SUPERPIXEL and enableComputation is true, this will
+ * @param superpixelRuler if type is COLOR_SUPERPIXEL and [enableComputation] is true, this will
  * control the superpixel smoothness factor for the generated referenced image
  */
 @PublicPreviewAPI