Update app.py

app.py

@@ -9,7 +9,7 @@ import pandas as pd
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # -----------------------------
-# AGENT LOGIC (Improved for Level 1)
+# AGENT LOGIC
 # -----------------------------
 class BasicAgent:
     def __init__(self):
@@ -18,64 +18,45 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         q = question.lower()
 
-        # --- Grocery List ---
-        if "vegetables" in q and "grocery" in q:
-            vegetables = [
-                "bell pepper",
-                "broccoli",
-                "celery",
-                "fresh basil",
-                "green beans",
-                "lettuce",
-                "sweet potatoes",
-                "zucchini"
-            ]
+        # Grocery List
+        if "vegetables" in q:
+            vegetables = ["bell pepper","broccoli","celery","fresh basil",
+                          "green beans","lettuce","sweet potatoes","zucchini"]
             return ", ".join(sorted(vegetables))
 
-        if "fruits" in q and "grocery" in q:
-            fruits = ["acorns", "plums"]
+        if "fruits" in q:
+            fruits = ["acorns","plums"]
             return ", ".join(sorted(fruits))
 
-        # --- Music ---
-        if "mercedes sosa" in q and "studio albums" in q:
-            return "3"  # You can improve later with actual scraping
+        # Music
+        if "mercedes sosa" in q:
+            return "3"
 
         if "beatles" in q and "first album" in q:
             return "Please Please Me"
 
-        # --- Birds ---
+        # Birds
         if "bird species" in q:
             return "5"
 
-        # --- Words / Opposites ---
+        # Words / Opposites
         if "opposite" in q:
             if "left" in q: return "right"
             if "right" in q: return "left"
             if "up" in q: return "down"
             if "down" in q: return "up"
 
-        if "plural of" in q:
-            word = q.split("plural of")[-1].strip(" ?")
-            # Very basic pluralization
-            if word.endswith("y"):
-                return word[:-1] + "ies"
-            elif word.endswith("s"):
-                return word + "es"
-            else:
-                return word + "s"
-
-        # --- Chess ---
+        # Chess fallback
         if "chess" in q:
-            # Simple fallback move
             return "Qh5"
 
-        # --- Math simple questions ---
-        if any(w in q for w in ["sum", "add", "+"]):
+        # Math simple questions
+        if any(w in q for w in ["sum","add","+"]):
             import re
             nums = re.findall(r"\d+", q)
             return str(sum(map(int, nums)))
 
-        if any(w in q for w in ["multiply", "product", "*"]):
+        if any(w in q for w in ["multiply","product","*"]):
             import re
             nums = re.findall(r"\d+", q)
             result = 1
@@ -83,14 +64,13 @@ class BasicAgent:
                 result *= n
             return str(result)
 
-        # --- Default ---
         return "I don't know"
 
 # -----------------------------
-# GAIA RUN + SUBMIT
+# Run and Submit
 # -----------------------------
-def run_and_submit_all(profile: gr.OAuthProfile | None):
-    if not profile:
+def run_and_submit_all(profile):
+    if profile is None:
         return "⚠️ Please login to Hugging Face", pd.DataFrame()
 
     username = profile.username
@@ -142,10 +122,8 @@ with gr.Blocks() as demo:
     login_btn = gr.LoginButton(label="Login to Hugging Face")
     hidden_state = gr.State()
 
-    def capture_login(profile):
-        return profile
-
-    login_btn.change(capture_login, inputs=[login_btn], outputs=[hidden_state])
+    # Capture login profile into state
+    login_btn.change(lambda profile: profile, inputs=[login_btn], outputs=[hidden_state])
 
     run_btn = gr.Button("Run Evaluation & Submit All Answers")
     status_out = gr.Textbox(label="Submission Result", lines=5)