Update demo

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
 import actions as a
-from examples import hello
+from examples import summarize_website
 from components import all_tasks, Tasks
 
 
@@ -10,18 +10,14 @@ with gr.Blocks() as demo:
     gr.Markdown(
         """
     # Toolkit
-    Assemble tasks to build an E2E application. Give instructions with text.
+    Assemble tasks to build an E2E application with everyday language.
     <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 to do something for you. Eg, get the text from a website.
-    <br> The code for the Code Tasks must be generated before executing the whole application.
+    <br> Use this task for things that ChatGPT can't do on its own, like access the internet or iterate over 4K+ tokens.
     <br>
     <br>Output from previous tasks can be referenced in subsequen tasks with {tn}. Max 10 tasks allowed (for now).
-    <br>
-    <br>Example application:
-    <br>1. Code Task: Get the text from a website.
-    <br>2. AI Task: Summarize {t0}.
     """
     )
     with gr.Tab("Toolkit"):
@@ -63,7 +59,7 @@ with gr.Blocks() as demo:
                 outputs=task.outputs + [error_message],
             )
             prev_tasks.append(task)
-    with gr.Tab("Example: Hello world"):
-        hello.render()
+    with gr.Tab("Example: Summarize website"):
+        summarize_website.render()
 
 demo.launch()

components.py

@@ -1,7 +1,7 @@
 from concurrent.futures import ThreadPoolExecutor
 import re
 from abc import ABC, abstractmethod
-from typing import Any, Dict, List, Union
+from typing import Any, Dict, List, Optional, Union
 
 import gradio as gr
 
@@ -102,20 +102,25 @@ class AITask(TaskComponent):
     def inputs(self) -> List[gr.Textbox]:
         return [self.input]
 
-    def execute(self, prompt: str, vars_in_scope: Dict[str, Any]) -> str:
+    def execute(self, prompt: str, vars_in_scope: Dict[str, Any]) -> Optional[str]:
         formatted_prompt = self.format_input(prompt, vars_in_scope)
-        return ai.llm.next([{"role": "user", "content": formatted_prompt}])
+        if formatted_prompt:
+            return ai.llm.next([{"role": "user", "content": formatted_prompt}])
 
 
 class CodeTask(TaskComponent):
     name = "Code Task"
 
+    def __init__(self, id_: int, value: str = "", visible: bool = False, code_value: str = ""):
+        super().__init__(id_, value, visible)
+        self._initial_code_value = code_value
+
     def _render(self) -> gr.Column:
         with gr.Column(visible=self._initial_visbility) as gr_component:
             code_prompt = gr.Textbox(
                 label="What would you like to do?",
                 interactive=True,
-                value=self._initial_value,
+                value=self._initial_code_value,
             )
             generate_code = gr.Button("Generate code")
             with gr.Row():
@@ -140,6 +145,7 @@ class CodeTask(TaskComponent):
                     self.input = gr.Textbox(
                         label="Input",
                         interactive=True,
+                        value=self._initial_value
                     )
                 with gr.Column():
                     self.output = gr.Textbox(
@@ -250,15 +256,19 @@ class CodeTask(TaskComponent):
     def execute(
         self, packages: str, function: str, input: str, vars_in_scope: Dict[str, Any]
     ):
+        if not function:
+            return None
+
         import inspect
-        import subprocess
-        import sys
 
-        function = function.strip()
+        def install():
+            import subprocess
+            import sys
 
-        for p in eval(packages):
-            subprocess.check_call([sys.executable, "-m", "pip", "install", p])
+            for p in eval(packages):
+                subprocess.check_call([sys.executable, "-m", "pip", "install", p])
 
+        function = function.strip()
         exec(function, locals())
         # Looking for the last defined function
         for var in reversed(locals().values()):
@@ -272,7 +282,11 @@ class CodeTask(TaskComponent):
                 formatted_input = eval(formatted_input)
             except:
                 pass
-            return self._toolkit_func(formatted_input)
+            if formatted_input:
+                install()
+                return self._toolkit_func(formatted_input)
+            return None
+        install()
         return self._toolkit_func()
 
 

examples/{hello.py → summarize_website.py}

@@ -6,8 +6,13 @@ import examples.actions as ea
 
 DEMO_ID = "hello"
 tasks = [
-    CodeTask(0, "Say hello world", visible=True),
-    AITask(1, "Explain this to me: {t0}", visible=True),
+    CodeTask(
+        0,
+        "https://openai.com/",
+        visible=True,
+        code_value="Get text from a website (no html). No empty lines.",
+    ),
+    AITask(1, "Summarize: {t0}", visible=True),
 ]
 ea.demo_tasks[DEMO_ID] = tasks