Prioritise support emails from paying customers

import base64
from email.message import EmailMessage
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import stripe
from clickruntime import on_trigger, credentials, state, ai, report_fetched

@on_trigger
def run(event):
    g = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    seen = set(state.get("seen") or [])
    new = [m for m in g.users().messages().list(userId="me", q="in:inbox newer_than:1d").execute().get("messages", []) if m["id"] not in seen]
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    stripe.api_key = credentials.secret("STRIPE_API_KEY")
    for m in new:
        msg = g.users().messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        sender = h["From"].split("<")[-1].strip("> ")
        subs = [s.status for c in stripe.Customer.search(query=f"email:'{sender}'", expand=["data.subscriptions"]).data for s in c.subscriptions.data]
        status = "paying" if "active" in subs else "trial" if "trialing" in subs else "unknown"
        g.users().messages().modify(userId="me", id=m["id"], body={"addLabelIds": [f"Label_{status.title()}"]}).execute()
        reply = EmailMessage(); reply["To"], reply["Subject"] = sender, "Re: " + h.get("Subject", "")
        reply.set_content(ai.generate(f"Draft a helpful support reply to a {status} customer.\nSubject: {h.get('Subject')}\n{msg['snippet']}", max_output_tokens=400).text)
        g.users().drafts().create(userId="me", body={"message": {"threadId": msg["threadId"], "raw": base64.urlsafe_b64encode(reply.as_bytes()).decode()}}).execute()
    state.set("seen", (list(seen) + [m["id"] for m in new])[-500:])
    return {"new": len(new)}