Update Modules/Code_Interpreter.py

Modules/Code_Interpreter.py

@@ -1,16 +1,14 @@
 from __future__ import annotations
 
-import os
-import sys
-from io import StringIO
 from typing import Annotated
 
 import gradio as gr
 from ._docstrings import autodoc
-from .File_System import ROOT_DIR
+from ._core import sandboxed_exec
 
 from app import _log_call_end, _log_call_start, _truncate_for_log
 
+
 # Single source of truth for the LLM-facing tool description
 TOOL_SUMMARY = (
     "Execute Python code from the tool root; returns captured stdout or the exception text."
@@ -22,42 +20,7 @@ TOOL_SUMMARY = (
 )
 def Code_Interpreter(code: Annotated[str, "Python source code to run; stdout is captured and returned."]) -> str:
     _log_call_start("Code_Interpreter", code=_truncate_for_log(code or "", 300))
-    if code is None:
-        result = "No code provided."
-        _log_call_end("Code_Interpreter", result)
-        return result
-    from .File_System import safe_open
-
-    old_stdout = sys.stdout
-    old_cwd = os.getcwd()
-    redirected_output = sys.stdout = StringIO()
-    
-    # Create safe builtins
-    safe_builtins = None
-    if isinstance(__builtins__, dict):
-        safe_builtins = __builtins__.copy()
-    else:
-        safe_builtins = vars(__builtins__).copy()
-    safe_builtins["open"] = safe_open
-    
-    env = {
-        "open": safe_open,
-        "__builtins__": safe_builtins,
-        "print": print,
-    }
-    
-    try:
-        os.chdir(ROOT_DIR)
-        exec(code, env)
-        result = redirected_output.getvalue()
-    except Exception as exc:  # pylint: disable=broad-except
-        result = str(exc)
-    finally:
-        sys.stdout = old_stdout
-        try:
-            os.chdir(old_cwd)
-        except Exception:
-            pass
+    result = sandboxed_exec(code, ast_mode=False)
     _log_call_end("Code_Interpreter", _truncate_for_log(result))
     return result