from google import genai
from google.genai import types
from PIL import Image
from io import BytesIO
import os

# Setup Client
client = genai.Client(api_key="AIzaSyB-nR0nkeftKrd42NrNIDcFCj3yFP8JLtw")

# Define the Prompt
prompt = "Create a simple modern app icon with a blue background and white envelope symbol, square format, flat design, minimalist"

print(f"Asking Gemini to generate: '{prompt}'...")

# Call the Gemini Model
try:
    response = client.models.generate_content(
        model='gemini-2.5-flash-image',
        contents=prompt,
        config=types.GenerateContentConfig(
            response_modalities=["IMAGE"]  # Tell Gemini to draw, not talk
        )
    )

    # Save the Image
    if response.candidates[0].content.parts:
        for part in response.candidates[0].content.parts:
            if part.inline_data:
                image_data = part.inline_data.data
                image = Image.open(BytesIO(image_data))
                image.save("test_gemini_native.png")
                print("Success! Saved test_gemini_native.png")
            else:
                print("Model returned text instead of image:", part.text)
    else:
        print("No content parts in response")

except Exception as e:
    print(f"Error: {type(e).__name__}: {e}")
    import traceback
    traceback.print_exc()