| 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_single = gr.Number(value=1e6,label="How many random steps?") | |
| iters_multi = gr.Number(value=1e6,label="How many random steps?") | |
| step_size_single = gr.Number(value=1,label="Step size") | |
| step_size_multi = gr.Number(value=1,label="Step size") | |
| agent_count = gr.Number(value=3, label = "Number of agents") | |
| random_seed_single = gr.Number(value=42,label="Random seed. Delete it to go full random mode, keep it for reproducibility") | |
| random_seed_multi = 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_single, step_size_single, random_seed_single], outputs=["image","file"], title="2-D Random Walk", description="Uniform steps along NEWS directions only") | |
| iface2 = gr.Interface(fn=multi_agent_walk, inputs=[iters_multi,iters_multi, step_size_multi, random_seed_multi], outputs=["image","file"], title="Multi-Agent 2D Random Walk") | |
| combinedinterface = gr.TabbedInterface([iface1,iface2],['Single Particle Random Walk', 'Multi-Particle Random Walk']) | |
| combinedinterface.launch() | |