Update app.py

app.py

@@ -7,9 +7,31 @@ from smolagents import (
     DuckDuckGoSearchTool,
     OpenAIServerModel,
 )
-from smolagents.tools.python_repl import PythonREPL
+#from smolagents.tools.python_repl import PythonREPL
 import traceback # Import traceback for detailed error logging
 
+from smolagents import BaseTool
+
+class PythonREPLTool(BaseTool):
+    def __init__(self, timeout=10):
+        self.timeout = timeout
+
+    def run(self, code: str) -> str:
+        import subprocess
+        try:
+            result = subprocess.run(
+                ["python3", "-c", code],
+                capture_output=True,
+                text=True,
+                timeout=self.timeout,
+            )
+            if result.returncode == 0:
+                return result.stdout
+            else:
+                return f"Error:\n{result.stderr}"
+        except subprocess.TimeoutExpired:
+            return "Execution timed out"
+
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -24,7 +46,7 @@ class GaiaAgent:
         )
         # 2) Define the tools
         self.search_tool = DuckDuckGoSearchTool()
-        self.python_tool = PythonREPL(timeout=10)  # Initialize PythonREPLTool
+        self.python_tool = PythonREPLTool(timeout=10)  # Initialize PythonREPLTool
         # 3) Create the CodeAgent
         self.agent = CodeAgent(
             model=self.model,