NIT for errors

actions.py

@@ -41,7 +41,7 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
     # We need to return outputs for all tasks in the row.
     outputs = [""] * n_avail_tasks
 
-    if active_index is None:  # Active index could be 0 == not active_index
+    if active_index is None or error_value:  # Active index could be 0 == not active_index
         return outputs + [
             gr.HighlightedText.update(
                 value=error_value, visible=error_value is not None
@@ -108,7 +108,7 @@ def execute_task(task_id: int, active_index: int, error_value, *args):
             )
         ]
     except Exception as e:
-        raise e
+        outputs[active_index] = "ERROR"
         return outputs + [
             gr.HighlightedText.update(
                 value=[(f"Error in Task {task_id} :: {e}", "ERROR")],

components.py

@@ -237,6 +237,7 @@ class CodeTask(TaskComponent):
             subprocess.check_call([sys.executable, "-m", "pip", "install", p])
             __import__(p)
         exec(function, locals())
+        # Should be last function in scope
         self._toolkit_func = list(locals().items())[-1][1]
 
         formatted_input = input.format(**vars_in_scope)