Save photos, receipts and voice notes from WhatsApp to Drive

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import base64, io
from googleapiclient.http import MediaIoBaseUpload
from clickruntime import on_event, credentials, http, ai

FOLDER_ID = "your-folder-id"
WA = "https://graph.facebook.com/v20.0/your-phone-number-id/messages"

def wa(to, text):
    http.post(WA, headers={"Authorization": f"Bearer {credentials.secret('WHATSAPP_TOKEN')}"},
              json={"messaging_product": "whatsapp", "to": to, "type": "text", "text": {"body": text}}).raise_for_status()

@on_event
def handle(event):
    h = {"Authorization": f"Bearer {credentials.secret('WHATSAPP_TOKEN')}"}
    msgs = [m for e in (event.json or {}).get("entry", []) for c in e["changes"] for m in c["value"].get("messages", [])]
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    for m in (m for m in msgs if m["type"] in ("image", "document", "audio")):
        media = m[m["type"]]
        info = http.get(f"https://graph.facebook.com/v20.0/{media['id']}", headers=h).json()
        data = http.get(info["url"], headers=h).content
        note = ""
        if m["type"] == "image":
            note = ai.generate(["If this is a receipt, give supplier, date and total in one line; otherwise reply NONE.",
                                base64.b64encode(data).decode()[:30000]], max_output_tokens=60).text.strip()
        name = media.get("filename") or f"{m['type']}-{m['id'][-8:]}.{info['mime_type'].split('/')[-1].split(';')[0]}"
        drive.files().create(body={"name": name, "parents": [FOLDER_ID], "description": note},
                             media_body=MediaIoBaseUpload(io.BytesIO(data), mimetype=info["mime_type"])).execute()
        wa(m["from"], f"Saved {name} to Drive." + (f" {note}" if note and note != "NONE" else ""))
    return {"saved": len(msgs)}