Update app.py

app.py

@@ -145,25 +145,36 @@ class SuperSmartAgent:
         return state
 
     def execute_python_code(self, state):
-        file_path = state.get("file_path")
-        if file_path and os.path.isfile(file_path):
+        question = state["question"]
+        file_name = state.get("file_name")
+    
+        # Case 1: Embedded code block
+        code_blocks = re.findall(r"```(?:python)?(.*?)```", question, re.DOTALL)
+        if code_blocks:
+            code = textwrap.dedent(code_blocks[0]).strip()
+        elif file_name and file_name.endswith(".py"):
+            # Case 2: Load code from attached Python file
+            file_path = os.path.join("data", file_name)  # Make sure this path matches the actual mount
             try:
-                with open(file_path, "r", encoding="utf-8") as f:
+                with open(file_path, "r") as f:
                     code = f.read()
-                code = textwrap.dedent(code).strip()
             except Exception as e:
-                state["response"] = f"Error reading file: {str(e)}"
+                state["response"] = f"Error loading Python file: {e}"
                 return state
         else:
-            # Fallback or error if no file is attached
-            state["response"] = "No valid Python file provided for execution."
-            return state
+            # Default fallback
+            code = """
+            def calculate():
+                return 5
+            result = calculate()
+            print(result)
+            """
     
         try:
-            result = code_interpreter(code)  # Your interpreter logic here
-            state["response"] = f"Python execution result:\n{result}"
+            result = code_interpreter(code)
+            state["response"] = str(result)
         except Exception as e:
-            state["response"] = f"Error executing Python code: {str(e)}"
+            state["response"] = f"Error executing Python code: {e}"
     
         return state