import gradio as gr

import actions as a
from components import all_tasks, Tasks


with gr.Blocks() as demo:
    # Initial layout
    gr.Markdown(
        """
    # Toolkit
    Assemble tasks to build an E2E application.
    <br>There are 2 types of tasks.
    <br>
    <br>**AI Task**: Ask ChatGPT to do something for you. Eg, summarize a text.
    <br>**Code Task**: ChatGPT will create a python function that will be executed on the fly. Eg, get the text from an url.
    <br>
    <br>Max 10 tasks allowed (for now).
    """
    )
    for t in all_tasks.values():
        t.render()
    with gr.Row():
        add_task_btn = gr.Button("Add task")
        remove_task_btn = gr.Button("Remove task")
    error_message = gr.HighlightedText(value=None, visible=False)
    execute_btn = gr.Button("Execute")

    # Edit layout
    add_task_btn.click(
        a.add_task,
        inputs=Tasks.visibilities(),
        outputs=Tasks.gr_components() + Tasks.visibilities(),
    )
    remove_task_btn.click(
        a.remove_task,
        inputs=Tasks.visibilities(),
        outputs=Tasks.gr_components() + Tasks.visibilities(),
    )

    # Sequential execution
    execution_event = execute_btn.click(
        # Clear error message
        lambda: gr.HighlightedText.update(value=None, visible=False),
        inputs=[],
        outputs=[error_message],
    )
    prev_tasks = []
    for i, task in all_tasks.items():
        execution_event = execution_event.then(
            a.execute_task,
            inputs=[task.component_id, task.active_index, error_message]
            + task.inputs
            + [t.active_index for t in prev_tasks]
            + [o for t in prev_tasks for o in t.outputs],
            outputs=task.outputs + [error_message],
        )
        prev_tasks.append(task)

demo.launch()