Update app.py

app.py

@@ -7,9 +7,32 @@ from smolagents import (
     DuckDuckGoSearchTool,
     OpenAIServerModel,
 )
-from smolagents.tools.python_repl import PythonREPLTool # Changed the import
 import traceback  # Import traceback for detailed error logging
 
+import subprocess
+
+class PythonREPLTool:
+    name = "python_repl"
+    description = "Runs Python code and returns the output or error."
+
+
+    def __init__(self, timeout=10):
+        self.timeout = timeout
+
+    def run(self, code: str) -> str:
+
+        try:
+            result = subprocess.run(
+                ["python3", "-c", code],
+                timeout=self.timeout,
+            )
+            if result.returncode == 0:
+                return result.stdout.strip()
+            else:
+                return f"Error:\n{result.stderr.strip()}"
+        except subprocess.TimeoutExpired:
+            return "Execution timed out."
+
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"