DebasishDhal99's picture
remove compact theme
5c25bbe
raw
history blame
1.55 kB
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=[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()