Track sponsorship offers from Gmail in Notion

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

DEALS_DB = "your-notion-database-id"
SCHEMA = {"type": "object", "properties": {"is_offer": {"type": "boolean"}, "brand": {"type": "string"}, "budget": {"type": "string"}, "deadline": {"type": "string"}}, "required": ["is_offer"]}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users().messages()
    seen = set(state.get("seen") or [])
    new = [m["id"] for m in gmail.list(userId="me", q="newer_than:1d -category:promotions", maxResults=50).execute().get("messages", []) if m["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"summary": "Nothing new"}
    notion = {"Authorization": f"Bearer {credentials.connection('notion', 'work')}", "Notion-Version": "2022-06-28"}
    added = 0
    for mid in new:
        msg = gmail.get(userId="me", id=mid, format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        info = ai.generate(f"From: {h.get('From')}\nSubject: {h.get('Subject')}\n{msg['snippet']}", system="Is this a sponsorship or brand deal offer? Extract brand, budget, deadline.", json_schema=SCHEMA, max_output_tokens=120).json
        if info.get("is_offer"):
            http.post("https://api.notion.com/v1/pages", headers=notion, json={"parent": {"database_id": DEALS_DB}, "properties": {
                "Name": {"title": [{"text": {"content": info.get("brand") or h.get("Subject", "Offer")}}]},
                "Budget": {"rich_text": [{"text": {"content": info.get("budget", "")}}]}, "Deadline": {"rich_text": [{"text": {"content": info.get("deadline", "")}}]},
                "From": {"rich_text": [{"text": {"content": h.get("From", "")}}]}}}).raise_for_status()
            added += 1
    state.set("seen", (list(seen) + new)[-500:])
    return {"checked": len(new), "added": added}