Update app.py

app.py

@@ -10,6 +10,9 @@ from utils.utils import (
     DATASET_TYPE_HLE,
     get_dataset
 )
+import sys
+from io import StringIO
+from contextlib import contextmanager
 
 # Configuration
 
@@ -17,6 +20,21 @@ SPACE_ID = os.environ.get("SPACE_ID")
 
 BASE_URL = f"https://huggingface.co/spaces/{SPACE_ID}/blob/main"
 
+# Console output capture
+
+@contextmanager
+def capture_output():
+    """Context manager to capture stdout and stderr"""
+    old_stdout = sys.stdout
+    old_stderr = sys.stderr
+    try:
+        sys.stdout = StringIO()
+        sys.stderr = StringIO()
+        yield sys.stdout, sys.stderr
+    finally:
+        sys.stdout = old_stdout
+        sys.stderr = old_stderr
+
 # MCP server functions
 
 def ask(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name = ""):
@@ -31,23 +49,23 @@ def ask(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name =
         file_name (str): Optional file name
 
     Returns:
-        str: The answer to the question
+        tuple: (answer, console_logs) - The answer to the question and captured console output
     """
     if not question:
         gr.Warning("Question is required.")
-        return ""
+        return "", ""
 
     if not openai_api_key:
         gr.Warning("OpenAI API Key is required.")
-        return ""
+        return "", ""
     
     if not gemini_api_key:
         gr.Warning("Gemini API Key is required.")
-        return ""
+        return "", ""
 
     if not anthropic_api_key:
         gr.Warning("Anthropic API Key is required.")
-        return ""
+        return "", ""
         
     if file_name:
         file_name = f"files/{file_name}"
@@ -63,10 +81,20 @@ def ask(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name =
         os.environ["GEMINI_API_KEY"] = gemini_key
         os.environ["ANTHROPIC_API_KEY"] = anthropic_key
 
-        return run_crew(question, file_name)
+        # Capture console output
+        with capture_output() as (stdout, stderr):
+            answer = run_crew(question, file_name)
+            
+        # Get captured logs
+        stdout_logs = stdout.getvalue()
+        stderr_logs = stderr.getvalue()
+        console_logs = stdout_logs + ("\n" + stderr_logs if stderr_logs else "")
+        
+        return answer, console_logs
     except Exception as e:
-        gr.Warning(str(e))
-        return ""
+        error_msg = str(e)
+        gr.Warning(error_msg)
+        return "", f"Error: {error_msg}"
 
 # Helper functions
 
@@ -183,9 +211,6 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                         interactive=True
                     )
                 with gr.Row():
-                    clear_btn = gr.ClearButton(
-                        components=[question, ground_truth, file_name, file_link]
-                    )
                     submit_btn = gr.Button("Submit", variant="primary")
             with gr.Column(scale=1):
                 answer = gr.Textbox(
@@ -194,11 +219,27 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                     lines=2,
                     max_lines=5
                 )
+        
+        # Console logs section
+        with gr.Accordion("Console Logs & Tool Calls", open=False):
+            console_logs = gr.Textbox(
+                label="Console Output",
+                interactive=False,
+                lines=15,
+                max_lines=30,
+                show_copy_button=True
+            )
+        
+        # Clear button (after all components are defined)
+        with gr.Row():
+            clear_btn = gr.ClearButton(
+                components=[question, ground_truth, file_name, file_link, answer, console_logs]
+            )
 
         submit_btn.click(
             fn=ask,
             inputs=[question, openai_api_key, gemini_api_key, anthropic_api_key, file_name],
-            outputs=answer
+            outputs=[answer, console_logs]
         )
 
         file_name.change(