Call simple gen-Image tool that can show image in ADK Web UI.

[NBestsuNgan]

Credit https://www.youtube.com/watch?v=HoXyGnPcEzA&t=329s

Just wanna share, I believe that there are many people who struggle with this use case and please for the youtube channel above is really good

import os
from typing import Dict, Any
from datetime import date
from google.genai import types
from google.adk.agents import Agent
from google.adk.agents.callback_context import CallbackContext
from google.adk.tools import load_artifacts
import base64
from google.adk.tools.tool_context import ToolContext
from .prompts import return_instructions_root
# from .tools import call_db_agent, call_ds_agent
import requests
import google.genai.types as types
    
async def gen_image(tool_context: ToolContext) -> Dict[str, Any]:


    image_bytes = ... #Your byte image data

    image_artifact = types.Part(
        inline_data=types.Blob(data=image_bytes, mime_type="image/png")
    )
    
    # Await the async method
    artifact_version = await tool_context.save_artifact(
        filename=file_name, artifact=image_artifact
    )
    
    # This might also need to be awaited
    await tool_context.load_artifact(filename=file_name)
    
    tool_context.state["file_name"] = file_name
    tool_context.state["file_version"] = artifact_version

    return {"status": "success"}

root_agent = Agent(
    name="test_agent",
    model="gemini-2.0-flash",
    description="test_agent",
    instruction=return_instructions_root(),
    tools=[gen_image],
)