Save a YouTube video's summary and social posts to Drive

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import re
from googleapiclient.http import MediaInMemoryUpload
from clickruntime import on_event, credentials, ai

def api(name, version):
    return build(name, version, credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)

FOLDER_ID = "video-notes-folder-id"

@on_event
def handle(event):
    url = (event.json or {})["url"]
    vid = re.search(r"(?:v=|youtu\.be/)([\w-]{11})", url).group(1)
    yt = api("youtube", "v3")
    s = yt.videos().list(part="snippet", id=vid).execute()["items"][0]["snippet"]
    text = s["description"]
    tracks = yt.captions().list(part="snippet", videoId=vid).execute().get("items", [])
    if tracks:
        text = yt.captions().download(id=tracks[0]["id"], tfmt="srt").execute().decode()[:24000]
    notes = ai.generate(f"Title: {s['title']}\n\n{text}", system="Write a one-paragraph summary, 5 key points, "
                        "then three social posts (LinkedIn, X, newsletter).", max_output_tokens=1200).text
    doc = api("drive", "v3").files().create(body={"name": f"Notes: {s['title']}", "parents": [FOLDER_ID],
          "mimeType": "application/vnd.google-apps.document"}, media_body=MediaInMemoryUpload(f"{url}\n\n{notes}".encode(), mimetype="text/plain"),
          fields="id,webViewLink").execute()
    return {"doc": doc["webViewLink"]}