Get reply drafts for customer emails

import base64
from email.message import EmailMessage
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, state, ai, notify, report_fetched

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    seen = set(state.get("seen") or [])
    msgs = gmail.users().messages().list(userId="me", q="in:inbox category:primary newer_than:1d", maxResults=15).execute().get("messages", [])
    new = [m for m in msgs if m["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    for m in new:
        msg = gmail.users().messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        reply = ai.generate(f"From: {h['From']}\nSubject: {h['Subject']}\n{msg['snippet']}", max_output_tokens=350,
                            system="Write a short, friendly customer reply. Don't promise refunds or dates.").text
        draft = EmailMessage()
        draft["To"], draft["Subject"] = h["From"], "Re: " + h["Subject"]
        draft.set_content(reply)
        created = gmail.users().drafts().create(userId="me", body={"message": {
            "raw": base64.urlsafe_b64encode(draft.as_bytes()).decode(), "threadId": msg["threadId"]}}).execute()
        notify.slack(f"✉️ Draft ready for {h['From']}: {h['Subject']}\n"
                     f"https://mail.google.com/mail/u/0/#drafts?compose={created['message']['id']}", channel="#me")
    state.set("seen", [m["id"] for m in msgs])
    return {"drafts": len(new)}