Update app.py

app.py

@@ -172,10 +172,10 @@ class BasicAgent:
         return self.format_final_answer("COULD_NOT_SOLVE")
 
 # --- Evaluation Logic ---
-def run_and_submit_all(profile, test_mode):
+def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if profile:
-        username = profile
+        username = profile.username
         print(f"User logged in: {username}")
     else:
         return "Please Login to Hugging Face with the button.", None
@@ -249,21 +249,11 @@ with gr.Blocks() as demo:
         3. Click the button to run evaluation and submit your answers.
         """
     )
-    
-    # Simulate OAuth profile with a textbox for user
-    test_checkbox = gr.Checkbox(label="Enable Test Mode (Skip Submission)", value=False)
-    run_button = gr.Button("Run Evaluation")
+    gr.LoginButton()
+    run_button = gr.Button("Run Evaluation & Submit All Answers")
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
-
-    # Simulate OAuth Profile with a mock profile for now
-    mock_oauth_profile = gr.Textbox(label="Simulated OAuth Profile", value="mock_user", interactive=False)
-    
-    run_button.click(
-        fn=run_and_submit_all,
-        inputs=[mock_oauth_profile, test_checkbox],
-        outputs=[status_output, results_table]
-    )
+    run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 if __name__ == "__main__":
     print("Launching Gradio Interface...")