Updated run functions

app.py

@@ -31,18 +31,9 @@ tracer = trace.get_tracer(__name__)
 # Instrument smolagents with the configured provider
 SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
 
-def user_logged_in(profile: gr.OAuthProfile):
-    if profile:
-        username= f"{profile.username}"
-        print(f"User logged in: {username}")
-        return True
-    else:
-        print("User not logged in.")
-        return False
-        
 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):
     minutes = int(elapsed_time // 60)  # Get the whole number of minutes
     seconds = elapsed_time % 60  # Get the remaining seconds
@@ -60,24 +51,22 @@ def _run(questions: list, username: str) -> pd.DataFrame:
     return message, question_answer_pairs
     
 def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
-    if not user_logged_in(profile): 
+    if profile: 
+        return _run([evaluator.get_one_question()], profile.username)
+    else:
         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): 
+    if profile: 
+        return _run(evaluator.get_questions(), profile.username)
+    else:
         return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
-    questions = evaluator.get_questions()
-    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)}"
-    return message, question_answer_pairs
 
 def submit(profile: gr.OAuthProfile | None) -> str:
-    if not user_logged_in(profile): 
+    if profile: 
+        return evaluator.submit_answers(profile.username)
+    else:
         return LOGIN_MESSAGE
-    return evaluator.submit_answers(profile.username)
 
 
 # --- Build Gradio Interface using Blocks ---