| 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=1_000_000,label="How many random steps?") | |
| iters_multi = gr.Number(value=1_000_000,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=[agent_count,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() | |