Updated run functions

app.py

@@ -43,7 +43,7 @@ def user_logged_in(profile: gr.OAuthProfile):
 LOGIN_MESSAGE = "Please Login to Hugging Face with the button."
 EMPTY_RESULTS_TABLE = pd.DataFrame(columns=['task_id', 'question', 'answer'])
 
-def format_elapsed_time(elapsed_time):
+def _format_elapsed_time(elapsed_time):
     minutes = int(elapsed_time // 60)  # Get the whole number of minutes
     seconds = elapsed_time % 60  # Get the remaining seconds
 
@@ -51,16 +51,18 @@ def format_elapsed_time(elapsed_time):
         return f"Elapsed time: {minutes} minutes {seconds:.2f} seconds"
     else:
         return f"Elapsed time: {seconds:.2f} seconds"
-
-def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
-    if not user_logged_in(profile): 
-        return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
+    
+def _run(questions: list, username: str) -> pd.DataFrame:
     start_time = time.time()
-    questions = [evaluator.get_one_question()]
-    question_answer_pairs = runner.run_agent(questions, profile.username)
+    question_answer_pairs = runner.run_agent(questions, username)
     end_time = time.time()
-    message = f"Complete. {format_elapsed_time(end_time - start_time)}"
+    message = f"Complete. {_format_elapsed_time(end_time - start_time)}"
     return message, question_answer_pairs
+    
+def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
+    if not user_logged_in(profile): 
+        return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
+    return _run([evaluator.get_one_question()], profile.username)
 
 def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
     if not user_logged_in(profile): 
@@ -69,7 +71,7 @@ def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
     start_time = time.time()
     question_answer_pairs = runner.run_agent(questions, profile.username)
     end_time = time.time()
-    message = f"Complete. {format_elapsed_time(end_time - start_time)}"
+    message = f"Complete. {_format_elapsed_time(end_time - start_time)}"
     return message, question_answer_pairs
 
 def submit(profile: gr.OAuthProfile | None) -> str: