Update tools.py

tools.py

@@ -149,16 +149,50 @@ def read_excel_file(file_path: str) -> Dict[str, Any]:
     except Exception as e:
         return {"error": f"Failed to read Excel file: {str(e)}"}
 
+# def execute_python_code(file_path: str, timeout: int = 30) -> str:
+#     """Execute Python code safely with subprocess"""
+#     try:
+#         # Run in isolated subprocess with timeout
+#         result = subprocess.run(
+#             ['python', file_path],
+#             capture_output=True,
+#             text=True,
+#             timeout=timeout,
+#             cwd=tempfile.gettempdir()  # Run in temp directory
+#         )
+        
+#         if result.returncode == 0:
+#             return result.stdout.strip()
+#         else:
+#             return f"Error: {result.stderr}"
+            
+#     except subprocess.TimeoutExpired:
+#         return "Error: Code execution timed out"
+#     except Exception as e:
+#         return f"Error executing code: {str(e)}"
+
 def execute_python_code(file_path: str, timeout: int = 30) -> str:
     """Execute Python code safely with subprocess"""
     try:
+        # Check if file exists, if not try common locations
+        if not os.path.exists(file_path):
+            # Try in downloads directory
+            alt_path = os.path.join("downloads", os.path.basename(file_path))
+            if os.path.exists(alt_path):
+                file_path = alt_path
+            else:
+                return f"Error: File not found at {file_path} or {alt_path}"
+        
+        # Ensure we have absolute path
+        file_path = os.path.abspath(file_path)
+        
         # Run in isolated subprocess with timeout
         result = subprocess.run(
             ['python', file_path],
             capture_output=True,
             text=True,
             timeout=timeout,
-            cwd=tempfile.gettempdir()  # Run in temp directory
+            cwd=os.path.dirname(file_path)  # Run in the file's directory
         )
         
         if result.returncode == 0: