File email attachments in Drive and get a summary on Telegram

import base64
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from googleapiclient.http import MediaInMemoryUpload
from clickruntime import on_trigger, credentials, http, ai, state, report_fetched
FOLDER_ID, CHAT_ID = "your-drive-folder-id", "your-telegram-chat-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    gmail = build("gmail", "v1", credentials=creds, cache_discovery=False)
    seen = set(state.get("seen") or [])
    msgs = [m for m in gmail.users().messages().list(userId="me", q="has:attachment (filename:pdf OR filename:jpg OR filename:png) newer_than:1d")
            .execute().get("messages", []) if m["id"] not in seen]
    report_fetched(len(msgs))
    if not msgs:
        return {"files": 0}
    drive, token = build("drive", "v3", credentials=creds, cache_discovery=False), credentials.secret("TELEGRAM_BOT_TOKEN")
    for m in msgs:
        msg = gmail.users().messages().get(userId="me", id=m["id"]).execute()
        for p in [p for p in msg["payload"].get("parts", []) if p.get("filename")]:
            data = gmail.users().messages().attachments().get(userId="me", messageId=m["id"], id=p["body"]["attachmentId"]).execute()["data"]
            f = drive.files().create(body={"name": p["filename"], "parents": [FOLDER_ID]}, fields="id,webViewLink",
                                     media_body=MediaInMemoryUpload(base64.urlsafe_b64decode(data), mimetype=p["mimeType"])).execute()
            summary = ai.generate(f"Summarise in two lines what this attachment is, given the email:\n{msg['snippet']}\nFile: {p['filename']}",
                                  max_output_tokens=120).text
            http.post(f"https://api.telegram.org/bot{token}/sendMessage", json={"chat_id": CHAT_ID, "text": f"📎 {p['filename']}\n{summary}\n{f['webViewLink']}"})
        seen.add(m["id"])
    state.set("seen", list(seen)[-1000:])
    return {"emails": len(msgs)}