Adding user-controlled step size

app.py

@@ -5,7 +5,7 @@ import io
 import numpy as np
 
 
-def generate_random_walk(iters):
+def generate_random_walk(iters, step_size = 1):
     iters = int(iters)
     directions = ['east', 'north', 'west', 'south']
     start_point = [0, 0]
@@ -19,13 +19,13 @@ def generate_random_walk(iters):
     def step_determination():
         direction = random.choice(directions)
         if direction == 'east':
-            return [1, 0]
+            return [1*step_size, 0]
         elif direction == 'west':
-            return [-1, 0]
+            return [-1*step_size, 0]
         elif direction == 'north':
-            return [0, 1]
+            return [0, 1*step_size]
         elif direction == 'south':
-            return [0, -1]
+            return [0, -1*step_size]
     
     coordinate_list = [start_point]
     
@@ -65,6 +65,8 @@ def generate_random_walk(iters):
     image_array = np.array(fig.canvas.renderer.buffer_rgba())
     return image_array
 
+iters = gr.inputs.Number(label="How many random steps?")
+step_size = gr.inputs.Number(label="step size, defaults to 1")
 
-iface = gr.Interface(fn=generate_random_walk, inputs=gr.inputs.Number(label="How many random steps"), outputs="image", title="2-D Random Walk Plot", description="Steps along NEWS directions only, one step at a time")
+iface = gr.Interface(fn=generate_random_walk, inputs=[iters, step_size], outputs="image", title="2-D Random Walk Plot", description="Steps along NEWS directions only")
 iface.launch()