Answer WhatsApp questions from your Google Drive documents

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import time
from clickruntime import on_event, credentials, http, state, 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()
def incoming(event):
    changes = [c for e in (event.json or {}).get("entry", []) for c in e.get("changes", [])]
    return [(m["from"], m["text"]["body"]) for c in changes for m in c["value"].get("messages", []) if m.get("type") == "text"]
drive = lambda: build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)

def docs_text():
    cache = state.get("docs") or {}
    if time.time() - cache.get("at", 0) > 86400:
        files = drive().files().list(q=f"'{FOLDER_ID}' in parents and mimeType='application/vnd.google-apps.document'").execute()["files"]
        text = "\n\n".join(drive().files().export(fileId=f["id"], mimeType="text/plain").execute().decode()[:4000] for f in files)
        cache = {"at": time.time(), "text": text[:20000]}
        state.set("docs", cache)
    return cache["text"]

@on_event
def handle(event):
    for phone, text in incoming(event):
        answer = ai.generate(text[:1000], max_output_tokens=200,
                             system=f"Answer briefly from these documents only. If not covered, say the team will follow up.\n{docs_text()}").text
        wa(phone, answer.strip())
    return {"ok": True}