Update app.py

app.py

@@ -6,9 +6,10 @@ import gradio.utils, os
 import gradio as gr
 from agents.crew import run_crew
 from utils.utils import (
-    QUESTION_TYPE_GAIA,
-    QUESTION_TYPE_HLE,
-    get_questions_from_dataset
+    DATASET_TYPE_GAIA,
+    DATASET_TYPE_HLE,
+    get_dataset_from_file,
+    get_dataset
 )
 
 # MCP server functions
@@ -184,7 +185,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
         with gr.Tabs():
             with gr.TabItem("GAIA Benchmark Level 1"):
                 gr.Examples(
-                    examples=get_questions_from_dataset(QUESTION_TYPE_GAIA, 1),
+                    examples=get_dataset_from_file(QUESTION_TYPE_GAIA, 1),
                     inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
                     examples_per_page=3,
                     cache_examples=False
@@ -192,7 +193,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
 
             with gr.TabItem("GAIA Benchmark Level 2"):
                 gr.Examples(
-                    examples=get_questions_from_dataset(QUESTION_TYPE_GAIA, 2),
+                    examples=get_dataset_from_file(QUESTION_TYPE_GAIA, 2),
                     inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
                     examples_per_page=3,
                     cache_examples=False
@@ -200,7 +201,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
 
             with gr.TabItem("GAIA Benchmark Level 3"):
                 gr.Examples(
-                    examples=get_questions_from_dataset(QUESTION_TYPE_GAIA, 3),
+                    examples=get_dataset_from_file(QUESTION_TYPE_GAIA, 3),
                     inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
                     examples_per_page=3,
                     cache_examples=False
@@ -208,7 +209,7 @@ with gr.Blocks(elem_classes=["full-width-app"]) as gaia:
 
             with gr.TabItem("Humanity's Last Exam"):
                 gr.Examples(
-                    examples=get_questions_from_dataset(QUESTION_TYPE_HLE, 0),
+                    examples=get_dataset_from_file(QUESTION_TYPE_HLE, 0),
                     inputs=[question, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
                     examples_per_page=3,
                     cache_examples=False