Enhancing Images with Styles and Quality Modifiers in Python

A photo of a corn stalk

A beautiful high-detail HDR photo of a corn stalk taken by a professional photographer

A painting of a futuristic city

A photorealistic aerial photo of a futuristic city

from google import genai
from google.genai import types
import os
import time

GEMINI_IMAGE_MODEL = "gemini-3.1-flash-image"

api_key = os.getenv("GEMINI_API_KEY")

if not api_key:
    raise ValueError("GEMINI_API_KEY not found in environment variables. Set it before running the script.")

base_url = os.getenv("GEMINI_BASE_URL")

if not base_url:
    raise ValueError("GEMINI_BASE_URL not found in environment variables. Set it before running the script.")

client = genai.Client(
    api_key=api_key,
    http_options=types.HttpOptions(
        base_url=base_url,
    ),
)

style_quality_prompts = [
    "A beautiful landscape, photorealistic style, high-resolution details",
    "A portrait of a woman, oil painting style, high definition",
    "A futuristic cityscape, cyberpunk style, ultra-realistic, cinematic detail",
    "A serene beach scene, watercolor painting, high-quality details",
]

for index, prompt in enumerate(style_quality_prompts):
    response = client.models.generate_content(
        model=GEMINI_IMAGE_MODEL,
        contents=prompt,
        config=types.GenerateContentConfig(
            response_modalities=["IMAGE"],
            image_config=types.ImageConfig(
                aspect_ratio="1:1",
            )
        ),
    )

    image_parts = [part for part in response.parts if part.inline_data]

    if not image_parts:
        print(f"No image generated for prompt: {prompt}")
        continue

    image = image_parts[0].as_image()
    timestamp = int(time.time() * 1000)
    image_filename = f"image_{timestamp}_{index}.png"
    output_path = os.path.join("static", "images", image_filename)
    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    image.save(output_path)