import taipy as tp
import taipy.gui.builder as tgb
from taipy import Config, Gui, Scenario
from taipy.gui import State


def square(x: int) -> int:
    return x**2


x_cfg = Config.configure_data_node(id="x", default_data=5)
y_cfg = Config.configure_data_node(id="y")
square_task_cfg = Config.configure_task(id="square", function=square, input=x_cfg, output=y_cfg)
scenario_cfg = Config.configure_scenario(id="scenario", task_configs=[square_task_cfg])

selected_scenario = None

def on_create_scenario(state: State, id: str, payload: dict):
    config = payload["config"]
    date = payload["date"]
    label = payload["label"]
    properties = payload["properties"]

    label += f" {date.strftime('%Y-%m-%d %H:%M:%S')}"
    sc: Scenario = tp.create_scenario(config, name=label, creation_date=date)
    sc.properties.update(properties)
    sc.properties["another_key"] = "another_value"
    return sc


with tgb.Page() as page:
    tgb.toggle(theme=True)
    with tgb.part(class_name="container"):
        tgb.text("# Select scenario:", mode="md")
        with tgb.layout(columns="1 1"):
            tgb.scenario_selector("{selected_scenario}", on_creation=on_create_scenario, filter=["another_key", "Label"])
            tgb.scenario_selector("{selected_scenario}", on_creation=on_create_scenario)
        tgb.scenario("{selected_scenario}")

if __name__ == "__main__":
    Config.configure_core()
    tp.Orchestrator().run()
    Gui(page=page).run(title="Filter by Custom Property", run_browser=False)