chore: code refactor

Author: msampathkumar

genai/image_generation/imggen_mmflash_edit_img_with_txt_img.py

@@ -16,7 +16,7 @@
 def generate_content() -> str:
     # [START googlegenaisdk_imggen_mmflash_edit_img_with_txt_img]
     from google import genai
-    from google.genai.types import GenerateContentConfig
+    from google.genai.types import GenerateContentConfig, Modality
     from PIL import Image
     from io import BytesIO
 
@@ -28,7 +28,7 @@ def generate_content() -> str:
     response = client.models.generate_content(
         model="gemini-2.0-flash-exp",
         contents=[image, "Edit this image to make it look like a cartoon."],
-        config=GenerateContentConfig(response_modalities=["Text", "Image"]),
+        config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
     )
     for part in response.candidates[0].content.parts:
         if part.text:

genai/image_generation/imggen_mmflash_txt_and_img_with_txt.py

@@ -16,7 +16,7 @@
 def generate_content() -> int:
     # [START googlegenaisdk_imggen_mmflash_txt_and_img_with_txt]
     from google import genai
-    from google.genai.types import GenerateContentConfig
+    from google.genai.types import GenerateContentConfig, Modality
     from PIL import Image
     from io import BytesIO
 
@@ -28,7 +28,7 @@ def generate_content() -> int:
             "Generate an illustrated recipe for a paella."
             "Create images to go alongside the text as you generate the recipe"
         ),
-        config=GenerateContentConfig(response_modalities=["Text", "Image"]),
+        config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
     )
     with open("paella-recipe.md", "w") as fp:
         for i, part in enumerate(response.candidates[0].content.parts):

genai/image_generation/imggen_mmflash_with_txt.py

@@ -16,7 +16,7 @@
 def generate_content() -> str:
     # [START googlegenaisdk_imggen_mmflash_with_txt]
     from google import genai
-    from google.genai.types import GenerateContentConfig
+    from google.genai.types import GenerateContentConfig, Modality
     from PIL import Image
     from io import BytesIO
 
@@ -27,7 +27,7 @@ def generate_content() -> str:
         contents=(
             "Generate an image of the Eiffel tower with fireworks in the background."
         ),
-        config=GenerateContentConfig(response_modalities=["Text", "Image"]),
+        config=GenerateContentConfig(response_modalities=[Modality.TEXT, Modality.IMAGE]),
     )
     for part in response.candidates[0].content.parts:
         if part.text: