Update app.py

app.py

@@ -20,7 +20,7 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all( profile: gr.OAuthProfile | None, limit: int | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -73,6 +73,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # 3. Run your Agent
     results_log = []
     answers_payload = []
+    # Limit for test purposes
+    if limit is not None:
+        question_data = question_data[:limit]
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
         task_id = item.get("task_id")
@@ -168,7 +171,7 @@ with gr.Blocks() as demo:
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
     run_button.click(
-        fn=run_and_submit_all,
+        fn=run_and_submit_all(limit=3),
         outputs=[status_output, results_table]
     )