Update app.py

app.py

@@ -4,7 +4,6 @@ from crew import run_crew
 from utils import get_questions
 
 QUESTION_FILE_PATH = "data/gaia_validation.jsonl"
-QUESTION_LEVEL     = 0
 
 def _run(question, openai_api_key, gemini_api_key, file_name = ""):
     """
@@ -51,7 +50,9 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
 
 gr.close_all()
 
-examples = get_questions(QUESTION_FILE_PATH, QUESTION_LEVEL)
+examples_1 = get_questions(QUESTION_FILE_PATH, 1)
+examples_2 = get_questions(QUESTION_FILE_PATH, 2)
+examples_3 = get_questions(QUESTION_FILE_PATH, 3)
 
 with gr.Blocks() as gaia:
     gr.Markdown("## General AI Assistant 🤖🤝🤖")
@@ -114,7 +115,21 @@ with gr.Blocks() as gaia:
     )
     
     gr.Examples(
-        examples=examples,
+        examples=examples_1,
+        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
+        outputs=answer,
+        cache_examples=False
+    )
+
+    gr.Examples(
+        examples=examples_2,
+        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
+        outputs=answer,
+        cache_examples=False
+    )
+
+    gr.Examples(
+        examples=examples_3,
         inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
         outputs=answer,
         cache_examples=False