Update app.py

app.py

@@ -66,27 +66,33 @@ def search_tool(query: str) -> str:
     except Exception as e:
         return f"Error running search: {e}"
 
-@tool
+@tooldef
 def code_interpreter(code: str) -> str:
     """
     Executes a string of Python code and returns its stdout, stderr, and any error.
-    Use this for calculations, data manipulation (including pandas on dataframes read from files), list operations, string manipulations, or to solve logic puzzles step-by-step.
-    When solving a logic problem, write out all the steps of your reasoning as comments in the code.
-    The code runs in a sandboxed environment. 'pandas' (as pd) and 'openpyxl' are available.
-    Ensure the code is complete and executable. If printing, use print().
+    ...
     """
     print(f"--- Calling Code Interpreter with code:\n{code}\n---")
     output_stream = io.StringIO()
     error_stream = io.StringIO()
     try:
         with contextlib.redirect_stdout(output_stream), contextlib.redirect_stderr(error_stream):
-            exec(code, {"pd": pd}, {})
+            # --- FIX IS HERE ---
+            # Create a safe environment that includes 'pd' and standard Python built-ins
+            safe_globals = {
+                "pd": pd,
+                "__builtins__": __builtins__  # This is the crucial addition
+            }
+            # Execute the code within this safe environment
+            exec(code, safe_globals, {})
+            # --- END FIX ---
+            
         stdout = output_stream.getvalue(); stderr = error_stream.getvalue()
         if stderr: return f"Error: {stderr}\nStdout: {stdout}"
         if stdout: return f"Success:\n{stdout}"
         return "Success: Code executed without error and produced no stdout."
     except Exception as e: return f"Execution failed with error: {str(e)}"
-
+        
 @tool
 def read_file(path: str) -> str:
     """Reads the content of a file at the specified path. Use this to examine files provided in the question."""