Update app.py

app.py

@@ -1,212 +1,113 @@
-# References:
-
-# https://www.gradio.app/guides/quickstart
-
-import os
-import gradio as gr
-from agents.crew import run_crew
-from utils.utils import get_questions
-
-# Initialization
-
-QUESTION_FILE_PATH_GAIA = "files/gaia_validation.jsonl"
-QUESTION_FILE_PATH_HLE  = "files/hle_validation.jsonl"
-
-DEFAULT_QUESTION = "In MCP's 1st Birthday Hackathon, hosted by Anthropic and Gradio, what percentage of participants submitted a solution so far?"
-
-# MCP server functions
-
-def ask(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name = ""):
-    """
-    Ask General AI Assistant a question to answer.
-
-    Args:
-        question (str): The question to answer
-        openai_api_key (str): OpenAI API key
-        gemini_api_key (str): Gemini API key
-        anthropic_api_key (str): Anthropic API key
-        file_name (str): Optional file name
-
-    Returns:
-        str: The answer to the question
-    """
-    if not question:
-        gr.Warning("Question is required.")
-        return ""
-
-    if not openai_api_key:
-        gr.Warning("OpenAI API Key is required.")
-        return ""
-    
-    if not gemini_api_key:
-        gr.Warning("Gemini API Key is required.")
-        return ""
-
-    if not anthropic_api_key:
-        gr.Warning("Anthropic API Key is required.")
-        return ""
-        
-    if file_name:
-        file_name = f"files/{file_name}"
-    
-    try:
-        os.environ["OPENAI_API_KEY"] = openai_api_key
-        os.environ["GEMINI_API_KEY"] = gemini_api_key
-        os.environ["MODEL_API_KEY"] = anthropic_api_key
-        
-        return run_crew(question, file_name)
-    except Exception as e:
-        gr.Warning(str(e))
-        return ""
-    finally:
-        for key in ["OPENAI_API_KEY", "GEMINI_API_KEY", "MODEL_API_KEY"]:
-            os.environ.pop(key, None)
-
-# Helper functions
-
-def update_file_link(file_name):
-    if file_name:
-        space_id = os.environ.get("SPACE_ID", "bstraehle/gaia")
-        base_url = f"https://huggingface.co/spaces/{space_id}/blob/main/files"
-        return f'<a href="{base_url}/{file_name}" target="_blank">Open File</a>'
-    
-    return ""
-
-# Graphical user interface
-
-CSS_FULL_WIDTH = """
-body {
-    margin: 0 !important;
-}
-.gradio-container {
-    max-width: 100% !important;
-    width: 100% !important;
-    margin: 0 !important;
-    padding-left: 0 !important;
-    padding-right: 0 !important;
-}
-.content-padding {
-    padding-left: 1.5rem;
-    padding-right: 1.5rem;
-}
-.full-width-tabs {
-    width: 100%;
-}
-.full-width-tabs .gradio-block {
-    max-width: 100%;
-    width: 100%;
-}
-"""
-
 with gr.Blocks() as gaia:
     gr.HTML(f"<style>{CSS_FULL_WIDTH}</style>")
-    
-    gr.Markdown("## General AI Assistant")
 
-    gr.Markdown(os.environ.get("DESCRIPTION"))
-    
-    with gr.Row():
-        with gr.Column(scale=3):
-            with gr.Row():
-                question = gr.Textbox(
-                    label="Question *",
-                    placeholder=DEFAULT_QUESTION,
-                    interactive=True,
+    with gr.Column(elem_classes=["content-padding"]):
+        gr.Markdown("## General AI Assistant")
+
+        gr.Markdown(os.environ.get("DESCRIPTION"))
+        
+        with gr.Row():
+            with gr.Column(scale=3):
+                with gr.Row():
+                    question = gr.Textbox(
+                        label="Question *",
+                        placeholder=DEFAULT_QUESTION,
+                        interactive=True,
+                        lines=1,
+                        max_lines=3
+                    )
+                with gr.Row():
+                    ground_truth = gr.Textbox(
+                        label="Ground Truth",
+                        interactive=True,
+                        lines=1,
+                        max_lines=2
+                    )
+                    file_name = gr.Textbox(
+                        label="File Name",
+                        interactive=True,
+                        scale=2,
+                        lines=1,
+                        max_lines=2
+                    )
+                    file_link = gr.HTML(
+                        label="File Link",
+                        value=""
+                    )
+                with gr.Row():
+                    openai_api_key = gr.Textbox(
+                        label="OpenAI API Key *",
+                        type="password",
+                        placeholder="sk‑...",
+                        interactive=True
+                    )
+                    gemini_api_key = gr.Textbox(
+                        label="Gemini API Key *",
+                        type="password",
+                        interactive=True
+                    )
+                    anthropic_api_key = gr.Textbox(
+                        label="Anthropic API Key *",
+                        type="password",
+                        placeholder="sk-ant-...",
+                        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=1,
                     max_lines=3
                 )
-            with gr.Row():
-                ground_truth = gr.Textbox(
-                    label="Ground Truth",
-                    interactive=True,
-                    lines=1,
-                    max_lines=2
-                )
-                file_name = gr.Textbox(
-                    label="File Name",
-                    interactive=True,
-                    scale=2,
-                    lines=1,
-                    max_lines=2
-                )
-                file_link = gr.HTML(
-                    label="File Link",
-                    value=""
-                )
-            with gr.Row():
-                openai_api_key = gr.Textbox(
-                    label="OpenAI API Key *",
-                    type="password",
-                    placeholder="sk‑...",
-                    interactive=True
-                )
-                gemini_api_key = gr.Textbox(
-                    label="Gemini API Key *",
-                    type="password",
-                    interactive=True
-                )
-                anthropic_api_key = gr.Textbox(
-                    label="Anthropic API Key *",
-                    type="password",
-                    placeholder="sk-ant-...",
-                    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=1,
-                max_lines=3
-            )
-
-    submit_btn.click(
-        fn=ask,
-        inputs=[question, openai_api_key, gemini_api_key, anthropic_api_key, file_name],
-        outputs=answer
-    )
-
-    file_name.change(
-        fn=update_file_link,
-        inputs=[file_name],
-        outputs=[file_link]
-    )
 
-    with gr.Tabs():
-        with gr.TabItem("GAIA Benchmark Level 1"):
-            gr.Examples(
-                examples=get_questions(QUESTION_FILE_PATH_GAIA, 1),
-                inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
-                examples_per_page=3,
-                cache_examples=False
-            )
+        submit_btn.click(
+            fn=ask,
+            inputs=[question, openai_api_key, gemini_api_key, anthropic_api_key, file_name],
+            outputs=answer
+        )
+
+        file_name.change(
+            fn=update_file_link,
+            inputs=[file_name],
+            outputs=[file_link]
+        )
+
+        with gr.Tabs(elem_classes=["full-width-tabs"]):
+            with gr.TabItem("GAIA Benchmark Level 1"):
+                gr.Examples(
+                    examples=get_questions(QUESTION_FILE_PATH_GAIA, 1),
+                    inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
+                    examples_per_page=3,
+                    cache_examples=False
+                )
 
-        with gr.TabItem("GAIA Benchmark Level 2"):
-            gr.Examples(
-                examples=get_questions(QUESTION_FILE_PATH_GAIA, 2),
-                inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
-                examples_per_page=3,
-                cache_examples=False
-            )
+            with gr.TabItem("GAIA Benchmark Level 2"):
+                gr.Examples(
+                    examples=get_questions(QUESTION_FILE_PATH_GAIA, 2),
+                    inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
+                    examples_per_page=3,
+                    cache_examples=False
+                )
 
-        with gr.TabItem("GAIA Benchmark Level 3"):
-            gr.Examples(
-                examples=get_questions(QUESTION_FILE_PATH_GAIA, 3),
-                inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
-                examples_per_page=3,
-                cache_examples=False
-            )
+            with gr.TabItem("GAIA Benchmark Level 3"):
+                gr.Examples(
+                    examples=get_questions(QUESTION_FILE_PATH_GAIA, 3),
+                    inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
+                    examples_per_page=3,
+                    cache_examples=False
+                )
 
-        with gr.TabItem("Humanity's Last Exam"):
-            gr.Examples(
-                examples=get_questions(QUESTION_FILE_PATH_HLE, 0),
-                inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
-                examples_per_page=3,
-                cache_examples=False
-            )
+            with gr.TabItem("Humanity's Last Exam"):
+                gr.Examples(
+                    examples=get_questions(QUESTION_FILE_PATH_HLE, 0),
+                    inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
+                    examples_per_page=3,
+                    cache_examples=False
+                )
 
 gaia.launch(mcp_server=True, share=False, ssr_mode=False)