Update shared_vis_python_exe.py

shared_vis_python_exe.py

@@ -281,66 +281,66 @@ class PythonExecutor:
             
     #     return result, report
 
-def execute(
-    self,
-    code,
-    messages,
-    get_answer_from_stdout=True,
-    runtime_class=None,
-    answer_symbol=None,
-    answer_expr=None,
-) -> Tuple[Union[str, Dict[str, Any]], str]:
-    print(runtime_class)
-    runtime = self.persistent_runtime
-
-    try:
-        if get_answer_from_stdout:
-            program_io = io.StringIO()
-            with redirect_stdout(program_io):
+    def execute(
+        self,
+        code,
+        messages,
+        get_answer_from_stdout=True,
+        runtime_class=None,
+        answer_symbol=None,
+        answer_expr=None,
+    ) -> Tuple[Union[str, Dict[str, Any]], str]:
+        print(runtime_class)
+        runtime = self.persistent_runtime
+    
+        try:
+            if get_answer_from_stdout:
+                program_io = io.StringIO()
+                with redirect_stdout(program_io):
+                    runtime.exec_code("\n".join(code))
+                program_io.seek(0)
+                result = program_io.read()
+            elif answer_symbol:
                 runtime.exec_code("\n".join(code))
-            program_io.seek(0)
-            result = program_io.read()
-        elif answer_symbol:
-            runtime.exec_code("\n".join(code))
-            result = runtime._global_vars.get(answer_symbol, "")
-        elif answer_expr:
-            runtime.exec_code("\n".join(code))
-            result = runtime.eval_code(answer_expr)
-        else:
-            if len(code) > 1:
-                runtime.exec_code("\n".join(code[:-1]))
-                result = runtime.eval_code(code[-1])
-            else:
+                result = runtime._global_vars.get(answer_symbol, "")
+            elif answer_expr:
                 runtime.exec_code("\n".join(code))
-                result = ""
-
-        # Check for captured figures
-        captured_figures = runtime._global_vars.get("_captured_figures", [])
-        if captured_figures:
+                result = runtime.eval_code(answer_expr)
+            else:
+                if len(code) > 1:
+                    runtime.exec_code("\n".join(code[:-1]))
+                    result = runtime.eval_code(code[-1])
+                else:
+                    runtime.exec_code("\n".join(code))
+                    result = ""
+    
+            # Check for captured figures
+            captured_figures = runtime._global_vars.get("_captured_figures", [])
+            if captured_figures:
+                result = {
+                    'text': result,
+                    'images': captured_figures
+                } if result else {'images': captured_figures}
+            else:
+                result = {'text': result} if result else {}
+    
+            report = "Done"
+    
+        except Exception as e:
             result = {
-                'text': result,
-                'images': captured_figures
-            } if result else {'images': captured_figures}
-        else:
-            result = {'text': result} if result else {}
-
-        report = "Done"
-
-    except Exception as e:
-        result = {
-            'error': str(e),
-            'traceback': traceback.format_exc()
-        }
-        report = f"Error: {str(e)}\n{traceback.format_exc()}"
-
-    # Ensure result is serializable
-    try:
-        pickle.dumps(result)
-    except Exception as e:
-        result = f"Result serialization error: {str(e)}"
-        report = f"Serialization Error: {str(e)}"
-
-    return result, report
+                'error': str(e),
+                'traceback': traceback.format_exc()
+            }
+            report = f"Error: {str(e)}\n{traceback.format_exc()}"
+    
+        # Ensure result is serializable
+        try:
+            pickle.dumps(result)
+        except Exception as e:
+            result = f"Result serialization error: {str(e)}"
+            report = f"Serialization Error: {str(e)}"
+    
+        return result, report