DebasishDhal99's picture
Adding multi agent random walk
5e24ac5
raw
history blame
983 Bytes
import matplotlib.pyplot as plt
import random
import gradio as gr
import numpy as np
import pandas as pd
from single_agent_2D import generate_random_walk
from multi_agent_2D import multi_agent_walk
iters = gr.Number(value=1e6,label="How many random steps?")
step_size = gr.Number(value=1,label="Step size")
agent_count = gr.Number(value=3, labale = "Number of agents")
random_seed = gr.Number(value=42,label="Random seed. Delete it to go full random mode, keep it for reproducibility")
iface1 = gr.Interface(fn=generate_random_walk, inputs=[iters, step_size, random_seed], outputs=["image","file"], title="2-D Random Walk", description="Uniform steps along NEWS directions only")
iface2 = gr.Interface(fn=multi_agent_walk, inputs=[iters,iters, step_size, random_seed], output=["image","file"], title="Multi-Agent 2D Random Walk")
combinedinterface = gr.TabbedInterface([iface1,iface2],['Single Particle Random Walk', 'Multi-Particle Random Walk'])
combinedinterface.launch()