Write captions for Drive videos and track them in Airtable

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

BASE_ID, FOLDER_ID = "appXXXXXXXXXXXXXX", "1FolderIdOfReels"
TONE = "Friendly neighbourhood café, warm and a little playful, no more than two emojis."
SCHEMA = {"type": "object", "properties": {"caption": {"type": "string"}, "hashtags": {"type": "string"}},
          "required": ["caption", "hashtags"]}

@on_trigger
def run(event):
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    since = state.get("since") or "2026-01-01T00:00:00Z"
    q = (f"'{FOLDER_ID}' in parents and createdTime > '{since}' and trashed = false "
         "and (mimeType contains 'video/' or mimeType contains 'image/')")
    files = drive.files().list(q=q, orderBy="createdTime",
                               fields="files(id,name,description,createdTime,webViewLink)").execute()["files"]
    report_fetched(len(files))
    if not files:
        return {"new": 0}
    headers = {"Authorization": f"Bearer {credentials.connection('airtable', 'work')}"}
    for f in files:
        post = ai.generate(f"File name: {f['name']}\nDescription: {f.get('description', '')}", json_schema=SCHEMA,
                           system=f"Write a short social caption (under 40 words) and 5 hashtags. Tone: {TONE}",
                           max_output_tokens=200).json
        http.post(f"https://api.airtable.com/v0/{BASE_ID}/Content", headers=headers, json={"typecast": True,
                  "records": [{"fields": {"File": f["name"], "Link": f["webViewLink"], "Caption": post["caption"],
                               "Hashtags": post["hashtags"], "Status": "Ready to post"}}]}).raise_for_status()
        state.set("since", f["createdTime"])
    return {"new": len(files)}