Execute tasks and gen code for demos

actions.py

@@ -98,7 +98,3 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
                 visible=True,
             )
         ]
-
-
-def generate_code(task_id: int, active_index: int, code_prompt: str):
-    return all_tasks[int(task_id)].generate_code(active_index, code_prompt)

app.py

@@ -27,7 +27,7 @@ with gr.Blocks() as demo:
             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")
+        execute_btn = gr.Button("Execute tasks")
 
         # Edit layout
         add_task_btn.click(

components.py

@@ -337,36 +337,6 @@ class Task(Component):
         print(f"Executing {self._source}: {self._id}")
         return inner_task.execute(*args, vars_in_scope)
 
-    @property
-    def generate_code_input(self) -> gr.Textbox:
-        "This function only works for CodeTask"
-        return self._inner_tasks[1].code_prompt
-
-    @property
-    def generate_code_outputs(self):
-        "This function only works for CodeTask"
-        inner_task = self._inner_tasks[1]
-        return (
-            inner_task.raw_output,
-            inner_task.packages,
-            inner_task.script,
-            inner_task.error_message,
-            inner_task.accordion,
-        )
-
-    def generate_code(self, active_index, code_prompt: str = ""):
-        "This function only works for CodeTask"
-        inner_task = self._inner_tasks[active_index]
-        if isinstance(inner_task, CodeTask):
-            return CodeTask.generate_code(code_prompt)
-        return (
-            gr.Textbox.update(),
-            gr.Textbox.update(),
-            gr.Textbox.update(),
-            gr.HighlightedText.update(),
-            gr.Accordion.update(),
-        )
-
 
 MAX_TASKS = 10
 

examples/__init__.py

@@ -1,12 +1,12 @@
 from typing import List
 import gradio as gr
 
-from components import Task, TaskComponent
+from components import CodeTask, Task, TaskComponent
 
 
 def demo_buttons(demo_id, tasks: List[TaskComponent]):
     error_message = gr.HighlightedText(value=None, visible=False)
-    execute_btn = gr.Button("Execute")
+    execute_btn = gr.Button("Generate code and execute tasks")
 
     # Sequential execution
     execution_event = execute_btn.click(
@@ -17,6 +17,18 @@ def demo_buttons(demo_id, tasks: List[TaskComponent]):
     )
     prev_tasks = []
     for task in tasks:
+        if isinstance(task, CodeTask):
+            execution_event = execution_event.then(
+                task.generate_code,
+                inputs=[task.code_prompt],
+                outputs=[
+                    task.raw_output,
+                    task.packages,
+                    task.script,
+                    task.error_message,
+                    task.accordion,
+                ],
+            )
         execution_event = execution_event.then(
             execute_task,
             inputs=[demo_id, task.component_id, error_message]

examples/best_clubs.py

@@ -0,0 +1,54 @@
+import gradio as gr
+from components import AITask, CodeTask
+
+from examples import demo_buttons, demo_tasks
+
+
+DEMO_ID = __name__
+tasks = [
+    CodeTask(
+        0,
+        "https://openai.com/",
+        visible=True,
+        code_value="Make a google search.",
+    ),
+    AITask(
+        1,
+        """Your goal is to create an ad for a website.
+You will use an AI image generator to generate an image for your ad.
+
+Here is the text from the website:
+{t0}
+
+Create a prompt for the AI image generator.
+Avoid logos.""",
+        visible=True,
+    ),
+    CodeTask(
+        2,
+        "{t1}",
+        visible=True,
+        code_value="Use openai key <put_your_key_in_here>. Generate an image from a prompt. Return the url.",
+    ),
+    AITask(
+        1,
+        """Here is the text from a website:
+{t0}
+
+Here is a prompt that was used by an AI image generator to generate an image for an ad:
+{t1}
+
+Consider the website content and the prompt to create a headline for an ad.""",
+        visible=True,
+    ),
+]
+demo_tasks[DEMO_ID] = tasks
+
+
+def render():
+    demo_id = gr.Textbox(DEMO_ID, visible=False)
+    tasks[0].render()
+    tasks[1].render()
+    tasks[2].render()
+    tasks[3].render()
+    demo_buttons(demo_id, tasks)

examples/generate_ad.py

@@ -15,13 +15,13 @@ tasks = [
     AITask(
         1,
         """Your goal is to create an ad for a website.
-You are an expert in prompt engineering.
 You will use an AI image generator to generate an image for your ad.
 
 Here is the text from the website:
 {t0}
 
-Create a prompt for the AI image generator.""",
+Create a prompt for the AI image generator.
+Avoid logos.""",
         visible=True,
     ),
     CodeTask(
@@ -38,7 +38,7 @@ Create a prompt for the AI image generator.""",
 Here is a prompt that was used by an AI image generator to generate an image for an ad:
 {t1}
 
-Create a headline for the ad.""",
+Consider the website content and the prompt to create a headline for an ad.""",
         visible=True,
     ),
 ]