Log expenses and catch budget overruns

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from datetime import date
from clickruntime import on_event, credentials, state, ai, notify

SHEET_ID = "your-spreadsheet-id"
CATEGORIES = ["Software", "Travel", "Meals", "Office", "Other"]

@on_event
def handle(event):
    exp = event.json or {}
    sheets = build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets()
    cat = ai.generate(f"Pick one category from {CATEGORIES} for this expense, reply with it only: {exp['description']}",
                      max_output_tokens=10).text.strip()
    cat = cat if cat in CATEGORIES else "Other"
    today = date.today().isoformat()
    sheets.values().append(spreadsheetId=SHEET_ID, range="Expenses!A:D", valueInputOption="USER_ENTERED",
                           body={"values": [[today, exp["description"], float(exp["amount"]), cat]]}).execute()
    rows = sheets.values().get(spreadsheetId=SHEET_ID, range="Expenses!A2:D").execute().get("values", [])
    spent = sum(float(r[2]) for r in rows if len(r) > 3 and r[3] == cat and r[0][:7] == today[:7])
    limits = dict(sheets.values().get(spreadsheetId=SHEET_ID, range="Budget!A2:B").execute().get("values", []))
    limit = float(limits.get(cat, "inf"))
    alert_key = f"alerted:{today[:7]}:{cat}"
    if spent > limit and not state.get(alert_key):
        notify.email_me(f"{cat} is over budget", f"{cat}: {spent:,.2f} spent this month against a limit of {limit:,.2f}.")
        state.set(alert_key, True)
    return {"category": cat, "spent": spent}