Categorise bank statement transactions into Supabase

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

FOLDER_ID, SUPABASE = "your-statements-folder-id", "https://yourproject.supabase.co"

@on_trigger
def run(event):
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    seen = set(state.get("seen") or [])
    files = drive.files().list(q=f"'{FOLDER_ID}' in parents and mimeType='text/csv' and trashed=false", fields="files(id,name)").execute()["files"]
    new = [f for f in files if f["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"summary": "Nothing new"}
    key = credentials.secret("SUPABASE_SERVICE_KEY")
    sb = {"apikey": key, "Authorization": f"Bearer {key}"}
    cats = [c["name"] for c in http.get(f"{SUPABASE}/rest/v1/categories", headers=sb, params={"select": "name"}).json()]
    for f in new:
        rows = list(csv.DictReader(io.StringIO(drive.files().get_media(fileId=f["id"]).execute().decode())))
        for i in range(0, len(rows), 50):
            batch = rows[i:i + 50]
            listing = "\n".join(f"{n + 1}. {r['Description']} {r['Amount']}" for n, r in enumerate(batch))
            labels = ai.generate(f"Categories: {', '.join(cats)}\nGive one category per line for:\n{listing}", max_output_tokens=600).text.splitlines()
            http.post(f"{SUPABASE}/rest/v1/transactions", headers=sb, json=[{"date": r["Date"], "description": r["Description"], "amount": float(r["Amount"]),
                      "category": (labels[n].split(". ", 1)[-1].strip() if n < len(labels) else "Uncategorised")} for n, r in enumerate(batch)])
        seen.add(f["id"])
    state.set("seen", list(seen)[-500:])
    return {"files": len(new)}