#!/bin/bash
# Run DPA simulation on local machine (CPU/MPS, no GPU needed)
set -e

cd "$(dirname "$0")/.."

echo "=== Generating agent trajectory dataset ==="
python3 src/data/agent_trajectory.py

echo ""
echo "=== Running DPA simulation benchmark ==="
python3 src/eval/benchmark.py \
  --mode simulate \
  --hidden-size 512 \
  --num-layers 6 \
  --num-heads 8 \
  --seq-len 1024 \
  --batch-size 4 \
  --output-dir results/simulation

echo ""
echo "=== Generating figures ==="
python3 src/eval/visualize.py results/simulation/simulation_results.json
python3 src/eval/visualize.py data/agent_trajectories.json  # trajectory analysis

echo ""
echo "=== Done! Check results/ and figures/ ==="