Clear out old promotional emails every night

from collections import Counter
from googleapiclient.discovery import build
from google.oauth2.credentials import Credentials
from clickruntime import on_trigger, credentials, ai, http

KEEP = ["bank.example", "acme.example"]
CHAT_ID = "your-telegram-chat-id"

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users()
    ids = gmail.messages().list(userId="me", q="category:promotions older_than:30d", maxResults=100).execute().get("messages", [])
    rows = []
    for m in ids:
        h = {x["name"]: x["value"] for x in gmail.messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Subject"]).execute()["payload"]["headers"]}
        if not any(k in h.get("From", "") for k in KEEP):
            rows.append((m["id"], h.get("From", ""), h.get("Subject", "")))
    trashed = Counter()
    for i in range(0, len(rows), 25):
        batch = rows[i:i + 25]
        verdict = ai.generate("For each line answer junk or keep (keep receipts, invoices, account notices):\n" +
                              "\n".join(f"{n}. {f} | {s}" for n, (_, f, s) in enumerate(batch, 1)), max_output_tokens=200).text.lower().splitlines()
        for (mid, sender, _), v in zip(batch, verdict):
            if "junk" in v:
                gmail.messages().trash(userId="me", id=mid).execute()
                trashed[sender.split("<")[0].strip()] += 1
    text = f"Cleared {sum(trashed.values())} old promotions. Top: " + ", ".join(f"{s} ({n})" for s, n in trashed.most_common(5))
    http.post(f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}/sendMessage", json={"chat_id": CHAT_ID, "text": text})
    return {"trashed": sum(trashed.values())}