Adding random seed for reproducibility

app.py

@@ -1,11 +1,13 @@
 import matplotlib.pyplot as plt
 import random
 import gradio as gr
-import io
 import numpy as np
 
+# seed = random.randint(1, 1000)
+# random.seed(seed)
 
-def generate_random_walk(iters, step_size = 1):
+def generate_random_walk(iters, step_size = 1, random_seed=42):
+    random.seed(random_seed)
     iters = int(iters)
     directions = ['east', 'north', 'west', 'south']
     start_point = [0, 0]
@@ -58,7 +60,7 @@ def generate_random_walk(iters, step_size = 1):
     info_text = 'Start point=' + str(start_point) + '\n'  +'End point=' + str([x[-1],y[-1]]) + '\n' +'Displacement =' + str(distance_from_start([x[-1], y[-1]], start_point)) + '\n' +'Max x = ' + str(max(x)) + '\n' + 'Min x = ' + str(min(x)) + '\n' + 'Max y = ' + str(max(y)) + '\n' + 'Min y = ' + str(min(y)) 
     ax.legend([info_text], loc='best', handlelength=0, handletextpad=0, fancybox=True, fontsize=8)
     
-    plt.title('2D Random Walk, steps = ' + str(iters)+', step size = '+ str(step_size))
+    plt.title('2D Random Walk, steps = ' + str(iters)+', step size = '+ str(step_size)+ 'seed = '+str(random_seed))
     plt.grid()
     
     fig.canvas.draw()
@@ -67,6 +69,7 @@ def generate_random_walk(iters, step_size = 1):
 
 iters = gr.inputs.Number(label="How many random steps?", default=1e5)
 step_size = gr.inputs.Number(label="Step size", default=1)
+random_seed = gr.inputs.Number(label="Random seed", default=42)
 
-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 = gr.Interface(fn=generate_random_walk, inputs=[iters, step_size, random_seed], outputs="image", title="2-D Random Walk Plot", description="Steps along NEWS directions only")
 iface.launch()