#!/bin/bash

# Set working directory to project root
cd "$(dirname "$0")/../../../" || exit

# Set Python environment variable
export PYTHONUNBUFFERED=1

# Default parameters
NUM_OF_DATA=200
# NUM_OF_DATA=10  # For debugging
MAX_ATTEMPTS=10000
DIFFICULTY=5
STATEMENT_TYPE=0

# Create data directory
mkdir -p ./games/tasks/web_of_lies/data

for NUM_PERSON in 10 12 14 16 18; do
  # Output current parameters
  echo "Generating Web of Lies game data with the following parameters:"
  echo "Number of problems: $NUM_OF_DATA"
  echo "Maximum attempts: $MAX_ATTEMPTS"
  echo "Number of characters: $NUM_PERSON"
  echo "Difficulty level: $DIFFICULTY"
  echo "Statement type: $STATEMENT_TYPE"

  # Run game generation script
  python -m games.tasks.web_of_lies.scripts.web_of_lies \
    --num_of_data "$NUM_OF_DATA" \
    --max_attempts "$MAX_ATTEMPTS" \
    --num_person "$NUM_PERSON" \
    --difficulty "$DIFFICULTY" \
    --statement_type "$STATEMENT_TYPE"

  echo "Game data generation completed!"

done