Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from crew import run_crew
 from utils import get_questions
 
-#QUESTION_FILE_PATH = "data/gaia_validation.jsonl"
+QUESTION_FILE_PATH = "data/gaia_validation.jsonl"
 
 def _run(question, openai_api_key, gemini_api_key, file_name = ""):
     """
@@ -50,12 +50,6 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
 
 gr.close_all()
 
-#examples_1 = get_questions(QUESTION_FILE_PATH, 1)
-#examples_2 = get_questions(QUESTION_FILE_PATH, 2)
-#examples_3 = get_questions(QUESTION_FILE_PATH, 3)
-
-examples_1, examples_2, examples_3 = get_questions()
-
 with gr.Blocks() as gaia:
     gr.Markdown("## General AI Assistant 🧠")
     gr.Markdown(os.environ.get("DESCRIPTION"))
@@ -114,7 +108,7 @@ with gr.Blocks() as gaia:
     
     gr.Examples(
         label="Level 1",
-        examples=examples_1,
+        examples=get_questions(QUESTION_FILE_PATH, 1),
         inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
         outputs=answer,
         cache_examples=False
@@ -122,7 +116,7 @@ with gr.Blocks() as gaia:
 
     gr.Examples(
         label="Level 2",
-        examples=examples_2,
+        examples=get_questions(QUESTION_FILE_PATH, 2),
         inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
         outputs=answer,
         cache_examples=False
@@ -130,7 +124,7 @@ with gr.Blocks() as gaia:
 
     gr.Examples(
         label="Level 3",
-        examples=examples_3,
+        examples=get_questions(QUESTION_FILE_PATH, 3),
         inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
         outputs=answer,
         cache_examples=False