Update app.py

app.py

@@ -2,9 +2,11 @@
 
 # https://www.gradio.app/guides/quickstart
 
-import gradio.utils, os
+import gradio.utils, os, sys
 import gradio as gr
 from agents.crew import run_crew
+from contextlib import contextmanager
+from io import StringIO
 from utils.utils import (
     DATASET_TYPE_GAIA,
     DATASET_TYPE_HLE,
@@ -17,6 +19,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():
+    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 +48,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 +80,18 @@ 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)
+        with capture_output() as (stdout, stderr):
+            answer = run_crew(question, file_name)
+            
+        stdout_logs = stdout.getvalue()
+        stderr_logs = stderr.getvalue()
+        console_logs = stdout_logs + (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
 
@@ -89,7 +114,9 @@ DESCRIPTION = (
     f"The system can solve multiple <a href='https://arxiv.org/pdf/2311.12983'>GAIA Benchmark</a> "
     f"Level 1, 2, 3 and even <a href='https://arxiv.org/pdf/2501.14249'>Humanity's Last Exam</a> "
     f"problems. To get started, select from the examples below. "
-    f"API keys are provided thanks to sponsor credit. "
+    f"Processing can take minutes depending on question complexity. "
+    f"Console logs are provided below for transparency. "
+    f"API keys are provided temporarily thanks to sponsor credit. "
     f"<a href='{BASE_URL}/README.md'>Documentation</a></p>"
 )
 
@@ -138,22 +165,22 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                         label="Question *",
                         value=DEFAULT_QUESTION,
                         interactive=True,
-                        lines=2,
-                        max_lines=5
+                        lines=1,
+                        max_lines=1
                     )
                 with gr.Row():
                     ground_truth = gr.Textbox(
                         label="Ground Truth",
                         value=DEFAULT_GROUND_TRUTH,
                         interactive=True,
-                        lines=2,
-                        max_lines=2
+                        lines=1,
+                        max_lines=1
                     )
                     file_name = gr.Textbox(
                         label="File Name",
                         interactive=True,
-                        lines=2,
-                        max_lines=2,
+                        lines=1,
+                        max_lines=1,
                         scale=2
                     )
                     file_link = gr.HTML(
@@ -182,22 +209,32 @@ 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(
                     label="Answer",
                     interactive=False,
-                    lines=2,
-                    max_lines=5
+                    lines=1,
+                    max_lines=1
                 )
 
+        with gr.Accordion("Console Logs", open=False):
+            console_logs = gr.Textbox(
+                label="Output",
+                interactive=False,
+                lines=25,
+                max_lines=25
+            )
+        
+        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(
@@ -211,7 +248,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                 gr.Examples(
                     examples=get_dataset(DATASET_TYPE_GAIA, 1),
                     inputs=[question, ground_truth, file_name, "", "", ""],
-                    examples_per_page=3,
+                    examples_per_page=5,
                     cache_examples=False
                 )
 
@@ -219,7 +256,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                 gr.Examples(
                     examples=get_dataset(DATASET_TYPE_GAIA, 2),
                     inputs=[question, ground_truth, file_name, "", "", ""],
-                    examples_per_page=3,
+                    examples_per_page=5,
                     cache_examples=False
                 )
 
@@ -227,7 +264,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                 gr.Examples(
                     examples=get_dataset(DATASET_TYPE_GAIA, 3),
                     inputs=[question, ground_truth, file_name, "", "", ""],
-                    examples_per_page=3,
+                    examples_per_page=5,
                     cache_examples=False
                 )
 
@@ -235,7 +272,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                 gr.Examples(
                     examples=get_dataset(DATASET_TYPE_HLE, 0),
                     inputs=[question, ground_truth, file_name, "", "", ""],
-                    examples_per_page=3,
+                    examples_per_page=5,
                     cache_examples=False
                 )