Turn companies in a sheet into research briefs in Google Docs

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

SHEET_ID = "your-spreadsheet-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    sheets = build("sheets", "v4", credentials=creds, cache_discovery=False).spreadsheets()
    rows = sheets.values().get(spreadsheetId=SHEET_ID, range="Prospects!A2:C").execute().get("values", [])
    todo = [(i + 2, r) for i, r in enumerate(rows) if len(r) >= 2 and (len(r) < 3 or not r[2])]
    report_fetched(len(todo))
    if not todo:
        return {"summary": "Nothing new"}
    docs = build("docs", "v1", credentials=creds, cache_discovery=False).documents()
    for row, (name, site, *_) in todo[:5]:
        text = BeautifulSoup(http.get(site if site.startswith("https") else f"https://{site}", timeout=15).text, "html.parser").get_text(" ", strip=True)[:4000]
        brief = ai.generate(f"Company: {name}\nWebsite text:\n{text}", system="Write a sales research brief with sections: Overview, Customers, Products, Talking points.", max_output_tokens=900).text
        doc = docs.create(body={"title": f"Brief: {name}"}).execute()
        docs.batchUpdate(documentId=doc["documentId"], body={"requests": [{"insertText": {"location": {"index": 1}, "text": brief}}]}).execute()
        sheets.values().update(spreadsheetId=SHEET_ID, range=f"Prospects!C{row}", valueInputOption="RAW",
                               body={"values": [[f"https://docs.google.com/document/d/{doc['documentId']}"]]}).execute()
    return {"briefs": min(len(todo), 5)}