Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio
 import kaboom
 from kaboom.carMakers import runCarDesignProblem
 
-def run_kaboom(reps):
+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)
@@ -17,5 +19,10 @@ def run_kaboom(reps):
     return team.getBestScore()*-1
 
 gradio.Interface(
-    fn = run_kaboom, inputs = [gradio.Number(label="Number of teams", value=1)], outputs = [gradio.Number(label="Performance")]
+    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)