Update app.py

app.py

@@ -33,7 +33,7 @@ class BasicAgent:
         return answer[14:]
 
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all(nb_questions: int, profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -87,6 +87,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
+
+    questions_data = questions_data[:nb_questions]
+    
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
@@ -176,12 +179,15 @@ with gr.Blocks() as demo:
 
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
+    nb_questions = gr.Number(value=3)
+
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     # Removed max_rows=10 from DataFrame constructor
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
     run_button.click(
         fn=run_and_submit_all,
+        inputs=[nb_questions],
         outputs=[status_output, results_table]
     )