Update gensim/sim_runner.py

gensim/sim_runner.py

@@ -79,13 +79,13 @@ class SimulationRunner:
 
             self.generated_task = {'task-name': 'TASK_NAME_TEMPLATE', 'task-description': 'TASK_STRING_TEMPLATE', 'assets-used': ['ASSET_1', 'ASSET_2', Ellipsis]}
             print("generated_task\n", self.generated_task)
-            yield "Task Generated ==>", None, None
+            yield "Task Generated ==>", "", None, None
             self.generated_asset = self.agent.propose_assets()
             # self.generated_asset = {}
             print("generated_asset\n", self.generated_asset)
-            yield "Task Generated ==> Asset Generated ==> ", None, None
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", None, None
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", None, None
+            yield "Task Generated ==> Asset Generated ==> ", "", None, None
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", "", None, None
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", "", None, None
 
             online_code_buffer = {}
             for task_file in json.load(open(os.path.join('prompts/data', "generated_task_codes.json"))):
@@ -101,7 +101,7 @@ class SimulationRunner:
             self.generated_code = online_code_buffer[random_task_file]
             print("generated_code\n", self.generated_code)
             print("curr_task_name\n", self.curr_task_name)
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> ", self.generated_code, None
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> ", "", self.generated_code, None
 
             self.generated_tasks.append(self.generated_task)
             self.generated_task_assets.append(self.generated_asset)
@@ -110,9 +110,10 @@ class SimulationRunner:
         except:
             to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
             print("Task Creation Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
-
-            raise gradio.Error(f"Task Creation Exception: {str(traceback.format_exc())}")
+            self.log = to_print
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generation Failed", self.log, "", None
             self.task_creation_pass = False
+            return
 
         # self.curr_task_name = self.generated_task['task-name']
         print("task creation time {:.3f}".format(time.time() - start_time))