Update app.py

app.py

@@ -7,6 +7,7 @@ from io import BytesIO
 from PIL import Image
 import argparse
 from vis_python_exe import PythonExecutor
+from shared_vis_python_exe import SharedRuntimeExecutor
 from openai import OpenAI
 from typing import Optional, Union
 import gradio as gr
@@ -243,7 +244,8 @@ def o3_chat(api_key, base_url, question, image):
     print("done!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
     # 初始化组件
     client = OpenAI(api_key=api_key, base_url=base_url)
-    executor = PythonExecutor()
+    # executor = PythonExecutor()
+    executor = SharedRuntimeExecutor(var_whitelist="RETAIN_ALL_VARS")
 
     prompt_template = json.load(open("./prompt_template_vis.json", "r", encoding="utf-8"))
     prompt_type = 'vistool'