Update app.py

app.py

@@ -9,6 +9,7 @@ import base64
 import json
 import sys
 import contextlib
+import traceback
 #from tavily import TavilyClient
 from io import StringIO
 from langchain_tavily import TavilySearch
@@ -20,6 +21,7 @@ from langchain_core.tools import tool
 from langchain_core.messages import HumanMessage, SystemMessage, BaseMessage
 from langchain.chat_models import init_chat_model
 from typing import Annotated,Sequence, TypedDict, Literal, Dict
+from contextlib import redirect_stdout, redirect_stderr
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -86,15 +88,6 @@ web_search = TavilySearch(
         topic="general",
     )
 
-@contextlib.contextmanager
-def stdoutIO(stdout=None):
-    old = sys.stdout
-    if stdout is None:
-        stdout = StringIO()
-    sys.stdout = stdout
-    yield stdout
-    sys.stdout = old
-
 def python_code_runner_tool(task_id:str) -> str:
     """
         Download and run python code, capturing the output.
@@ -110,6 +103,26 @@ def python_code_runner_tool(task_id:str) -> str:
 
     print(f"python_code: {python_code}")
 
+    saida = io.StringIO()
+    erros = io.StringIO()
+
+    try:
+        # Captura stdout e stderr usando contexto
+        with redirect_stdout(saida), redirect_stderr(erros):
+            exec(python_code, {'__name__': '__main__'})
+
+        # Pega o conteúdo das saídas
+        saida_valor = saida.getvalue()
+        erro_valor = erros.getvalue()
+
+        if erro_valor:
+            return f"[ERRO DE EXECUÇÃO]:\n{erro_valor}"
+            
+        return saida_valor if saida_valor.strip() else "[SEM SAÍDA]"
+
+    except Exception:
+        return f"[EXCEÇÃO DURANTE EXECUÇÃO]:\n{traceback.format_exc()}"
+        
     #with stdoutIO() as s:
     #    exec(python_code)