Update agent.py

agent.py

@@ -11,6 +11,8 @@ import json
 import io
 import contextlib
 import os
+import subprocess
+import tempfile
 
 OPENAI = os.getenv("OPENAI")
 
@@ -74,32 +76,75 @@ If the question specifies a format, please return it in the specified format. Do
     state['messages'] = state['messages'] + [llm.invoke([sys_msg])]
     return state
 
-
+    
 def python_interpreter(state: AgentState):
+    print("python interpreter")
+
     question_id = state['task_id']
     url = f"https://agents-course-unit4-scoring.hf.space/files/{question_id}"
-
-    # Step 1: Fetch Python code as bytes and decode
+    
+    # Step 1: Download the code
     response = requests.get(url)
     python_code = response.content.decode("utf-8")
-    # print(python_code)
-
-    # Step 2: Capture stdout
-    stdout = io.StringIO()
-    with contextlib.redirect_stdout(stdout):
-        try:
-            exec(python_code)  # Use empty dict for isolated global context
-        except Exception as e:
-            output = f"Error during execution: {e}"
+
+    print("Running user code with subprocess...")
+
+    # Step 2: Write code to temp file
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".py", delete=False) as tmp_file:
+        tmp_file.write(python_code)
+        tmp_file_path = tmp_file.name
+
+    try:
+        # Step 3: Run using subprocess (set timeout if needed)
+        result = subprocess.run(
+            ["python3", tmp_file_path],
+            capture_output=True,
+            text=True,
+            timeout=30  # seconds
+        )
+        if result.returncode != 0:
+            output = f"⚠️ Error:\n{result.stderr.strip()}"
         else:
-            output = stdout.getvalue().strip()
-    print("Code output : ", output)
-    # Step 3: Save output as message
+            output = result.stdout.strip() or "Code ran but produced no output."
+    except subprocess.TimeoutExpired:
+        output = "Execution timed out."
+    except Exception as e:
+        output = f"Unexpected error: {e}"
+
+    # Step 4: Update agent state
     state['messages'] = state['messages'] + [
-        AIMessage(content=f"The output of the python code is : {output}")
+        AIMessage(content=f"The output of the Python code is:\n```\n{output}\n```")
     ]
+    
     return state
 
+
+
+# def python_interpreter(state: AgentState):
+#     question_id = state['task_id']
+#     url = f"https://agents-course-unit4-scoring.hf.space/files/{question_id}"
+
+#     # Step 1: Fetch Python code as bytes and decode
+#     response = requests.get(url)
+#     python_code = response.content.decode("utf-8")
+#     # print(python_code)
+
+#     # Step 2: Capture stdout
+#     stdout = io.StringIO()
+#     with contextlib.redirect_stdout(stdout):
+#         try:
+#             exec(python_code)  # Use empty dict for isolated global context
+#         except Exception as e:
+#             output = f"Error during execution: {e}"
+#         else:
+#             output = stdout.getvalue().strip()
+#     print("Code output : ", output)
+#     # Step 3: Save output as message
+#     state['messages'] = state['messages'] + [
+#         AIMessage(content=f"The output of the python code is : {output}")
+#     ]
+#     return state
+
 def image_interpreter(state: AgentState):
     print("image interpreter")
     """Answers a question about an image. Provide raw image bytes and a question."""