Write a weekly AI brief, save it as a PDF in Drive and share it in Slack

from datetime import date
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from googleapiclient.http import MediaInMemoryUpload
from clickruntime import on_trigger, credentials, ai, notify

NOTES_DOC_ID = "your-notes-doc-id"
FOLDER_ID = "your-briefs-folder-id"

@on_trigger
def run(event):
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    notes = drive.files().export(fileId=NOTES_DOC_ID, mimeType="text/plain").execute().decode()[:20000]
    brief = ai.generate(f"Write a one-page pricing brief with sections Summary, Changes, Risks, Actions from these notes:\n{notes}",
                        max_output_tokens=1500).text
    title = f"Pricing brief {date.today().isoformat()}"
    doc = drive.files().create(body={"name": title, "mimeType": "application/vnd.google-apps.document"},
                               media_body=MediaInMemoryUpload(brief.encode(), mimetype="text/plain")).execute()
    pdf = drive.files().export(fileId=doc["id"], mimeType="application/pdf").execute()
    saved = drive.files().create(body={"name": f"{title}.pdf", "parents": [FOLDER_ID]},
                                 media_body=MediaInMemoryUpload(pdf, mimetype="application/pdf"),
                                 fields="id,webViewLink").execute()
    drive.files().delete(fileId=doc["id"]).execute()
    notify.slack(f"This week's pricing brief: {saved['webViewLink']}", channel="#strategy")
    return {"file": saved["id"]}