Schedule study sessions around your calendar until the exam

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from datetime import datetime, timezone
from clickruntime import on_trigger, credentials, ai

SHEET_ID = "your-spreadsheet-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    sheet = build("sheets", "v4", credentials=creds, cache_discovery=False).spreadsheets().values()
    rows = sheet.get(spreadsheetId=SHEET_ID, range="Syllabus!A2:D").execute().get("values", [])
    todo = [(i, r) for i, r in enumerate(rows, start=2) if len(r) < 4 or r[3] != "Scheduled"]
    if not todo:
        return {"scheduled": 0}
    exam = rows[0][1]
    cal = build("calendar", "v3", credentials=creds, cache_discovery=False)
    busy = cal.freebusy().query(body={"timeMin": datetime.now(timezone.utc).isoformat(), "timeMax": f"{exam}T00:00:00Z",
                                      "items": [{"id": "primary"}]}).execute()["calendars"]["primary"]["busy"]
    schema = {"type": "object", "properties": {"sessions": {"type": "array", "items": {"type": "object", "properties": {
        "topic": {"type": "string"}, "start": {"type": "string"}, "end": {"type": "string"}}}}}}
    plan = ai.generate([f"Plan 1-2 hour study sessions between 08:00 and 21:00 before {exam}, avoiding these busy times, "
                        "harder topics get more time. Use ISO times.", str(busy), str([r for _, r in todo])],
                       json_schema=schema, max_output_tokens=2000).json["sessions"]
    for s in plan:
        cal.events().insert(calendarId="primary", body={"summary": f"Study: {s['topic']}",
                            "start": {"dateTime": s["start"]}, "end": {"dateTime": s["end"]}}).execute()
    for i, _ in todo:
        sheet.update(spreadsheetId=SHEET_ID, range=f"Syllabus!D{i}", valueInputOption="RAW", body={"values": [["Scheduled"]]}).execute()
    return {"scheduled": len(plan)}