Draft follow-ups for stalled deals and new leads

import base64
from datetime import date, timedelta
from email.message import EmailMessage
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, http, ai, state

ZOHO = "https://www.zohoapis.com/crm/v6"

@on_trigger
def run(event):
    zh = {"Authorization": f"Zoho-oauthtoken {credentials.connection('zoho', 'work')}"}
    cutoff = (date.today() - timedelta(days=10)).isoformat()
    deals = http.get(f"{ZOHO}/Deals/search", headers=zh,
                     params={"criteria": f"((Modified_Time:less_than:{cutoff}T00:00:00+00:00)and(Stage:not_equal:Closed Won))"}).json().get("data", [])
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    drafted = state.get("drafted") or {}
    for d in deals:
        if drafted.get(d["id"], "") > cutoff or not d.get("Email"):
            continue
        body = ai.generate(f"Short friendly follow-up for deal '{d['Deal_Name']}' at stage {d['Stage']}.", max_output_tokens=200).text
        msg = EmailMessage()
        msg["To"], msg["Subject"] = d["Email"], f"Checking in: {d['Deal_Name']}"
        msg.set_content(body)
        gmail.users().drafts().create(userId="me", body={"message": {"raw": base64.urlsafe_b64encode(msg.as_bytes()).decode()}}).execute()
        http.post(f"{ZOHO}/Notes", headers=zh, json={"data": [{"Parent_Id": {"id": d["id"]}, "$se_module": "Deals",
                                                                "Note_Content": "Follow-up drafted in Gmail."}]})
        drafted[d["id"]] = date.today().isoformat()
    state.set("drafted", drafted)
    return {"drafted": len(drafted)}