Update app.py

app.py

@@ -25,7 +25,7 @@ class BasicAgent:
                 return "Berlin"
             elif "italy" in question_lower:
                 return "Rome"
-            elif "spain" in question_lower:
+            elif "spain" in question_lower: 
                 return "Madrid"
             elif "japan" in question_lower:
                 return "Tokyo"
@@ -33,26 +33,26 @@ class BasicAgent:
         if "who" in question_lower and "ceo" in question_lower: 
             if "tesla" in question_lower or "spacex" in question_lower: 
                 return "Elon Musk"
-            elif "apple" in question_lower:
+            elif "apple" in question_lower: 
                 return "Tim Cook"
-            elif "microsoft" in question_lower: 
+            elif "microsoft" in question_lower:
                 return "Satya Nadella"
         
-        if "what year" in question_lower or "when" in question_lower: 
-            if "world war 2" in question_lower or "wwii" in question_lower: 
+        if "what year" in question_lower or "when" in question_lower:
+            if "world war 2" in question_lower or "wwii" in question_lower:
                 return "1939-1945"
-            elif "world war 1" in question_lower or "wwi" in question_lower:
+            elif "world war 1" in question_lower or "wwi" in question_lower: 
                 return "1914-1918"
         
-        if "how many" in question_lower:
-            if "planets" in question_lower:
+        if "how many" in question_lower: 
+            if "planets" in question_lower: 
                 return "8"
             elif "continents" in question_lower: 
                 return "7"
         
         # Default response
         answer = "I don't have enough information to answer this question accurately."
-        print(f"Agent returning answer:  {answer}")
+        print(f"Agent returning answer: {answer}")
         return answer
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -80,8 +80,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
 
+    # Build agent code URL
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
-    print(f"Agent code: {agent_code}")
+    print(f"Agent code:  {agent_code}")
 
     # 2. Fetch Questions
     print(f"Fetching questions from:  {questions_url}")
@@ -98,7 +99,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"Error fetching questions: {e}", None
     except requests.exceptions.JSONDecodeError as e:
         print(f"Error decoding JSON response from questions endpoint: {e}")
-        print(f"Response text: {response.text[:500]}")
+        print(f"Response text: {response.text[: 500]}")
         return f"Error decoding server response for questions: {e}", None
     except Exception as e:
         print(f"An unexpected error occurred fetching questions: {e}")
@@ -108,8 +109,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data: 
-        task_id = item. get("task_id")
+    for item in questions_data:
+        task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question:  {item}")
@@ -124,30 +125,56 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             answers_payload.append({"task_id": task_id, "answer": f"Error: {e}"})
             results_log.append((task_id, question_text[:50], f"Error: {e}"))
 
-    # 4. Submit answers
+    # 4. Submit answers with CORRECT payload format
     print(f"Submitting {len(answers_payload)} answers to {submit_url}...")
     try:
+        # THIS IS THE CORRECT PAYLOAD FORMAT!
+        payload = {
+            "username": username,
+            "answers": answers_payload,
+            "agent_code": agent_code
+        }
+        
+        print(f"Payload structure: username={username}, answers count={len(answers_payload)}, agent_code={agent_code}")
+        
         submit_response = requests.post(
             submit_url,
-            json={"username": username, "answers": answers_payload},
+            json=payload,
             timeout=30
         )
         submit_response.raise_for_status()
         submission_result = submit_response.json()
-        print(f"Submission successful: {submission_result}")
+        print(f"Submission successful:  {submission_result}")
+    except requests.exceptions.HTTPError as e:
+        print(f"HTTP Error submitting answers: {e}")
+        print(f"Response text: {submit_response.text[: 500]}")
+        return f"Error submitting answers: {e}\n\nResponse:  {submit_response.text[: 200]}", None
     except Exception as e:
         print(f"Error submitting answers: {e}")
         return f"Error submitting answers: {e}", None
 
     # Display results
     results_df = pd.DataFrame(results_log, columns=["task_id", "question", "answer"])
-    result_message = f"✅ Submission Result:  {submission_result}\n\n🔗 [View Code]({agent_code})"
+    
+    # Extract score from submission result
+    score = submission_result.get('score', 'N/A')
+    result_message = f"""
+## ✅ Submission Successful!
+
+**Score:** {score}
+
+**Username:** {username}
+
+**Questions Answered:** {len(answers_payload)}
+
+🔗 [View Your Agent Code]({agent_code})
+    """
     
     return result_message, results_df
 
 # --- Gradio UI Setup ---
 with gr.Blocks() as demo:
-    gr. Markdown("""
+    gr.Markdown("""
     # 🤖 Agent Final Assignment - Unit 4
     
     **Instructions:**
@@ -163,7 +190,7 @@ with gr.Blocks() as demo:
         submit_button = gr.Button("🚀 Run Evaluation & Submit All Answers", variant="primary", size="lg")
     
     output_text = gr. Markdown()
-    output_table = gr. Dataframe(label="Results")
+    output_table = gr.Dataframe(label="Results")
 
     # The profile is automatically injected when using gr.LoginButton()
     submit_button.click(