#!/bin/bash

# Get the directory of this script
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Default configuration options (no environment variable dependencies)
NUM_PATIENTS=100
MODEL_NAME="gpt-4o"

# Parse command line arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        --num_patients)
            NUM_PATIENTS="$2"
            shift 2
            ;;
        --model_name)
            MODEL_NAME="$2"
            shift 2
            ;;
        *)
            echo "Unknown option: $1"
            echo "Usage: $0 [--num_patients NUM] [--model_name MODEL]"
            exit 1
            ;;
    esac
done

EXPERIMENT_FOLDER="experiments/history_taking/history_taking_${MODEL_NAME}"

echo "Running history taking simulator experiment..."
echo "Number of patients: ${NUM_PATIENTS}"
echo "Model name: ${MODEL_NAME}"
echo "Experiment folder: ${EXPERIMENT_FOLDER}"

cd "${SCRIPT_DIR}" && python experiment.py --experiment_type history_taking --experiment_folder "${EXPERIMENT_FOLDER}" --num_patients "${NUM_PATIENTS}" 