Turn a website address into a company profile in Supabase and Drive

from bs4 import BeautifulSoup
from googleapiclient.http import MediaInMemoryUpload
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_event, credentials, http, ai

SUPABASE_URL, FOLDER_ID = "https://your-project.supabase.co", "your-profiles-folder-id"

@on_event
def handle(event):
    url = event.event["url"]
    text = BeautifulSoup(http.get(url, timeout=20).text, "html.parser").get_text(" ", strip=True)[:15000]
    schema = {"type": "object", "properties": {k: {"type": "string"} for k in ["name", "mission", "services", "contacts", "keywords"]}}
    p = ai.generate(f"Extract the company name, mission, services, contacts and keywords from this website:\n{text}",
                    json_schema=schema, max_output_tokens=600).json
    key = credentials.secret("SUPABASE_SERVICE_KEY")
    http.post(f"{SUPABASE_URL}/rest/v1/companies", json={"website": url, **p},
              headers={"apikey": key, "Authorization": f"Bearer {key}", "Prefer": "resolution=merge-duplicates"})
    doc = "\n\n".join(f"{k.title()}: {v}" for k, v in p.items())
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    drive.files().create(body={"name": f"{p.get('name', url)} profile", "parents": [FOLDER_ID],
                               "mimeType": "application/vnd.google-apps.document"},
                         media_body=MediaInMemoryUpload(f"{url}\n\n{doc}".encode(), mimetype="text/plain")).execute()
    return {"company": p.get("name")}