Update tools.py

tools.py

@@ -14,6 +14,7 @@ from transformers import pipeline
 import asyncio
 import os
 import io
+import ast
 from dotenv import load_dotenv
 #from PIL import Image
 #from io import StringIO
@@ -227,7 +228,7 @@ def detect_objects(image_path: str) -> str:
 @tool
 def run_python(code: str) -> str:
     """
-    Executes Python code safely and returns stdout.
+    Executes Python code safely and returns stdout or the last expression result.
     
     Args:
         code (str): The Python code to execute.
@@ -235,14 +236,34 @@ def run_python(code: str) -> str:
     Returns:
         str: Captured stdout and/or result.
     """
+    
     stdout = io.StringIO()
     local_vars = {}
+    
     try:
+        # Parse code into AST
+        parsed = ast.parse(code, mode="exec")
+        last_expr = None
+        if parsed.body and isinstance(parsed.body[-1], ast.Expr):
+            # If last node is an expression, separate it
+            last_expr = parsed.body.pop()
+        
         with contextlib.redirect_stdout(stdout):
-            exec(code, {}, local_vars)
-        # Return printed output or last expression result
+            # Run everything except the last expression
+            exec(compile(parsed, filename="<ast>", mode="exec"), {}, local_vars)
+            
+            # Evaluate last expression if present
+            if last_expr is not None:
+                _result = eval(compile(ast.Expression(last_expr.value), 
+                                       filename="<ast>", mode="eval"), {}, local_vars)
+                local_vars["_result"] = _result
+        
+        # Return _result if set
         if "_result" in local_vars:
             return str(local_vars["_result"])
+        
+        # Otherwise, return stdout
         return stdout.getvalue().strip() or "Code executed successfully."
+    
     except Exception as e:
         return f"Execution error: {e}"