Get urgent emails on WhatsApp

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

MY_PHONE = "447700900123"
URGENT_LABEL = "Label_123"
SCHEMA = {"type": "object", "properties": {"urgent": {"type": "boolean"}, "summary": {"type": "string"}}, "required": ["urgent", "summary"]}
WA = "https://graph.facebook.com/v20.0/your-phone-number-id/messages"

def wa(to, text):
    http.post(WA, headers={"Authorization": f"Bearer {credentials.secret('WHATSAPP_TOKEN')}"},
              json={"messaging_product": "whatsapp", "to": to, "type": "text", "text": {"body": text}}).raise_for_status()
gmail = lambda: build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users()

@on_trigger
def run(event):
    seen = state.get("seen") or []
    ids = [m["id"] for m in gmail().messages().list(userId="me", q="in:inbox newer_than:1d", maxResults=30).execute().get("messages", [])]
    new = [i for i in ids if i not in seen]
    report_fetched(len(new))
    if not new:
        return {"summary": "Nothing new"}
    for i in new:
        m = gmail().messages().get(userId="me", id=i, format="metadata", metadataHeaders=["From", "Subject"]).execute()
        head = {h["name"]: h["value"] for h in m["payload"]["headers"]}
        r = ai.generate(f"From: {head.get('From')}\nSubject: {head.get('Subject')}\n{m['snippet']}", json_schema=SCHEMA, max_output_tokens=60,
                        system="Urgent = a client problem, a payment issue, or something needed today.").json
        if r["urgent"]:
            wa(MY_PHONE, f"📧 {head.get('From')}: {r['summary']}")
            gmail().messages().modify(userId="me", id=i, body={"addLabelIds": [URGENT_LABEL]}).execute()
    state.set("seen", (seen + new)[-300:])
    return {"checked": len(new)}