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()