Get a risk review of every new contract in Slack

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

FOLDER_ID = "your-contracts-folder-id"

@on_trigger
def run(event):
    token = credentials.connection("google", "work")
    drive = build("drive", "v3", credentials=Credentials(token), cache_discovery=False)
    seen = set(state.get("seen") or [])
    files = drive.files().list(q=f"'{FOLDER_ID}' in parents and mimeType='application/pdf' and trashed=false",
                               fields="files(id,name,webViewLink)").execute().get("files", [])
    new = [f for f in files if f["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    for f in new:
        pdf = http.get(f"https://www.googleapis.com/drive/v3/files/{f['id']}?alt=media",
                       headers={"Authorization": f"Bearer {token}"}).content
        text = "\n".join(p.extract_text() or "" for p in PdfReader(io.BytesIO(pdf)).pages)[:60000]
        review = ai.generate(["List the key terms and any risky clauses in this contract as short bullets.", text],
                             max_output_tokens=700).text
        notify.slack(f"*{f['name']}*\n{review}\n{f['webViewLink']}", channel="#legal")
        seen.add(f["id"])
    state.set("seen", list(seen))
    return {"new": len(new)}