Sort your inbox with AI and get urgent emails on Telegram

import time
from clickruntime import on_trigger, credentials, http, state, ai, report_fetched

G = "https://gmail.googleapis.com/gmail/v1/users/me"
LABELS = {"customer": "Label_1", "supplier": "Label_2", "admin": "Label_3", "newsletter": "Label_4", "promo": "Label_5"}
SCHEMA = {"type": "object", "properties": {"category": {"type": "string", "enum": list(LABELS)},
          "urgent": {"type": "boolean"}, "summary": {"type": "string"}}, "required": ["category", "urgent", "summary"]}

@on_trigger
def run(event):
    h = {"Authorization": f"Bearer {credentials.connection('google', 'work')}"}
    since = state.get("since") or 0
    msgs = http.get(f"{G}/messages", headers=h, params={"q": f"in:inbox after:{since}"}).json().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"new": 0}
    bot, chat = credentials.secret("TELEGRAM_BOT_TOKEN"), credentials.secret("TELEGRAM_CHAT_ID")
    for m in msgs:
        msg = http.get(f"{G}/messages/{m['id']}", headers=h, params={"format": "metadata", "metadataHeaders": ["From", "Subject"]}).json()
        hdr = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        r = ai.generate(f"From: {hdr.get('From')}\nSubject: {hdr.get('Subject')}\n{msg['snippet']}\nUrgent means a cancelled delivery, complaint or deadline today.",
                        json_schema=SCHEMA, max_output_tokens=80).json
        remove = ["INBOX"] if r["category"] in ("newsletter", "promo") else []
        http.post(f"{G}/messages/{m['id']}/modify", headers=h, json={"addLabelIds": [LABELS[r["category"]]], "removeLabelIds": remove})
        if r["urgent"]:
            http.post(f"https://api.telegram.org/bot{bot}/sendMessage", json={"chat_id": chat, "text": f"{hdr.get('From')}: {r['summary']}"})
    state.set("since", int(time.time()))
    return {"sorted": len(msgs)}