import gradio as gr
from huggingface_hub import HfApi
from substra_launcher import launch_substra_space

api = HfApi()

gr.Interface(
    title="Susbtra Demo - Orchestrator",
    fn=lambda *args, **kwargs: launch_substra_space(api, *args, **kwargs),
    inputs=[
        gr.Slider(label="num_hospitals", value=3, minimum=2, maximum=5, step=1),
        gr.Text(label="space_name")
    ],
    outputs="text",
    examples=[[2, "substra-test"]]
).launch()