feat(genai): Add new image Generation samples (#13549)

* feat(genai): Add new image Generation samples

Local aware & Multiple images generation

* feat(genai): code formatting

Author: msampathkumar

genai/image_generation/imggen_canny_ctrl_type_with_txt_img.py

@@ -16,7 +16,12 @@
 def canny_edge_customization(output_gcs_uri: str) -> str:
     # [START googlegenaisdk_imggen_canny_ctrl_type_with_txt_img]
     from google import genai
-    from google.genai.types import ControlReferenceConfig, ControlReferenceImage, EditImageConfig, Image
+    from google.genai.types import (
+        ControlReferenceConfig,
+        ControlReferenceImage,
+        EditImageConfig,
+        Image,
+    )
 
     client = genai.Client()
 

genai/image_generation/imggen_inpainting_insert_mask_with_txt_img.py

@@ -18,18 +18,25 @@
 def edit_inpainting_insert_mask(output_file: str) -> Image:
     # [START googlegenaisdk_imggen_inpainting_insert_mask_with_txt_img]
     from google import genai
-    from google.genai.types import RawReferenceImage, MaskReferenceImage, MaskReferenceConfig, EditImageConfig
+    from google.genai.types import (
+        RawReferenceImage,
+        MaskReferenceImage,
+        MaskReferenceConfig,
+        EditImageConfig,
+    )
 
     client = genai.Client()
 
     # TODO(developer): Update and un-comment below line
     # output_file = "output-image.png"
 
     raw_ref = RawReferenceImage(
-        reference_image=Image.from_file(location='test_resources/fruit.png'), reference_id=0)
+        reference_image=Image.from_file(location="test_resources/fruit.png"),
+        reference_id=0,
+    )
     mask_ref = MaskReferenceImage(
         reference_id=1,
-        reference_image=Image.from_file(location='test_resources/fruit_mask.png'),
+        reference_image=Image.from_file(location="test_resources/fruit_mask.png"),
         config=MaskReferenceConfig(
             mask_mode="MASK_MODE_USER_PROVIDED",
             mask_dilation=0.01,
@@ -56,4 +63,4 @@ def edit_inpainting_insert_mask(output_file: str) -> Image:
 
 
 if __name__ == "__main__":
-    edit_inpainting_insert_mask(output_file="test_resources/fruit_edit.png")
+    edit_inpainting_insert_mask(output_file="output_folder/fruit_edit.png")

genai/image_generation/imggen_inpainting_insert_with_txt_img.py

@@ -18,15 +18,22 @@
 def edit_inpainting_insert(output_file: str) -> Image:
     # [START googlegenaisdk_imggen_inpainting_insert_with_txt_img]
     from google import genai
-    from google.genai.types import RawReferenceImage, MaskReferenceImage, MaskReferenceConfig, EditImageConfig
+    from google.genai.types import (
+        RawReferenceImage,
+        MaskReferenceImage,
+        MaskReferenceConfig,
+        EditImageConfig,
+    )
 
     client = genai.Client()
 
     # TODO(developer): Update and un-comment below line
     # output_file = "output-image.png"
 
     raw_ref = RawReferenceImage(
-        reference_image=Image.from_file(location='test_resources/fruit.png'), reference_id=0)
+        reference_image=Image.from_file(location="test_resources/fruit.png"),
+        reference_id=0,
+    )
     mask_ref = MaskReferenceImage(
         reference_id=1,
         reference_image=None,
@@ -56,4 +63,4 @@ def edit_inpainting_insert(output_file: str) -> Image:
 
 
 if __name__ == "__main__":
-    edit_inpainting_insert(output_file="test_resources/fruit_edit.png")
+    edit_inpainting_insert(output_file="output_folder/fruit_edit.png")

genai/image_generation/imggen_inpainting_removal_mask_with_txt_img.py

@@ -18,18 +18,25 @@
 def edit_inpainting_removal_mask(output_file: str) -> Image:
     # [START googlegenaisdk_imggen_inpainting_removal_mask_with_txt_img]
     from google import genai
-    from google.genai.types import RawReferenceImage, MaskReferenceImage, MaskReferenceConfig, EditImageConfig
+    from google.genai.types import (
+        RawReferenceImage,
+        MaskReferenceImage,
+        MaskReferenceConfig,
+        EditImageConfig,
+    )
 
     client = genai.Client()
 
     # TODO(developer): Update and un-comment below line
     # output_file = "output-image.png"
 
     raw_ref = RawReferenceImage(
-        reference_image=Image.from_file(location='test_resources/fruit.png'), reference_id=0)
+        reference_image=Image.from_file(location="test_resources/fruit.png"),
+        reference_id=0,
+    )
     mask_ref = MaskReferenceImage(
         reference_id=1,
-        reference_image=Image.from_file(location='test_resources/fruit_mask.png'),
+        reference_image=Image.from_file(location="test_resources/fruit_mask.png"),
         config=MaskReferenceConfig(
             mask_mode="MASK_MODE_USER_PROVIDED",
             mask_dilation=0.01,
@@ -56,4 +63,4 @@ def edit_inpainting_removal_mask(output_file: str) -> Image:
 
 
 if __name__ == "__main__":
-    edit_inpainting_removal_mask(output_file="test_resources/fruit_edit.png")
+    edit_inpainting_removal_mask(output_file="output_folder/fruit_edit.png")

genai/image_generation/imggen_inpainting_removal_with_txt_img.py

@@ -18,15 +18,22 @@
 def edit_inpainting_removal(output_file: str) -> Image:
     # [START googlegenaisdk_imggen_inpainting_removal_with_txt_img]
     from google import genai
-    from google.genai.types import RawReferenceImage, MaskReferenceImage, MaskReferenceConfig, EditImageConfig
+    from google.genai.types import (
+        RawReferenceImage,
+        MaskReferenceImage,
+        MaskReferenceConfig,
+        EditImageConfig,
+    )
 
     client = genai.Client()
 
     # TODO(developer): Update and un-comment below line
     # output_file = "output-image.png"
 
     raw_ref = RawReferenceImage(
-        reference_image=Image.from_file(location='test_resources/fruit.png'), reference_id=0)
+        reference_image=Image.from_file(location="test_resources/fruit.png"),
+        reference_id=0,
+    )
     mask_ref = MaskReferenceImage(
         reference_id=1,
         reference_image=None,
@@ -55,4 +62,4 @@ def edit_inpainting_removal(output_file: str) -> Image:
 
 
 if __name__ == "__main__":
-    edit_inpainting_removal(output_file="test_resources/fruit_edit.png")
+    edit_inpainting_removal(output_file="output_folder/fruit_edit.png")

genai/image_generation/imggen_mask_free_edit_with_txt_img.py

@@ -26,7 +26,9 @@ def edit_mask_free(output_file: str) -> Image:
     # output_file = "output-image.png"
 
     raw_ref = RawReferenceImage(
-        reference_image=Image.from_file(location='test_resources/latte.jpg'), reference_id=0)
+        reference_image=Image.from_file(location="test_resources/latte.jpg"),
+        reference_id=0,
+    )
 
     image = client.models.edit_image(
         model="imagen-3.0-capability-001",
@@ -48,4 +50,4 @@ def edit_mask_free(output_file: str) -> Image:
 
 
 if __name__ == "__main__":
-    edit_mask_free(output_file="test_resources/latte_edit.png")
+    edit_mask_free(output_file="output_folder/latte_edit.png")

genai/image_generation/imggen_mmflash_edit_img_with_txt_img.py

@@ -23,7 +23,7 @@ def generate_content() -> str:
     client = genai.Client()
 
     # Using an image of Eiffel tower, with fireworks in the background.
-    image = Image.open("example-image.png")
+    image = Image.open("test_resources/example-image-eiffel-tower.png")
 
     response = client.models.generate_content(
         model="gemini-2.0-flash-preview-image-generation",
@@ -35,15 +35,15 @@ def generate_content() -> str:
             print(part.text)
         elif part.inline_data:
             image = Image.open(BytesIO((part.inline_data.data)))
-            image.save("bw-example-image.png")
+            image.save("output_folder/bw-example-image.png")
     # Example response:
     #  Here's the cartoon-style edit of the image:
     #  Cartoon-style edit:
     #  - Simplified the Eiffel Tower with bolder lines and slightly exaggerated proportions.
     #  - Brightened and saturated the colors of the sky, fireworks, and foliage for a more vibrant, cartoonish look.
     #  ....
     # [END googlegenaisdk_imggen_mmflash_edit_img_with_txt_img]
-    return "bw-example-image.png"
+    return "output_folder/bw-example-image.png"
 
 
 if __name__ == "__main__":

genai/image_generation/imggen_mmflash_locale_aware_with_txt.py

@@ -0,0 +1,45 @@
+# 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
+#
+#    https://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.
+
+
+def generate_content() -> str:
+    # [START googlegenaisdk_imggen_mmflash_locale_aware_with_txt]
+    from google import genai
+    from google.genai.types import GenerateContentConfig, Modality
+    from PIL import Image
+    from io import BytesIO
+
+    client = genai.Client()
+
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-preview-image-generation",
+        contents=("Generate a photo of a breakfast meal."),
+        config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
+    )
+    for part in response.candidates[0].content.parts:
+        if part.text:
+            print(part.text)
+        elif part.inline_data:
+            image = Image.open(BytesIO((part.inline_data.data)))
+            image.save("output_folder/example-breakfast-meal.png")
+    # Example response:
+    #   Generates a photo of a vibrant and appetizing breakfast meal.
+    #   The scene will feature a white plate with golden-brown pancakes
+    #   stacked neatly, drizzled with rich maple syrup and ...
+    # [END googlegenaisdk_imggen_mmflash_locale_aware_with_txt]
+    return "output_folder/example-breakfast-meal.png"
+
+
+if __name__ == "__main__":
+    generate_content()

genai/image_generation/imggen_mmflash_multiple_imgs_with_txt.py

@@ -0,0 +1,58 @@
+# 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
+#
+#    https://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.
+
+
+def generate_content() -> str:
+    # [START googlegenaisdk_imggen_mmflash_multiple_imgs_with_txt]
+    from google import genai
+    from google.genai.types import GenerateContentConfig, Modality
+    from PIL import Image
+    from io import BytesIO
+
+    client = genai.Client()
+
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-preview-image-generation",
+        contents=("Generate 3 images a cat sitting on a chair."),
+        config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
+    )
+    saved_files = []
+    image_counter = 1
+    for part in response.candidates[0].content.parts:
+        if part.text:
+            print(part.text)
+        elif part.inline_data:
+            image = Image.open(BytesIO((part.inline_data.data)))
+            filename = f"output_folder/example-cats-0{image_counter}.png"
+            image.save(filename)
+            saved_files.append(filename)
+            image_counter += 1
+    # Example response:
+    #   Image 1: A fluffy calico cat with striking green eyes is perched elegantly on a vintage wooden
+    #   chair with a woven seat. Sunlight streams through a nearby window, casting soft shadows and
+    #   highlighting the cat's fur.
+    #
+    #   Image 2: A sleek black cat with intense yellow eyes is sitting upright on a modern, minimalist
+    #   white chair. The background is a plain grey wall, putting the focus entirely on the feline's
+    #   graceful posture.
+    #
+    #   Image 3: A ginger tabby cat with playful amber eyes is comfortably curled up asleep on a plush,
+    #   oversized armchair upholstered in a soft, floral fabric. A corner of a cozy living room with a
+    #   warm lamp in the background can be seen.
+    # [END googlegenaisdk_imggen_mmflash_multiple_imgs_with_txt]
+    return saved_files
+
+
+if __name__ == "__main__":
+    generate_content()

genai/image_generation/imggen_mmflash_txt_and_img_with_txt.py

@@ -30,14 +30,14 @@ def generate_content() -> int:
         ),
         config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
     )
-    with open("paella-recipe.md", "w") as fp:
+    with open("output_folder/paella-recipe.md", "w") as fp:
         for i, part in enumerate(response.candidates[0].content.parts):
             if part.text is not None:
                 fp.write(part.text)
             elif part.inline_data is not None:
                 image = Image.open(BytesIO((part.inline_data.data)))
-                image.save(f"example-image-{i+1}.png")
-                fp.write(f"![image](./example-image-{i+1}.png)")
+                image.save(f"output_folder/example-image-{i+1}.png")
+                fp.write(f"![image](example-image-{i+1}.png)")
     # Example response:
     #  A markdown page for a Paella recipe(`paella-recipe.md`) has been generated.
     #   It includes detailed steps and several images illustrating the cooking process.