Get your important emails in Slack

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

LABELS = ["urgent", "client", "invoice", "newsletter"]

@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 newer_than:1d", maxResults=25).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}
    label_ids = {l["name"]: l["id"] for l in gmail.users().labels().list(userId="me").execute()["labels"]}
    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"]}
        r = ai.generate(f"From: {h.get('From')}\nSubject: {h.get('Subject')}\n{msg['snippet']}", max_output_tokens=60,
                        json_schema={"type": "object", "properties": {"category": {"enum": LABELS}, "summary": {"type": "string"}}}).json
        if r["category"] in label_ids:
            gmail.users().messages().modify(userId="me", id=m["id"], body={"addLabelIds": [label_ids[r["category"]]]}).execute()
        if r["category"] == "urgent":
            notify.slack(f"🚨 {h.get('From')}: {r['summary']}", channel="#me")
    state.set("seen", [m["id"] for m in msgs])
    return {"triaged": len(new)}