import google.generativeai as genai
import os
from PIL import Image
import io

# Configure API Key
genai.configure(api_key="AIzaSyB-nR0nkeftKrd42NrNIDcFCj3yFP8JLtw")

# Try to initialize the model
try:
    generation_model = genai.GenerativeModel('gemini-pro-vision')
    print("Model initialized successfully")

    # Define prompt
    prompt = "A modern app icon with a blue background and white envelope symbol"
    print(f"Generating image for: '{prompt}'...")

    # Try to generate
    response = generation_model.generate_content(prompt)

    print(f"Response type: {type(response)}")
    print(f"Response: {response}")

    if hasattr(response, 'parts') and response.parts:
        print(f"Parts: {response.parts}")
        if hasattr(response.parts[0], 'inline_data'):
            print("Has inline_data!")
            image_data = response.parts[0].inline_data.data
            image = Image.open(io.BytesIO(image_data))
            image.save("test_output.png")
            print("Success! Image saved to test_output.png")
        else:
            print("No inline_data attribute")
            print(f"Part attributes: {dir(response.parts[0])}")
    else:
        print("No parts in response or response.parts doesn't exist")
        if hasattr(response, 'text'):
            print(f"Response text: {response.text}")

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