Data Analysis in a Sandbox

import { Sandbox } from "@declaw/sdk";

const SALES_CSV = `date,product,region,quantity,unit_price
2024-01-05,Widget A,North,120,9.99
2024-01-05,Widget B,South,85,14.50
...
`;

const ANALYSIS_SCRIPT = `import csv, json
from collections import defaultdict

rows = []
with open("/home/user/data/sales.csv") as f:
    reader = csv.DictReader(f)
    for row in reader:
        row["quantity"] = int(row["quantity"])
        row["unit_price"] = float(row["unit_price"])
        row["revenue"] = row["quantity"] * row["unit_price"]
        rows.append(row)

total_revenue = sum(r["revenue"] for r in rows)
rev_by_product = defaultdict(float)
for r in rows:
    rev_by_product[r["product"]] += r["revenue"]

results = {
    "total_revenue": round(total_revenue, 2),
    "top_product": max(rev_by_product, key=rev_by_product.get),
    "revenue_by_product": {k: round(v, 2) for k, v in sorted(rev_by_product.items())},
}

with open("/home/user/data/results.json", "w") as f:
    json.dump(results, f, indent=2)

print("Analysis complete.")
`;

const sbx = await Sandbox.create({ template: "python", timeout: 300 });
try {
    // Upload dataset and script
    await sbx.files.write("/home/user/data/sales.csv", SALES_CSV);
    await sbx.files.write("/home/user/data/analyze.py", ANALYSIS_SCRIPT);

    // Run the analysis
    const result = await sbx.commands.run("python3 /home/user/data/analyze.py");
    console.log(result.stdout);

    if (result.exitCode !== 0) {
        console.error(result.stderr);
        return;
    }

    // Read results back
    const resultsJson = await sbx.files.read("/home/user/data/results.json");
    const data = JSON.parse(resultsJson);

    console.log(`Total revenue:  $${data.total_revenue.toLocaleString()}`);
    console.log(`Top product:    ${data.top_product}`);
    for (const [product, rev] of Object.entries(data.revenue_by_product)) {
        console.log(`  ${product}: $${(rev as number).toLocaleString()}`);
    }
} finally {
    await sbx.kill();
}