Generate images from prompts in a sheet and save them to Drive

import base64
from googleapiclient.http import MediaInMemoryUpload
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, http, report_fetched

SHEET_ID, FOLDER_ID = "your-images-sheet-id", "your-drive-folder-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    values = build("sheets", "v4", credentials=creds, cache_discovery=False).spreadsheets().values()
    rows = values.get(spreadsheetId=SHEET_ID, range="Images!A2:B").execute().get("values", [])
    todo = [(i, r[0]) for i, r in enumerate(rows, start=2) if r and r[0] and len(r) < 2]
    report_fetched(len(todo))
    if not todo:
        return {"made": 0}
    drive = build("drive", "v3", credentials=creds, cache_discovery=False)
    for i, prompt in todo:
        img = http.post("https://api.openai.com/v1/images/generations", timeout=120,
                        headers={"Authorization": f"Bearer {credentials.secret('IMAGE_API_KEY')}"},
                        json={"model": "gpt-image-1", "prompt": prompt, "size": "1024x1024"}).json()["data"][0]["b64_json"]
        f = drive.files().create(body={"name": f"{prompt[:60]}.png", "parents": [FOLDER_ID]}, fields="webViewLink",
                                 media_body=MediaInMemoryUpload(base64.b64decode(img), mimetype="image/png")).execute()
        values.update(spreadsheetId=SHEET_ID, range=f"Images!B{i}", valueInputOption="RAW", body={"values": [[f["webViewLink"]]]}).execute()
    return {"made": len(todo)}