#!/bin/bash

# Set fixed parameter: generate 5000 data points for each case
NUM_OF_DATA=5000

echo "Starting to generate data for three different grid sizes..."

# Generate data for grid_size=4
echo "Generating data for grid_size=4..."
python -m games.tasks.calcudoko.scripts.calcudoko --num_of_data $NUM_OF_DATA --grid_size 4
echo "Data generation for grid_size=4 completed!"

# Generate data for grid_size=5
echo "Generating data for grid_size=5..."
python -m games.tasks.calcudoko.scripts.calcudoko --num_of_data $NUM_OF_DATA --grid_size 5
echo "Data generation for grid_size=5 completed!"

# Generate data for grid_size=6
echo "Generating data for grid_size=6..."
python -m games.tasks.calcudoko.scripts.calcudoko --num_of_data $NUM_OF_DATA --grid_size 6
echo "Data generation for grid_size=6 completed!"

echo "All data generation completed! Data is stored in:"
echo "- grid_size=4: ./data/grid_size_4/puzzles_grid4_n${NUM_OF_DATA}.jsonl"
echo "- grid_size=5: ./data/grid_size_5/puzzles_grid5_n${NUM_OF_DATA}.jsonl"
echo "- grid_size=6: ./data/grid_size_6/puzzles_grid6_n${NUM_OF_DATA}.jsonl"

for grid_size in 4 5 6
do
    python scripts/calcudoko_verifier.py \
        --grid_size ${grid_size} \
        --input_file ./data/grid_size_${grid_size}/puzzles_grid${grid_size}_n5000.jsonl \
        --output_file ./data/grid_size_${grid_size}/puzzles_grid${grid_size}_n5000.jsonl
done