Update app.py

app.py

@@ -29,9 +29,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
-        print("Initializing Advanced Agent logic within BasicAgent class...")
+        print("Initializing Advanced Agent logic...")
         try:
-            # Using hardcoded values for easier debugging, only API Key is from secrets
             self.llm = AzureChatOpenAI(
                 azure_endpoint="https://dsap.openai.azure.com/",
                 api_key=os.environ["AZURE_API_KEY"],
@@ -41,12 +40,11 @@ class BasicAgent:
                 max_retries=3,
             )
             print("AzureChatOpenAI client initialized successfully.")
-        except KeyError as e:
-            raise KeyError(f"CRITICAL: Missing environment variable 'AZURE_API_KEY'. Please add it to your Hugging Face Space secrets.")
+        except KeyError:
+            raise KeyError("CRITICAL: Missing 'AZURE_API_KEY'. Please add it to your Space secrets.")
         except Exception as e:
             print(f"FAILED to initialize Azure client. Error: {e}")
             raise
-
         self.tools = {
             "search": self.search, "browse": self.browse,
             "python": self.python_interpreter, "get_youtube_transcript": self.get_youtube_transcript,
@@ -148,18 +146,15 @@ Thought:
         return "Agent failed to find an answer within the loop limit."
 
 
-# *** THE FIX IS HERE ***
-def run_and_submit_all(request: gr.Request):
-    """
-    This function now accepts a gr.Request object to safely get user profile info.
-    """
+# The function signature is changed to accept the profile object directly
+def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
-    # Get username from the request object
-    if request and request.username:
-        username = request.username
+    if profile and profile.username:
+        username = profile.username
         print(f"User logged in: {username}")
     else:
+        # This is the message you were seeing.
         return "Could not get user profile. Please log in using the button above.", None
 
     api_url, submit_url = DEFAULT_API_URL, f"{DEFAULT_API_URL}/submit"
@@ -169,7 +164,6 @@ def run_and_submit_all(request: gr.Request):
     except Exception as e: return f"Error initializing agent: {e}", None
 
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "https://github.com/huggingface/agents-course"
-    print(f"Agent code URL: {agent_code}")
 
     try:
         response = requests.get(questions_url, timeout=20)
@@ -204,24 +198,35 @@ def run_and_submit_all(request: gr.Request):
                         f"Overall Score: {result_data.get('score', 'N/A')}% "
                         f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)")
         return final_status, pd.DataFrame(results_log)
-    except requests.exceptions.HTTPError as e:
-        error_detail = f"Server responded with {e.response.status_code}. Detail: {e.response.text[:500]}"
-        return f"Submission Failed: {error_detail}", pd.DataFrame(results_log)
     except Exception as e: return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
 
 # --- Build Gradio Interface ---
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Agent Evaluation Runner")
-    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)
 
-    # *** AND THE FIX IS HERE ***
-    # The click event now passes the request object implicitly to the function.
+    # This state object will hold the user profile. It's invisible.
+    profile_state = gr.State(value=None)
+    
+    with gr.Row():
+        # The login button is now just for show and triggering the login flow.
+        gr.LoginButton()
+
+    with gr.Row():
+        run_button = gr.Button("Run Evaluation & Submit All Answers", variant="primary")
+
+    with gr.Column():
+        status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
+        results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
+
+    # *** THE NEW, ROBUST FIX IS HERE ***
+    # 1. When the page loads, get the user's auth info and store it in our invisible state object.
+    demo.load(lambda request: request.auth, inputs=[], outputs=[profile_state])
+
+    # 2. When the run button is clicked, take the stored profile from the state object as input.
     run_button.click(
         fn=run_and_submit_all,
+        inputs=[profile_state],
         outputs=[status_output, results_table]
     )