Turn emailed press releases into draft articles in Drive

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

FOLDER = "articles-folder-id"
STYLE = "Neutral news style, 300 words, headline first, no superlatives."

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    gmail = build("gmail", "v1", credentials=creds, cache_discovery=False).users()
    msgs = gmail.messages().list(userId="me", q='label:press-releases -label:drafted').execute().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"summary": "No new releases"}
    for m in msgs:
        snippet = gmail.messages().get(userId="me", id=m["id"]).execute()["snippet"]
        body = gmail.messages().get(userId="me", id=m["id"], format="full").execute()["payload"]
        text = snippet if not body.get("parts") else str(body["parts"][0].get("body", {}).get("data", snippet))
        article = ai.generate(f"{STYLE}\nRewrite as an article:\n{text[:8000]}", max_output_tokens=900).text
        doc = http.post("https://www.googleapis.com/upload/drive/v3/files?uploadType=multipart", headers={"Authorization": f"Bearer {creds.token}"},
                        files={"metadata": (None, f'{{"name": "{article.splitlines()[0][:80]}", "mimeType": "application/vnd.google-apps.document", "parents": ["{FOLDER}"]}}', "application/json"),
                               "file": ("a.txt", article, "text/plain")}).json()
        notify.email_me("New article draft", f"https://docs.google.com/document/d/{doc['id']}/edit")
    return {"drafted": len(msgs)}