Update app.py

app.py

@@ -44,6 +44,7 @@ class AgentState(TypedDict):
     is_reversed: bool
     is_riddle: bool
     is_python: bool
+    file_path: str | None 
 
 class SuperSmartAgent:
     def __init__(self):
@@ -144,29 +145,23 @@ class SuperSmartAgent:
         return state
 
     def execute_python_code(self, state):
-        question = state["question"]
-    
-        # Extract code block from ```python ... ```
-        code_blocks = re.findall(r"```python(.*?)```", question, re.DOTALL)
-        if code_blocks:
-            code = code_blocks[0].strip()
+        file_path = state.get("file_path")
+        if file_path and os.path.isfile(file_path):
+            try:
+                with open(file_path, "r", encoding="utf-8") as f:
+                    code = f.read()
+                code = textwrap.dedent(code).strip()
+            except Exception as e:
+                state["response"] = f"Error reading file: {str(e)}"
+                return state
         else:
-            # Default fallback code
-            code = """
-                def calculate():
-                    return 5
-    
-                result = calculate()
-                print(result)
-            """
-    
-        # Clean up indentation
-        import textwrap
-        code = textwrap.dedent(code).strip()
+            # Fallback or error if no file is attached
+            state["response"] = "No valid Python file provided for execution."
+            return state
     
         try:
-            result = code_interpreter(code)
-            state["response"] = f"{result}"
+            result = code_interpreter(code)  # Your interpreter logic here
+            state["response"] = f"Python execution result:\n{result}"
         except Exception as e:
             state["response"] = f"Error executing Python code: {str(e)}"
     
@@ -175,10 +170,6 @@ class SuperSmartAgent:
 
 
 
-
-
-
-
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,