import gradio as gr
from ui import __version__
from ui.tabs.simple_example_tab import SimpleExampleTab
from ui.tabs.single_pop_tab import SinglePopTab
from ui.tabs.with_agent_tab import WithAgentTab

with gr.Blocks(theme=gr.themes.Glass()) as interface:
    gr.Markdown("**Social distancing sim**")
    gr.Markdown(
        """Graph based disease spread simulator. Uses https://github.com/garethjns/social-distancing-sim -
        see [README](https://github.com/garethjns/social-distancing-sim/blob/master/README.MD)
        for details.

        Individuals in the population are represented as graph nodes, with edges modelling connections
        allowing the possibility of disease transmission. between individuals.
        """
    )
    gr.Markdown(f"App version: {__version__} using `social-distancing-sim==0.11.2`")
    SimpleExampleTab().build()
    SinglePopTab().build()
    WithAgentTab().build()

if __name__ == "__main__":
    interface.launch()