Turn enquiry emails into Salesforce leads

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

API = "https://yourcompany.my.salesforce.com/services/data/v61.0"
SCHEMA = {"type": "object", "properties": {k: {"type": "string"} for k in
          ["first_name", "last_name", "email", "company", "phone", "need"]}, "required": ["last_name", "email", "company"]}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).users()
    labels = {l["name"]: l["id"] for l in gmail.labels().list(userId="me").execute()["labels"]}
    msgs = gmail.messages().list(userId="me", labelIds=[labels["Leads"]], maxResults=20).execute().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"summary": "No new lead emails"}
    headers, created = {"Authorization": f"Bearer {credentials.connection('salesforce', 'work')}"}, 0
    for m in msgs:
        full = gmail.messages().get(userId="me", id=m["id"], format="full").execute()
        body = next((base64.urlsafe_b64decode(p["body"]["data"]).decode(errors="ignore") for p in full["payload"].get("parts") or [full["payload"]]
                     if p.get("mimeType") == "text/plain" and p["body"].get("data")), full["snippet"])
        d = ai.generate("Pull the sender's contact details and what they need from this enquiry email:\n" + body[:3000], json_schema=SCHEMA, max_output_tokens=200).json
        email = d["email"].replace("'", "")
        if not http.get(f"{API}/query", headers=headers, params={"q": f"SELECT Id FROM Lead WHERE Email = '{email}' LIMIT 1"}).json()["records"]:
            http.post(f"{API}/sobjects/Lead", headers=headers, json={"FirstName": d.get("first_name"), "LastName": d["last_name"], "Email": email,
                      "Company": d["company"] or "Unknown", "Phone": d.get("phone"), "LeadSource": "Email", "Description": d.get("need")}).raise_for_status()
            created += 1
        gmail.messages().modify(userId="me", id=m["id"], body={"removeLabelIds": [labels["Leads"]], "addLabelIds": [labels["Leads/Done"]]}).execute()
    return {"emails": len(msgs), "created": created}