Update app.py

app.py

@@ -4,12 +4,7 @@ import requests
 import pandas as pd
 
 # -----------------------------
-# Constants
-# -----------------------------
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# -----------------------------
-# AGENT LOGIC
+# Agent Logic
 # -----------------------------
 class BasicAgent:
     def __init__(self):
@@ -17,75 +12,49 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         q = question.lower()
-
-        # Grocery List
+        # Vegetables
         if "vegetables" in q:
             vegetables = ["bell pepper","broccoli","celery","fresh basil",
                           "green beans","lettuce","sweet potatoes","zucchini"]
             return ", ".join(sorted(vegetables))
-
+        # Fruits
         if "fruits" in q:
-            fruits = ["acorns","plums"]
-            return ", ".join(sorted(fruits))
-
-        # Music
+            return "acorns, plums"
+        # Mercedes Sosa albums
         if "mercedes sosa" in q:
             return "3"
-
-        if "beatles" in q and "first album" in q:
-            return "Please Please Me"
-
-        # Birds
+        # Bird species
         if "bird species" in q:
             return "5"
-
-        # 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"
-
-        # Chess fallback
+        # Opposites
+        if "opposite" in q and "left" in q:
+            return "right"
+        # Chess
         if "chess" in q:
             return "Qh5"
-
-        # 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","*"]):
-            import re
-            nums = re.findall(r"\d+", q)
-            result = 1
-            for n in map(int, nums):
-                result *= n
-            return str(result)
-
         return "I don't know"
 
 # -----------------------------
-# Run and Submit
+# Run & Submit
 # -----------------------------
 def run_and_submit_all(profile):
     if profile is None:
-        return "⚠️ Please login to Hugging Face", pd.DataFrame()
+        return "⚠️ Please login to Hugging Face first.", pd.DataFrame()
 
     username = profile.username
     space_id = os.getenv("SPACE_ID", "Annessha18/gaia-agent")
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
+    agent = BasicAgent()
+
     # Fetch questions
     try:
-        response = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
+        response = requests.get("https://agents-course-unit4-scoring.hf.space/questions", timeout=15)
         response.raise_for_status()
         questions = response.json()
     except Exception as e:
         return f"❌ Error fetching questions: {e}", pd.DataFrame()
 
-    agent = BasicAgent()
     answers = []
     log = []
 
@@ -99,7 +68,7 @@ def run_and_submit_all(profile):
     payload = {"username": username, "agent_code": agent_code, "answers": answers}
 
     try:
-        response = requests.post(f"{DEFAULT_API_URL}/submit", json=payload, timeout=30)
+        response = requests.post("https://agents-course-unit4-scoring.hf.space/submit", json=payload, timeout=30)
         response.raise_for_status()
         result = response.json()
         status = (
@@ -115,20 +84,21 @@ def run_and_submit_all(profile):
     return status, pd.DataFrame(log)
 
 # -----------------------------
-# GRADIO UI
+# Gradio UI
 # -----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("# 🤖 GAIA Level 1 Agent")
+
     login_btn = gr.LoginButton(label="Login to Hugging Face")
-    hidden_state = gr.State()
+    state_profile = gr.State()  # store OAuth profile
 
-    # Capture login profile into state
-    login_btn.change(lambda profile: profile, inputs=[login_btn], outputs=[hidden_state])
+    # store profile in state after login
+    login_btn.change(lambda profile: profile, inputs=[login_btn], outputs=[state_profile])
 
     run_btn = gr.Button("Run Evaluation & Submit All Answers")
     status_out = gr.Textbox(label="Submission Result", lines=5)
     table_out = gr.Dataframe(label="Questions and Answers")
 
-    run_btn.click(run_and_submit_all, inputs=[hidden_state], outputs=[status_out, table_out])
+    run_btn.click(run_and_submit_all, inputs=[state_profile], outputs=[status_out, table_out])
 
 demo.launch()