Update app.py

app.py

@@ -8,6 +8,24 @@ from langgraph.graph import StateGraph, END
 from typing import TypedDict
 import string
 
+import sys
+import io
+import contextlib
+
+def code_interpreter(code: str) -> str:
+    """
+    Executes the given Python code string and captures its final printed output.
+    """
+    buffer = io.StringIO()
+    try:
+        # Redirect stdout to buffer
+        with contextlib.redirect_stdout(buffer):
+            exec(code, {})
+        return buffer.getvalue().strip()
+    except Exception as e:
+        return f"Execution Error: {e}"
+
+
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -124,28 +142,29 @@ class SuperSmartAgent:
 
     def execute_python_code(self, state):
         question = state["question"]
-
-        # Check if the question is about executing Python code
-        if "code" in question.lower():
-            # Placeholder for actual file handling and execution logic
-            # In a real implementation, you would read the attached file and execute it
-            python_code = """
-                # Example Python code
-                def calculate():
-                    return 42
-                
-                result = calculate()
-                print(result)
-                """
+    
+        # Extract code from the question (optional enhancement)
+        if "```python" in question:
             try:
-                # Use code_interpreter to execute the Python code
-                result = code_interpreter(code=python_code)
-                state["response"] = f"The final numeric output is: {result}"
-            except Exception as e:
-                state["response"] = f"Error executing Python code: {str(e)}"
+                code = question.split("```python")[1].split("```")[0].strip()
+            except IndexError:
+                code = ""
         else:
-            state["response"] = "This question does not seem to involve executing Python code."
-
+            # Fallback code
+            code = """
+    def calculate():
+        return 42
+    
+    result = calculate()
+    print(result)
+    """
+    
+        try:
+            result = code_interpreter(code)
+            state["response"] = f"Python execution result:\n{result}"
+        except Exception as e:
+            state["response"] = f"Error executing Python code: {str(e)}"
+    
         return state
 
 
@@ -153,6 +172,7 @@ class SuperSmartAgent:
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,