import gradio as gr

from utils import get_best_path

# ------------- Constants
TITLE = "Travel Planner" # TODO: Change title as we need

# ------------- Main App
with gr.Blocks(
    title=TITLE,
    theme=gr.themes.Default(primary_hue=gr.themes.colors.red).set(slider_color="#ff0000"),
    css=".travel-block {max-width: 50%; margin: 0 auto}",
) as demo:

    with gr.Column(elem_classes="travel-block") as left:
        # Input
        with gr.Row() as header:
                gr.Markdown(
                    """
                    ## Search For You Next Trip
                    """,
                )

        A = gr.Textbox(label="Start", placeholder="Lausanne")
        B = gr.Textbox(label="Destination", placeholder="Basel")

        with gr.Row() as metainfo:
            limit = gr.Slider(minimum=1, maximum=5, step=1, label="Limit", interactive=True, value=3)
            days = gr.Textbox(label="📅 Day", placeholder="YYYY-MM-DD")
            time = gr.Textbox(label="🕒 Time", placeholder="hh:mm")
            sustainable = gr.Checkbox(label="🌎 Sustainable", value=False)
            outage = gr.Checkbox(label="🚧 Outage", value=False)

        submit = gr.Button(value="Search", variant="primary")

        # Output
        with gr.Column(visible=False) as output:
            route = gr.Markdown()

        def search(A, B, limit, day, time, sustainable, outage):
            """Search for the best route from A to B"""

            md = get_best_path(A, B, day, time, limit, outage, sustainable)

            return {
                output: gr.Column(visible=True),
                route: md
            }

        submit.click(
            search,
            inputs=[A, B, limit, days, time, sustainable, outage],
            outputs=[route, output]
        )

demo.launch()