Image Gen 과 PDF Processing

from google import genai
from google.genai import types
from pathlib import Path

client = genai.Client()

response = client.models.generate_content(
    model='gemini-3-pro-image-preview',  # preview image model
    contents='A red-haired anime girl with blue eyes sitting at a desk full of code, '
             'soft warm light, oil painting style, cozy atmosphere.',
    config={'response_modalities': ['IMAGE']},
)

for part in response.parts:
    if part.mime_type and part.mime_type.startswith('image/'):
        Path('out.png').write_bytes(part.as_image().image_bytes)
        print('Wrote out.png')

import httpx, io

# Upload from local or from a URL
pdf_bytes = httpx.get('https://arxiv.org/pdf/2305.10601.pdf').content
doc = client.files.upload(
    file=io.BytesIO(pdf_bytes),
    config=dict(mime_type='application/pdf'),
)

response = client.models.generate_content(
    model='gemini-2.5-flash',
    contents=[
        'Summarize this paper in 5 bullet points. Focus on the contribution and method.',
        doc,
    ],
)
print(response.text)

client.files.delete(name=doc.name)

img_bytes = Path('original.png').read_bytes()

response = client.models.generate_content(
    model='gemini-3-pro-image-preview',
    contents=[
        types.Part.from_bytes(data=img_bytes, mime_type='image/png'),
        'Replace the cloudy sky with a starry night sky. Keep everything else exactly the same.',
    ],
    config={'response_modalities': ['IMAGE']},
)

for part in response.parts:
    if part.mime_type and part.mime_type.startswith('image/'):
        Path('edited.png').write_bytes(part.as_image().image_bytes)