Update gensim/sim_runner.py

gensim/sim_runner.py

@@ -9,7 +9,7 @@ import imageio
 
 from pygments import highlight
 from pygments.lexers import PythonLexer
-from pygments.formatters import TerminalFormatter
+from pygments.formatters import HtmlFormatter, TerminalFormatter
 import gradio
 import time
 import random
@@ -31,6 +31,7 @@ class SimulationRunner:
         self.agent = agent
         self.critic = critic
         self.memory = memory
+        self.log = ""
 
         # statistics
         self.syntax_pass_rate = 0
@@ -107,8 +108,8 @@ class SimulationRunner:
             self.generated_task_programs.append(self.generated_code)
             self.generated_task_names.append(self.generated_task_name)
         except:
-            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter())
-            print("Task Creation Exception:", to_print)
+            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.task_creation_pass = False
@@ -127,30 +128,32 @@ 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()
             print("generated_asset\n", self.generated_asset)
-            yield "Task Generated ==> Asset Generated ==> ", None, None
+            yield "Task Generated ==> Asset Generated ==> ", "", None, None
             self.agent.api_review()
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", None, None
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> ", "", None, None
             self.critic.error_review(self.generated_task)
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", None, None
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> ", "", None, None
             self.generated_code, self.curr_task_name = self.agent.implement_task()
             self.task_asset_logs.append(self.generated_task["assets-used"])
             self.generated_task_name = self.generated_task["task-name"]
             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.log, self.generated_code, None
 
             self.generated_tasks.append(self.generated_task)
             self.generated_task_assets.append(self.generated_asset)
             self.generated_task_programs.append(self.generated_code)
             self.generated_task_names.append(self.generated_task_name)
         except:
-            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter())
-            print("Task Creation Exception:", to_print)
-            raise gradio.Error(f"Task Creati'BuildWheel'on Exception: {str(traceback.format_exc())}")
+            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
+            print("Task Creation Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
+            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))
@@ -236,11 +239,13 @@ class SimulationRunner:
             self.syntax_pass_rate += 1
 
         except:
-            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter())
+            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
             save_text(self.cfg['model_output_dir'], self.generated_task_name + '_error', str(traceback.format_exc()))
             print("========================================================")
-            print("Syntax Exception:", to_print)
-            raise gradio.Error(f"Syntax Exception: {str(traceback.format_exc())}")
+            print("Syntax Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
+            self.log = to_print
+
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> Code Syntax Parse Failed", self.log, self.generated_code, None
             return
 
         try:
@@ -296,7 +301,7 @@ class SimulationRunner:
             video_writer.close()
             print("save video to ", self.video_path)
 
-            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> Simulation Running completed", self.generated_code, self.video_path
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> Simulation Running completed", self.log, self.generated_code, self.video_path
             episode.append((obs, None, reward, info))
 
 
@@ -306,10 +311,10 @@ class SimulationRunner:
             # self.runtime_pass_rate += 1
             print("Runtime Test Pass!")
         except:
-            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter())
+            to_print = highlight(f"{str(traceback.format_exc())}", PythonLexer(), HtmlFormatter())
             save_text(self.cfg['model_output_dir'], self.generated_task_name + '_error', str(traceback.format_exc()))
             print("========================================================")
-            print("Runtime Exception:", to_print)
-
-            raise gradio.Error(f"Runtime Exception: {str(traceback.format_exc())}")
+            print("Runtime Exception:", highlight(f"{str(traceback.format_exc())}", PythonLexer(), TerminalFormatter()))
+            self.log = to_print
+            yield "Task Generated ==> Asset Generated ==> API Reviewed ==> Error Reviewed ==> Code Generated ==> Simulation Running Failed", self.log, self.generated_code, None
         self.memory.save_run(self.generated_task)