Update app.py

app.py

@@ -11,6 +11,9 @@ import string
 import sys
 import io
 import contextlib
+import re
+import textwrap
+
 
 def code_interpreter(code: str) -> str:
     """
@@ -143,21 +146,24 @@ class SuperSmartAgent:
     def execute_python_code(self, state):
         question = state["question"]
     
-        # Extract code from the question (optional enhancement)
-        if "```python" in question:
-            try:
-                code = question.split("```python")[1].split("```")[0].strip()
-            except IndexError:
-                code = ""
+        # Extract code block from ```python ... ```
+        code_blocks = re.findall(r"```python(.*?)```", question, re.DOTALL)
+        if code_blocks:
+            code = code_blocks[0].strip()
         else:
-            # Fallback code
+            # Default fallback code
             code = """
-                    def calculate():
-                        return 42
+                def calculate():
+                    return 42
+    
+                result = calculate()
+                print(result)
+            """
+    
+        # Clean up indentation
+        import textwrap
+        code = textwrap.dedent(code).strip()
     
-                    result = calculate()
-                    print(result)
-                    """
         try:
             result = code_interpreter(code)
             state["response"] = f"Python execution result:\n{result}"
@@ -172,6 +178,7 @@ class SuperSmartAgent:
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,