feat: Convert `google.genai.types.Image` to `PIL.Image` for Gradio display using `io.BytesIO`.

app.py

@@ -117,7 +117,18 @@ The output must be a single image that looks like the completed infographic, mai
         # Extract and return the generated image
         for part in response.parts:
             if image := part.as_image():
-                return image
+                # Convert google.genai.types.Image to PIL Image for Gradio
+                try:
+                    # Try saving to a bytes buffer
+                    img_buffer = io.BytesIO()
+                    image.save(img_buffer)
+                    img_buffer.seek(0)
+                    return Image.open(img_buffer)
+                except Exception:
+                    # Fallback: if save() doesn't accept buffer, try accessing bytes directly if possible
+                    if hasattr(image, '_image_bytes'):
+                         return Image.open(io.BytesIO(image._image_bytes))
+                    raise
         
         raise gr.Error("The model completed but did not return an image. Check inputs or safety filters.")