Update app.py

app.py

@@ -15,7 +15,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ON A SUPPRIMÉ L'ANCIENNE CLASSE BasicAgent D'ICI.
 # C'EST MAINTENANT GÉRÉ PAR L'IMPORT CI-DESSUS.
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -138,7 +138,7 @@ with gr.Blocks() as demo:
         """
     )
 
-    # On assigne le bouton de connexion à une variable
+    # Create the login button
     login_button = gr.LoginButton()
 
     run_button = gr.Button("Run Evaluation & Submit All Answers")
@@ -146,10 +146,17 @@ with gr.Blocks() as demo:
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
-    # On utilise la variable 'login_button' comme 'inputs' pour le clic
+    # Create a state to store the profile
+    profile_state = gr.State()
+    
+    # Update the profile state when login status changes
+    login_button.login(lambda profile: profile, outputs=profile_state)
+    login_button.logout(lambda: None, outputs=profile_state)
+
+    # Use the profile state as input
     run_button.click(
         fn=run_and_submit_all,
-        inputs=[login_button],
+        inputs=profile_state,
         outputs=[status_output, results_table]
     )