Content Generation Agent with Image Creation

from deepagents import create_deep_agent
from deepagents.backends import FilesystemBackend
from langchain_core.tools import tool

@tool
def generate_cover(prompt: str, slug: str) -> str:
    """Generate a cover image for a blog post."""
    from google import genai
    client = genai.Client()
    response = client.models.generate_content(
        model="gemini-2.5-flash-image",
        contents=[prompt],
    )
    for part in response.parts:
        if part.inline_data is not None:
            image = part.as_image()
            image.save(f"blogs/{slug}/hero.png")
            return f"Image saved to blogs/{slug}/hero.png"
    return "No image generated"

agent = create_deep_agent(
    memory=["./AGENTS.md"],
    skills=["./skills/"],
    tools=[generate_cover],
    subagents=[{"name": "researcher", "description": "Research topics", "tools": [web_search]}],
    backend=FilesystemBackend(root_dir="."),
)