File size: 2,417 Bytes
d5b1bca a35fa4d 5c20801 d5b1bca 759ceaa 169710f db9c02a 169710f bc38777 846ae23 bc38777 846ae23 169710f db9c02a 846ae23 d371605 846ae23 a35fa4d b92b58b 5c20801 b92b58b 846ae23 a35fa4d b92b58b 5c20801 b92b58b 846ae23 a35fa4d b92b58b 5c20801 b92b58b 846ae23 a35fa4d b92b58b 5c20801 b92b58b 846ae23 5c20801 a35fa4d 5c20801 169710f 5c20801 a35fa4d 169710f 5c20801 846ae23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import gradio as gr
import actions as a
from components import all_inputs, all_tasks
with gr.Blocks() as demo:
# Initial layout
gr.Markdown(
"""
# Toolkit
Define input variables to be used in your tasks.
<br>Task outputs can be used in subsequent tasks.
<br>
<br>AI tasks call into ChatGPT to perform actions.
<br>Chain inputs and tasks to build an E2E application.
<br>
<br>Example prompt: "Translate the following text into spanish and add {v0} more sentences: {t0}".
"""
)
for i in all_inputs.values():
i.render()
with gr.Row():
add_input_btn = gr.Button("Add input variable")
remove_input_btn = gr.Button("Remove input variable")
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_input_btn.click(
a.add_input,
inputs=[i.visible for i in all_inputs.values()],
outputs=[i.gr_component for i in all_inputs.values()] # type: ignore
+ [i.visible for i in all_inputs.values()],
)
remove_input_btn.click(
a.remove_input,
inputs=[i.visible for i in all_inputs.values()],
outputs=[i.gr_component for i in all_inputs.values()] # type: ignore
+ [i.visible for i in all_inputs.values()],
)
add_task_btn.click(
a.add_task,
inputs=[i.visible for i in all_tasks.values()],
outputs=[i.gr_component for i in all_tasks.values()] # type: ignore
+ [i.visible for i in all_tasks.values()],
)
remove_task_btn.click(
a.remove_task,
inputs=[i.visible for i in all_tasks.values()],
outputs=[i.gr_component for i in all_tasks.values()] # type: ignore
+ [i.visible for i in all_tasks.values()],
)
# Sequential execution
execution_event = execute_btn.click(
a._clear_error, inputs=[], outputs=[error_message]
)
for i, task in all_tasks.items():
execution_event = execution_event.then(
a.execute_task,
inputs=[task.component_id, task.internal.prompt, error_message] + a._get_all_vars_up_to(i), # type: ignore
outputs=[task.output, error_message],
)
demo.launch()
|