Image Generation Model

This example shows how to use the image generation model to generate images with whatsapp app.

Code

cookbook/apps/whatsapp/image_generation_model.py

from agno.agent import Agent
from agno.app.whatsapp.app import WhatsappAPI
from agno.models.google import Gemini

image_agentg = Agent(
    model=Gemini(
        id="gemini-2.0-flash-exp-image-generation",
        response_modalities=["Text", "Image"],
    ),
    debug_mode=True,
)

whatsapp_app = WhatsappAPI(
    agent=image_agent,
    name="Image Generation Model",
    app_id="image_generation_model",
    description="A model that generates images using the Gemini API.",
)

app = whatsapp_app.get_app()

if __name__ == "__main__":
    whatsapp_app.serve(app="image_generation_model:app", port=8000, reload=True)

Usage

1. Create a virtual environment

   Open the Terminal and create a python virtual environment.

   Mac

   python3 -m venv .venv
   source .venv/bin/activate
   

2. Set your API key

   export GOOGLE_API_KEY=xxx
   

3. Install libraries

   pip install -U agno google-generativeai "uvicorn[standard]"
   

4. Run Agent

   Mac

   python cookbook/apps/whatsapp/image_generation_model.py