import gradio
import kaboom
from kaboom.carMakers import runCarDesignProblem

def run_kaboom(reps, steps, comms):
    
    #create a parameters object
    #parameters (p.nAgents = 33, p.nTeams = 11, p.nDims = 56) are automatically set for this problem.
    parameters = kaboom.params.Params()
    parameters.reps = reps
    parameters.pComm = comms
    parameters.steps = steps
    
    #run the simulation with the car designer objective
    team = runCarDesignProblem(parameters)
    
    #check the performance of the team
    #invert score *-1 so that higher score = better performance
    return team.getBestScore()*-1

gradio.Interface(
    fn = run_kaboom, 
    inputs = [
        gradio.Number(label="Number of teams", value=2),
        gradio.Number(label="Steps", value=300),
        gradio.Number(label="Probability of Communication", value=0.2),
    ], outputs = [gradio.Number(label="Performance")]
).launch(debug=True)