Ad example

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
 import actions as a
-from examples import summarize_website
+from examples import generate_ad, summarize_website
 from components import all_tasks, Tasks
 
 
@@ -61,5 +61,7 @@ with gr.Blocks() as demo:
             prev_tasks.append(task)
     with gr.Tab("Example: Summarize website"):
         summarize_website.render()
+    with gr.Tab("Example: Generate ad"):
+        generate_ad.render()
 
 demo.launch()

components.py

@@ -193,41 +193,35 @@ class CodeTask(TaskComponent):
         print(f"Generating code.")
         try:
             raw_output = llm_call(
-                f"""
-                Write one python function for the following request:
-                {code_prompt}
-
-                Use pip packages where available.
-                For example, if you wanted to make a google search, use the googlesearch-python package instead of scraping google.
-                Include only the necessary imports.
-                Instead of printing or saving to disk, the function should return the data.
-                """
+                f"""Write one python function for the following request:
+{code_prompt}
+
+Use pip packages where available.
+For example, if you wanted to make a google search, use the googlesearch-python package instead of scraping google.
+Include only the necessary imports.
+Instead of printing or saving to disk, the function should return the data."""
             )
             with ThreadPoolExecutor() as executor:
                 packages, script = tuple(
                     executor.map(
                         llm_call,
                         [
-                            f"""
-                            The following text has some python code:
-                            {raw_output}
+                            f"""The following text has some python code:
+{raw_output}
 
-                            Find the pip packages that need to be installed and get their corresponsing names in pip.
-                            Package names in the imports and in pip might be different. Use the correct pip names.
+Find the pip packages that need to be installed and get their corresponsing names in pip.
+Package names in the imports and in pip might be different. Use the correct pip names.
                             
-                            Put them in a JSON:
-                            ```
-                            {{
-                                "packages": Python list to be used with eval(). If no packages, empty list.
-                            }}
-                            ```
-                            """,
-                            f"""
-                            The following text has some python code:
-                            {raw_output}
-
-                            Extract it. Remove anything after the function definition.
-                            """,
+Put them in a JSON:
+```
+{{
+    "packages": Python list to be used with eval(). If no packages, empty list.
+}}
+```""",
+                            f"""The following text has some python code:
+{raw_output}
+
+Extract it. Remove anything after the function definition.""",
                         ],
                     )
                 )

examples/__init__.py

@@ -0,0 +1,70 @@
+from typing import List
+import gradio as gr
+
+from components import Task, TaskComponent
+
+
+def demo_buttons(demo_id, tasks: List[TaskComponent]):
+    error_message = gr.HighlightedText(value=None, visible=False)
+    execute_btn = gr.Button("Execute")
+
+    # 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 task in tasks:
+        execution_event = execution_event.then(
+            execute_task,
+            inputs=[demo_id, task.component_id, error_message]
+            + task.inputs
+            + [t.output for t in prev_tasks],
+            outputs=[task.output, error_message],
+        )
+        prev_tasks.append(task)
+
+
+demo_tasks = {}
+
+
+def execute_task(demo_id: str, task_id: int, error_value, *args):
+    error_update = gr.HighlightedText.update(
+        value=error_value, visible=error_value is not None
+    )
+
+    if error_value:
+        return ["", error_update]
+
+    task_id = int(task_id)
+    n_inputs = demo_tasks[demo_id][task_id].n_inputs
+
+    task_inputs = args[:n_inputs]
+    prev_task_outputs = args[n_inputs:]
+
+    non_empty_inputs = [i for i in task_inputs if i]
+    if not non_empty_inputs:
+        return ["", error_update]
+
+    # Put task outputs in a dictionary with names.
+    vars_in_scope = {f"{Task.vname}{i}": o for i, o in enumerate(prev_task_outputs)}
+
+    try:
+        # Task logic gets inserted into the right index
+        output = demo_tasks[demo_id][task_id].execute(
+            *task_inputs, vars_in_scope=vars_in_scope
+        )
+        return [output, error_update]
+    except Exception as e:
+        import traceback
+
+        print(traceback.format_tb(e.__traceback__))
+        return [
+            "ERROR",
+            gr.HighlightedText.update(
+                value=[(f"Error in Task {task_id} :: {e}", "ERROR")],
+                visible=True,
+            ),
+        ]

examples/actions.py

@@ -1,45 +0,0 @@
-import gradio as gr
-
-from components import Task
-
-demo_tasks = {}
-
-
-def execute_task(demo_id: str, task_id: int, error_value, *args):
-    error_update = gr.HighlightedText.update(
-        value=error_value, visible=error_value is not None
-    )
-
-    if error_value:
-        return ["", error_update]
-
-    task_id = int(task_id)
-    n_inputs = demo_tasks[demo_id][task_id].n_inputs
-
-    task_inputs = args[:n_inputs]
-    prev_task_outputs = args[n_inputs:]
-
-    non_empty_inputs = [i for i in task_inputs if i]
-    if not non_empty_inputs:
-        return ["", error_update]
-
-    # Put task outputs in a dictionary with names.
-    vars_in_scope = {f"{Task.vname}{i}": o for i, o in enumerate(prev_task_outputs)}
-
-    try:
-        # Task logic gets inserted into the right index
-        output = demo_tasks[demo_id][task_id].execute(
-            *task_inputs, vars_in_scope=vars_in_scope
-        )
-        return [output, error_update]
-    except Exception as e:
-        import traceback
-
-        print(traceback.format_tb(e.__traceback__))
-        return [
-            "ERROR",
-            gr.HighlightedText.update(
-                value=[(f"Error in Task {task_id} :: {e}", "ERROR")],
-                visible=True,
-            ),
-        ]

examples/generate_ad.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="Get text from a website. No html. No empty lines.",
+    ),
+    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.""",
+        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}
+
+Create a headline for the 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/summarize_website.py

@@ -1,43 +1,24 @@
 import gradio as gr
 from components import AITask, CodeTask
 
-import examples.actions as ea
+from examples import demo_buttons, demo_tasks
 
 
-DEMO_ID = "hello"
+DEMO_ID = __name__
 tasks = [
     CodeTask(
         0,
-        "https://openai.com/",
+        "https://huggingface.co/",
         visible=True,
-        code_value="Get text from a website (no html). No empty lines.",
+        code_value="Get text from a website. No html. No empty lines.",
     ),
     AITask(1, "Summarize: {t0}", visible=True),
 ]
-ea.demo_tasks[DEMO_ID] = tasks
+demo_tasks[DEMO_ID] = tasks
 
 
 def render():
     demo_id = gr.Textbox(DEMO_ID, visible=False)
     tasks[0].render()
     tasks[1].render()
-    error_message = gr.HighlightedText(value=None, visible=False)
-    execute_btn = gr.Button("Execute")
-
-    # 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 task in tasks:
-        execution_event = execution_event.then(
-            ea.execute_task,
-            inputs=[demo_id, task.component_id, error_message]
-            + task.inputs
-            + [t.output for t in prev_tasks],
-            outputs=[task.output, error_message],
-        )
-        prev_tasks.append(task)
+    demo_buttons(demo_id, tasks)