Image Generation

from lyzr import Studio
from lyzr.image_models import Gemini, DallE

studio = Studio(api_key="your-api-key")

# Create agent with image generation
agent = studio.create_agent(
    name="Image Creator",
    provider="gpt-4o",
    role="Visual designer",
    goal="Create images from descriptions",
    image_model=Gemini.PRO  # Enable image generation
)

# Generate an image
response = agent.run("Create an image of a futuristic city at sunset")

# Access generated images
if response.has_files():
    for img in response.files:
        print(f"Image URL: {img.url}")
        img.download("./images/city.png")

from lyzr.image_models import Gemini

# Available Gemini models
Gemini.PRO     # gemini/gemini-3-pro-image-preview - High quality
Gemini.FLASH   # gemini/gemini-2.5-flash-image - Fast generation

from lyzr.image_models import DallE

# Available DALL-E models
DallE.DALL_E_3      # dall-e-3 - Highest quality
DallE.DALL_E_2      # dall-e-2 - Standard quality
DallE.GPT_IMAGE_1   # gpt-image-1 - GPT-based
DallE.GPT_IMAGE_1_5 # gpt-image-1.5 - Enhanced GPT-based

from lyzr.image_models import Gemini

agent = studio.create_agent(
    name="Image Agent",
    provider="gpt-4o",
    role="Image creator",
    image_model=Gemini.PRO  # Use Gemini PRO
)

from lyzr.image_models import DallE

# Create with one model
agent = studio.create_agent(
    name="Image Agent",
    provider="gpt-4o",
    image_model=Gemini.PRO
)

# Switch to different model
agent = agent.set_image_model(DallE.DALL_E_3)

from lyzr.image_models import Gemini

config = Gemini.PRO
print(config.model)          # gemini/gemini-3-pro-image-preview
print(config.credential_id)  # lyzr_google
print(config.provider)       # ImageProvider.GOOGLE

response = agent.run("Create an illustration")

if response.has_files():
    images = [f for f in response.files if f.format_type == "image"]
    print(f"Generated {len(images)} images")

for img in response.files:
    print(f"Name: {img.name}")
    print(f"Format: {img.format_type}")
    print(f"URL: {img.url}")
    print(f"ID: {img.artifact_id}")

# Download single image
response.files[0].download("./output/image.png")

# Download all images
for img in response.files:
    img.download(f"./images/{img.name}")

from lyzr.image_models import DallE

agent = studio.create_agent(
    name="Product Visualizer",
    provider="gpt-4o",
    role="Product photographer",
    goal="Create professional product images",
    instructions="Generate clean, professional product images on white backgrounds",
    image_model=DallE.DALL_E_3
)

response = agent.run("""
Create a product image:
- Subject: Modern wireless earbuds
- Style: Clean, minimal, white background
- Lighting: Soft studio lighting
- Angle: Three-quarter view showing both earbuds
""")

if response.has_files():
    response.files[0].download("./products/earbuds.png")

from lyzr.image_models import Gemini

agent = studio.create_agent(
    name="Marketing Designer",
    provider="gpt-4o",
    role="Marketing graphics designer",
    goal="Create eye-catching marketing visuals",
    image_model=Gemini.PRO
)

response = agent.run("""
Create a social media banner for a summer sale:
- Theme: Tropical, vibrant colors
- Text to include: "SUMMER SALE - UP TO 50% OFF"
- Size: Wide banner format
- Style: Modern, energetic
""")

if response.has_files():
    response.files[0].download("./marketing/summer_sale_banner.png")

from lyzr.image_models import DallE

agent = studio.create_agent(
    name="Illustrator",
    provider="gpt-4o",
    role="Digital illustrator",
    goal="Create beautiful illustrations",
    instructions="Generate high-quality digital illustrations",
    image_model=DallE.DALL_E_3
)

response = agent.run("""
Create a children's book illustration:
- Scene: A friendly dragon reading a book in a cozy library
- Style: Warm, whimsical, colorful
- Mood: Peaceful, inviting
- Details: Soft lighting, books stacked around, comfortable reading nook
""")

if response.has_files():
    response.files[0].download("./illustrations/dragon_reading.png")

from lyzr.image_models import Gemini

agent = studio.create_agent(
    name="Diagram Creator",
    provider="gpt-4o",
    role="Technical illustrator",
    goal="Create clear technical diagrams",
    image_model=Gemini.FLASH
)

response = agent.run("""
Create a system architecture diagram showing:
- Frontend (React)
- API Gateway
- Microservices (3 services)
- Database layer
- Cache layer
Use clean, professional style with labeled components
""")

if response.has_files():
    response.files[0].download("./diagrams/architecture.png")

from lyzr.image_models import DallE

agent = studio.create_agent(
    name="Social Media Creator",
    provider="gpt-4o",
    role="Social media content creator",
    goal="Create engaging social media visuals",
    image_model=DallE.DALL_E_3
)

prompts = [
    "Create an Instagram post for a coffee shop featuring a latte art",
    "Create a LinkedIn banner for a tech company",
    "Create a Twitter header for a podcast about AI"
]

for prompt in prompts:
    response = agent.run(prompt)
    if response.has_files():
        name = prompt.split()[3]  # Simplified naming
        response.files[0].download(f"./social/{name}_post.png")

# Good: Detailed description
response = agent.run("""
Create an image of a modern office space:
- Large windows with city view
- Minimalist furniture in neutral colors
- Green plants for accents
- Natural lighting
- A few people working at desks
- Style: Photorealistic
""")

# Less effective: Vague description
response = agent.run("Create an office image")

# Specify artistic style
response = agent.run("""
Create a portrait in the style of:
- Impressionist painting
- Soft brushstrokes
- Pastel color palette
- Natural lighting
""")

styles = ["photorealistic", "watercolor", "oil painting", "digital art"]
subject = "mountain landscape at sunrise"

for style in styles:
    response = agent.run(f"Create a {style} image of a {subject}")
    if response.has_files():
        response.files[0].download(f"./variations/{style}_landscape.png")

try:
    response = agent.run("Generate an image")

    if response.has_files():
        for img in response.files:
            try:
                img.download(f"./images/{img.name}")
                print(f"Saved: {img.name}")
            except Exception as e:
                print(f"Download failed: {e}")
                print(f"URL: {img.url}")  # Manual download fallback
    else:
        print("No images generated")
        print(f"Response: {response.response}")

except Exception as e:
    print(f"Generation error: {e}")

from lyzr.image_models import Gemini

agent = studio.create_agent(
    name="Content Creator",
    provider="gpt-4o",
    role="Marketing content creator",
    file_output=True,           # Enable file generation
    image_model=Gemini.PRO      # Enable image generation
)

response = agent.run("""
Create a complete marketing package:
1. A PDF brochure with product information
2. A hero image for the website
3. Social media graphics
""")

for artifact in response.files:
    if artifact.format_type == "pdf":
        artifact.download("./marketing/brochure.pdf")
    elif artifact.format_type == "image":
        artifact.download(f"./marketing/images/{artifact.name}")