Add AI chapters to your new YouTube videos

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, report_fetched, http, credentials, ai, state

def api(name, version):
    return build(name, version, credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)

AT = "https://api.airtable.com/v0/appXXXXXXXXXXXXXX"

def at(method, path, **kw):
    r = http.request(method, f"{AT}/{path}", headers={"Authorization": f"Bearer {credentials.connection('airtable', 'work')}"}, **kw)
    return r.raise_for_status() or r.json()

@on_trigger
def run(event):
    yt, done = api("youtube", "v3"), set(state.get("done") or [])
    uploads = yt.channels().list(part="contentDetails", mine=True).execute()["items"][0]["contentDetails"]["relatedPlaylists"]["uploads"]
    ids = [i["contentDetails"]["videoId"] for i in yt.playlistItems().list(part="contentDetails", playlistId=uploads, maxResults=5).execute()["items"]]
    new = [v for v in ids if v not in done and yt.captions().list(part="id", videoId=v).execute().get("items")]
    report_fetched(len(new))
    if not new:
        return {"summary": "No new videos with captions"}
    for vid in new:
        track = yt.captions().list(part="id", videoId=vid).execute()["items"][0]["id"]
        srt = yt.captions().download(id=track, tfmt="srt").execute().decode()[:30000]
        chapters = ai.generate(srt, system="Write 5-10 YouTube chapters as 'MM:SS Title' lines. The first must be 00:00.", max_output_tokens=300).text.strip()
        s = yt.videos().list(part="snippet", id=vid).execute()["items"][0]["snippet"]
        if "00:00" not in s["description"]:
            yt.videos().update(part="snippet", body={"id": vid, "snippet": {"title": s["title"], "categoryId": s["categoryId"],
                               "description": f"{s['description']}\n\nChapters\n{chapters}"}}).execute()
            at("POST", "Chapters", json={"records": [{"fields": {"Video ID": vid, "Title": s["title"], "Chapters": chapters}}]})
        done.add(vid)
    state.set("done", sorted(done)[-500:])
    return {"chaptered": len(new)}