run_game.jl

include("src/game.jl")
include("src/simulation.jl")
include("src/utils.jl")
using DataFrames
using CSV
using Serialization
using ProgressBars


# TODO set and variate text generation length? --> kinda covered with dialog steps. should be irrelevant
# TODO Number of teams and players does not affect bench data --> should also cut it?
# TODO Add sanity check (at least one action taken and add invalid flag?) --> do in data analysis

# Everything not set will result in usage of default values 
conf = init_sim_conf(
    # model="gpt-4-1106-preview",
    model="gpt-3.5-turbo-16k",
    # secret_key=get(ENV, "OPENAI_API_KEY", ""),
    # wargame_dir="wargame/",
    # output_dir="results/",
    # out_csv_file="",
    use_dummygpt=true,
    use_bench_players=true,
    no_dialog=false,
    no_chiefs=false,
    boostrap_players=true,
    pacificsm=false,
    sociopaths=false,
    more_disagreement=false,
    verbose=false,
    # save_results_to_csv=true,
    run_test_game=true,
    n_teams=10,
    n_players=6,
    n_dialog_steps=3,
)

function run_simulation(config::SimulationConfig)

    # Prepare the connection to the GPT model
    # TODO? turn into assert and combine with && !config.use_dummygpt
    if config.secret_key == ""
        @warn "OPENAI_API_KEY not set in ENV"
    end
    chat_setup = ChatSetup(config.secret_key, config.model, config.use_dummygpt)

    # Setup the results dataframe and game dir
    res = results_df(config)

    # Setup outpuf file names
    if config.out_csv_file == ""
        ending = create_file_ending(config.output_dir)
        data_filename = config.output_dir * "data" * ending * ".csv"
    else
        data_filename = config.output_dir * config.out_csv_file
    end 

    if conf.use_bench_players
        test_data = deserialize("wargame/test_data.jls")
        treatments = test_data[1]
        teams = test_data[2]
    else
        # Generate Treatments
        treatments = gen_all_treatments(config)

        # Generate teams
        teams = gen_teams(config)
    end
    
    if conf.pacificsm
        for (team_ind, team) in enumerate(teams)
            teams[team_ind] = [changeXP_player(p, "Strict pacifist") for p in team]
        end
    elseif conf.sociopaths
        for (team_ind, team) in enumerate(teams)
            teams[team_ind] = [changeXP_player(p, "Aggressive sociopath") for p in team]
        end
    end

    # Run a test game or all treatments * teams
    if config.run_test_game
        result = run_game(config, treatments[1], teams[1], chat_setup)
        push!(res, result)

        # Write the results DataFrame to a CSV file
        if config.save_results_to_csv
            CSV.write(data_filename, res)
        end
    else
        # Run the games
        for (j, team) in ProgressBar(enumerate(teams))
            println("Team: ", j)
            for (i, game) in ProgressBar(enumerate(treatments))
                println("Running treatment: ", i)
                try 
                    result = run_game(config, game, team, chat_setup)
                    push!(res, result)

                    # Write the results DataFrame to a CSV file
                    if config.save_results_to_csv
                        CSV.write(data_filename, res)
                    end
                catch e
                    println(e)
                    sleep(10)
                end
            end
        end
    end

    return res
end

# run_simulation(SimulationConfig())
# run_simulation(SimulationConfig(run_test_game=true, verbose=true, use_dummygpt=true))
run_simulation(conf)