Update tools.py

tools.py

@@ -9,6 +9,8 @@ from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
 import requests
 import os
+import subprocess
+import tempfile
 
 def download_file(task_id: str, file_name: str) -> str:
     """Downloads a file associated with a task_id and returns the local file path"""
@@ -145,4 +147,26 @@ def read_excel_file(file_path: str) -> Dict[str, Any]:
         return result
         
     except Exception as e:
-        return {"error": f"Failed to read Excel file: {str(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)}"