Update app.py

app.py

@@ -2,17 +2,16 @@
 
 # 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,
     get_dataset
 )
-import sys
-from io import StringIO
-from contextlib import contextmanager
 
 # Configuration
 
@@ -24,9 +23,9 @@ BASE_URL = f"https://huggingface.co/spaces/{SPACE_ID}/blob/main"
 
 @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()
@@ -81,14 +80,12 @@ 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
 
-        # 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 "")
+        console_logs = stdout_logs + (stderr_logs if stderr_logs else "")
         
         return answer, console_logs
     except Exception as e:
@@ -117,9 +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"Processing can take several minutes depending on the question complexity. "
-    f"Console logs are provided belowfor transparency.</p>"
-    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>"
 )
 
@@ -220,17 +217,15 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
                     lines=1,
                     max_lines=1
                 )
-        
-        # Console logs section
-        with gr.Accordion("Console Logs & Tool Calls", open=False):
+
+        with gr.Accordion("Console Logs", open=False):
             console_logs = gr.Textbox(
-                label="Console Output",
+                label="Output",
                 interactive=False,
                 lines=25,
                 max_lines=25
             )
         
-        # 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]
@@ -253,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
                 )
 
@@ -261,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
                 )
 
@@ -269,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
                 )
 
@@ -277,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
                 )