Update app.py

app.py

@@ -19,7 +19,7 @@ class BasicAgent:
         
         # Try to provide context-aware answers
         if "capital" in question_lower:
-            if "france" in question_lower:
+            if "france" in question_lower: 
                 return "Paris"
             elif "germany" in question_lower:
                 return "Berlin"
@@ -30,22 +30,22 @@ class BasicAgent:
             elif "japan" in question_lower:
                 return "Tokyo"
         
-        if "who" in question_lower and "ceo" in question_lower: 
-            if "tesla" in question_lower or "spacex" in question_lower: 
+        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: 
                 return "Tim Cook"
             elif "microsoft" in question_lower:
                 return "Satya Nadella"
         
-        if "what year" in question_lower or "when" 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:
                 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"
@@ -55,18 +55,20 @@ class BasicAgent:
         print(f"Agent returning answer: {answer}")
         return answer
 
-def run_and_submit_all(profile: gr.OAuthProfile | None):
+def run_and_submit_all(profile):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers, and displays the results. 
+    Fetches all questions, runs the BasicAgent on them, submits all answers, and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID")
-    if profile:
-        username = f"{profile.username}"
-        print(f"User logged in: {username}")
-    else:
+    
+    # Check if user is logged in
+    if profile is None:
         print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None
+        return "❌ Please login to Hugging Face using the button above.", None
+    
+    username = profile.username
+    print(f"User logged in: {username}")
 
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
@@ -96,10 +98,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print(f"Error fetching questions: {e}")
         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"Error decoding JSON response from questions endpoint:  {e}")
         print(f"Response text: {response.text[:500]}")
         return f"Error decoding server response for questions: {e}", None
-    except Exception as e: 
+    except Exception as e:
         print(f"An unexpected error occurred fetching questions: {e}")
         return f"An unexpected error occurred fetching questions: {e}", None
 
@@ -107,10 +109,10 @@ 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:
+        if not task_id or question_text is None: 
             print(f"Skipping item with missing task_id or question: {item}")
             continue
 
@@ -118,24 +120,24 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "answer": answer})
             results_log.append((task_id, question_text[: 50], answer[: 100]))
-        except Exception as e: 
-            print(f"Error running agent on question {task_id}:  {e}")
-            answers_payload.append({"task_id":  task_id, "answer": f"Error: {e}"})
-            results_log.append((task_id, question_text[:50], f"Error: {e}"))
+        except Exception as e:
+            print(f"Error running agent on question {task_id}: {e}")
+            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
     print(f"Submitting {len(answers_payload)} answers to {submit_url}...")
     try:
         submit_response = requests.post(
-            submit_url, 
-            json={"username": username, "answers": answers_payload}, 
+            submit_url,
+            json={"username": username, "answers": answers_payload},
             timeout=30
         )
         submit_response.raise_for_status()
         submission_result = submit_response.json()
         print(f"Submission successful: {submission_result}")
     except Exception as e:
-        print(f"Error submitting answers:  {e}")
+        print(f"Error submitting answers: {e}")
         return f"Error submitting answers: {e}", None
 
     # Display results
@@ -145,31 +147,31 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     return result_message, results_df
 
 # --- Gradio UI Setup ---
-with gr.Blocks() as demo:
+with gr. Blocks() as demo:
     gr.Markdown("""
     # 🤖 Agent Final Assignment - Unit 4
     
     **Instructions:**
-    1. Click "Sign in with Hugging Face" below
-    2. Click "Run Evaluation & Submit All Answers"
-    3. Wait for your score! 
+    1. Make sure you're logged in to Hugging Face (check top right)
+    2. Click "Run Evaluation & Submit All Answers" below
+    3. Wait for your score!
     
-    This agent will answer questions and submit results automatically.
+    This agent will answer questions and submit results automatically. 
     """)
-
-    with gr.Row():
-        hf_profile = gr.OAuthProfile()
     
     with gr.Row():
         submit_button = gr.Button("🚀 Run Evaluation & Submit All Answers", variant="primary", size="lg")
     
     output_text = gr.Markdown()
-    output_table = gr. Dataframe()
+    output_table = gr.Dataframe(label="Results")
 
     def submit_work(profile):
-        status, table = run_and_submit_all(profile)
-        return status, table
+        return run_and_submit_all(profile)
 
-    submit_button.click(submit_work, inputs=[hf_profile], outputs=[output_text, output_table])
+    submit_button. click(
+        submit_work,
+        inputs=gr.OAuthProfile(),
+        outputs=[output_text, output_table]
+    )
 
 demo.launch()