Update app.py

app.py

@@ -1,9 +1,8 @@
 """
 app.py
 This script provides the Gradio web interface to run the evaluation.
-## MODIFICATION: This version is simplified to work with the new agent architecture.
-It no longer performs file-type detection or prompt enhancement, as that responsibility
-has been moved into the agent's 'multimodal_router'.
+This version is simplified to work with the new agent architecture and has
+the correct Gradio wiring for the Hugging Face login functionality.
 """
 
 import os
@@ -18,7 +17,7 @@ from agent import create_agent_executor
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Helper function to parse the agent's output (remains the same) ---
+# --- Helper function to parse the agent's output ---
 def parse_final_answer(agent_response: str) -> str:
     match = re.search(r"FINAL ANSWER:\s*(.*)", agent_response, re.IGNORECASE | re.DOTALL)
     if match: return match.group(1).strip()
@@ -26,13 +25,6 @@ def parse_final_answer(agent_response: str) -> str:
     if lines: return lines[-1].strip()
     return "Could not parse a final answer."
 
-## MODIFICATION: The `detect_file_type` function has been removed.
-## It is now redundant as this logic is handled inside the agent.
-
-## MODIFICATION: The `create_enhanced_prompt` function has been removed.
-## It was causing errors by trying to instruct the agent to use tools that no longer exist.
-## The agent is now responsible for handling the raw input itself.
-
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the agent on them, submits all answers,
@@ -80,8 +72,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         
         file_url = item.get("file_url")
         
-        ## MODIFICATION: Prompt creation is now much simpler.
-        # We just combine the question and the URL into one string.
+        # We simply combine the question and the URL into one string.
         # The agent's multimodal_router will handle the rest.
         if file_url:
             full_question_text = f"{question_text}\n\nHere is the relevant file: {file_url}"
@@ -92,9 +83,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print(f"Raw Prompt for Agent:\n{full_question_text}")
 
         try:
-            # Pass the simple, raw question to the agent
             result = agent_executor.invoke({"messages": [("user", full_question_text)]})
-            
             raw_answer = result['messages'][-1].content
             submitted_answer = parse_final_answer(raw_answer)
             
@@ -102,6 +91,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"PARSED FINAL ANSWER: '{submitted_answer}'")
 
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+            # The log for the DataFrame no longer includes a 'File Type' column
             results_log.append({
                 "Task ID": task_id, 
                 "Question": question_text, 
@@ -123,7 +113,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not answers_payload:
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare and 5. Submit (remains the same)
+    # 4. Prepare and 5. Submit
     submission_data = {"username": username, "agent_code": agent_code, "answers": answers_payload}
     print(f"\nSubmitting {len(answers_payload)} answers for user '{username}'...")
     try:
@@ -139,30 +129,30 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print(status_message)
         return status_message, pd.DataFrame(results_log)
 
-# --- Gradio UI (remains the same) ---
+# --- Gradio UI ---
 with gr.Blocks(title="Multimodal Agent Evaluation") as demo:
     gr.Markdown("# Multimodal Agent Evaluation Runner")
     gr.Markdown("This agent can process images, YouTube videos, audio files, and perform web searches.")
     
-    gr.LoginButton()
+    ## MODIFICATION: Assign the LoginButton to a variable so we can use it as an input.
+    login_button = gr.LoginButton()
+    
     run_button = gr.Button("Run Evaluation & Submit All Answers", variant="primary")
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=6, interactive=False)
     results_table = gr.DataFrame(
         label="Questions and Agent Answers", 
         wrap=True, 
         row_count=10,
-        # MODIFICATION: Removed the 'File Type' column as it's no longer detected here.
-        column_widths=[80, 250, 200, 250] 
+        column_widths=[80, 250, 200, 250]
     )
     
-    # We also remove "File Type" from the results_log being displayed
-    def display_wrapper(profile):
-        status, df = run_and_submit_all(profile)
-        if df is not None and "File Type" in df.columns:
-            df = df.drop(columns=["File Type"])
-        return status, df
-
-    run_button.click(fn=display_wrapper, outputs=[status_output, results_table])
+    ## MODIFICATION: Wire the login_button as an input to the click event.
+    # This correctly passes the OAuth profile to the run_and_submit_all function.
+    run_button.click(
+        fn=run_and_submit_all, 
+        inputs=login_button, 
+        outputs=[status_output, results_table]
+    )
 
 if __name__ == "__main__":
     print("\n" + "-"*30 + " Multimodal App Starting " + "-"*30)