Summarise and sort new emails into a Notion inbox

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

DB_ID = "your-notion-database-id"
SCHEMA = {"type": "object", "properties": {"summary": {"type": "string"}, "category": {"type": "string",
          "enum": ["Customer", "Finance", "Partner", "Newsletter", "Other"]}, "action": {"type": "string"}}}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users().messages()
    seen = state.get("seen") or []
    new = [m["id"] for m in gmail.list(userId="me", q="in:inbox newer_than:1d -category:promotions").execute().get("messages", []) if m["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    notion = {"Authorization": f"Bearer {credentials.connection('notion', 'work')}", "Notion-Version": "2022-06-28"}
    for mid in new[:40]:
        m = gmail.get(userId="me", id=mid, format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in m["payload"]["headers"]}
        try:
            r = ai.generate(f"From: {h.get('From')}\nSubject: {h.get('Subject')}\n{m['snippet']}", json_schema=SCHEMA, max_output_tokens=120).json
        except ai.AIError:
            r = {"summary": m["snippet"][:200], "category": "Other", "action": ""}
        http.post("https://api.notion.com/v1/pages", headers=notion, json={"parent": {"database_id": DB_ID}, "properties": {
            "Name": {"title": [{"text": {"content": h.get("Subject", "(no subject)")[:200]}}]},
            "Category": {"select": {"name": r["category"]}}, "Summary": {"rich_text": [{"text": {"content": r["summary"]}}]},
            "Next action": {"rich_text": [{"text": {"content": r["action"]}}]}, "Link": {"url": f"https://mail.google.com/mail/u/0/#all/{mid}"}}})
        seen.append(mid)
    state.set("seen", seen[-1000:])
    return {"new": len(new)}