File size: 3,678 Bytes
846ae23 d5b1bca 846ae23 d5b1bca 846ae23 d5b1bca 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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 |
from abc import ABC, abstractmethod
import gradio as gr
MAX_INPUTS = 10
MAX_TASKS = 50
class Component(ABC):
def __init__(self, visible: bool = False):
self.component = None
self.visible = visible
@abstractmethod
def render(self) -> gr.Box:
...
class Input(Component):
def render(self) -> None:
with gr.Box(visible=self.visible) as component:
with gr.Row():
self.output_name = gr.Textbox(
label="Input name (can be referenced with {})",
interactive=True,
placeholder="Variable name",
)
self.output = gr.Textbox(
label="Input value",
interactive=True,
placeholder="Variable value",
)
self.component = component
class AITask(Component):
def render(self) -> None:
with gr.Box(visible=self.visible) as component:
gr.Markdown(f"AI task")
with gr.Row():
with gr.Column():
self.prompt = gr.Textbox(
label="Instructions",
lines=15,
interactive=True,
placeholder="What is the AI assistant meant to do?",
)
with gr.Column():
with gr.Box():
self.output_name = gr.Textbox(
label="Output name", interactive=True, placeholder="var"
)
self.output = gr.Textbox(
label="",
lines=10,
interactive=False,
)
self.component = component
all_inputs = [Input() for _ in range(MAX_INPUTS)]
all_tasks = [AITask() for _ in range(MAX_TASKS)]
all_inputs[0].visible = True
all_tasks[0].visible = True
next_input = 1
next_task = 1
def _update_components(i: int, max: int):
return [gr.Box.update(visible=True)] * i + [gr.Box.update(visible=False)] * (
max - i
)
def add_input():
global next_input
if next_input < MAX_INPUTS:
next_input += 1
return _update_components(next_input, MAX_INPUTS)
def remove_input():
global next_input
if next_input > 0:
next_input -= 1
return _update_components(next_input, MAX_INPUTS)
def add_task():
global next_task
if next_task < MAX_TASKS:
next_task += 1
return _update_components(next_task, MAX_TASKS)
def remove_task():
global next_task
if next_task > 0:
next_task -= 1
return _update_components(next_task, MAX_TASKS)
with gr.Blocks() as demo:
# Layout
for i in all_inputs:
i.render()
with gr.Row():
add_input_btn = gr.Button("Add input variable")
remove_input_btn = gr.Button("Remove input variable")
execute_btn = gr.Button("Execute")
for t in all_tasks:
t.render()
with gr.Row():
add_task_btn = gr.Button("Add task")
remove_task_btn = gr.Button("Remove task")
# Event handling
add_input_btn.click(
add_input,
inputs=[],
outputs=[i.component for i in all_inputs],
)
remove_input_btn.click(
remove_input,
inputs=[],
outputs=[i.component for i in all_inputs],
)
add_task_btn.click(
add_task,
inputs=[],
outputs=[t.component for t in all_tasks],
)
remove_task_btn.click(
remove_task,
inputs=[],
outputs=[t.component for t in all_tasks],
)
demo.launch()
|