Allow reference other task outputs

actions.py

@@ -55,8 +55,8 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
     # - start_inputs: Where the active task inputs start within args.
     # - end_inputs: End of the active task inputs.
     # - task_inputs: The active task inputs.
-    # - prev_active_indexes: Indexes of the active tasks in the previous tasks.
-    # - prev_task_outputs: Outputs of the previous tasks.
+    # - other_active_indexes: Indexes of the active tasks in the other tasks.
+    # - other_task_outputs: Outputs of every other task.
     start_inputs = 0
     end_inputs = 0
     end_all_inputs = sum(inner_n_inputs)
@@ -66,8 +66,8 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
             break
         start_inputs += n
     task_inputs = args[start_inputs:end_inputs]
-    prev_active_indexes = args[end_all_inputs : end_all_inputs + task_id]
-    prev_task_outputs = args[end_all_inputs + task_id :]
+    other_active_indexes = args[end_all_inputs : end_all_inputs + MAX_TASKS - 1]
+    other_task_outputs = args[end_all_inputs + MAX_TASKS - 1 :]
 
     # If no inputs, skip
     non_empty_inputs = [i for i in task_inputs if i]
@@ -76,10 +76,12 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
 
     # Put task outputs in a dictionary with names.
     vars_in_scope = {}
-    for i, prev_active_index in enumerate(prev_active_indexes):
-        vars_in_scope[f"{Task.vname}{i}"] = prev_task_outputs[
-            i * n_avail_tasks + int(prev_active_index)
-        ]
+    for i, other_active_index in enumerate(other_active_indexes):
+        if other_active_index is not None:
+            other_task_id = i if i < task_id else i + 1
+            vars_in_scope[f"{Task.vname}{other_task_id}"] = other_task_outputs[
+                i * n_avail_tasks + int(other_active_index)
+            ]
 
     try:
         # Task logic gets inserted into the right index

app.py

@@ -1,7 +1,13 @@
 import gradio as gr
 
 import actions as a
-from examples import authenticate_google, best_clubs, generate_ad, seo, summarize_website
+from examples import (
+    authenticate_google,
+    best_clubs,
+    generate_ad,
+    seo,
+    summarize_website,
+)
 from components import all_tasks, Tasks
 
 
@@ -17,7 +23,7 @@ with gr.Blocks() as demo:
     <br>**Code Task**: You will need code to do certain things that ChatGPT can't do, like access the internet or iterate over 4k+ tokens.
     <br> With this task, ChatGPT will generate code and then execute it. The code must be generated before executing all tasks.
     <br>
-    <br>Output from previous tasks can be referenced in subsequen tasks with {tn}. Max 10 tasks allowed (for now).
+    <br>Output from other tasks can be referenced in the current task with {tn}. Max 10 tasks allowed (for now).
     """
     )
     with gr.Tab("Toolkit"):
@@ -48,17 +54,15 @@ with gr.Blocks() as demo:
             inputs=[],
             outputs=[error_message],
         )
-        prev_tasks = []
         for i, task in all_tasks.items():
             execution_event = execution_event.then(
                 a.execute_task,
                 inputs=[task.component_id, task.active_index, error_message]
                 + task.inputs
-                + [t.active_index for t in prev_tasks]
-                + [o for t in prev_tasks for o in t.outputs],
+                + [t.active_index for t in all_tasks.values() if t != task]
+                + [o for t in all_tasks.values() if t != task for o in t.outputs],
                 outputs=task.outputs + [error_message],
             )
-            prev_tasks.append(task)
 
     # Examples
     summarize_website.render()