Update app.py

app.py

@@ -8,19 +8,14 @@ import subprocess
 import sys
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# Track API calls for rate limiting
 _last_call_time = 0
 
 def rate_limited_groq(api_key, prompt, system="", max_tokens=128):
-    """Call Groq with rate limiting — max 25 req/min to stay safe."""
     global _last_call_time
-    # Ensure at least 2.5 seconds between calls (= 24/min, safely under 30 limit)
     elapsed = time.time() - _last_call_time
     if elapsed < 2.5:
         time.sleep(2.5 - elapsed)
     _last_call_time = time.time()
-
     url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
     msgs = []
@@ -38,12 +33,16 @@ def rate_limited_groq(api_key, prompt, system="", max_tokens=128):
         raise Exception(f"Groq {resp.status_code}: {resp.text[:200]}")
     return resp.json()["choices"][0]["message"]["content"].strip()
 
-def download_task_file(task_id):
+def download_task_file(task_id, hf_token=None):
+    """Download with HF OAuth token for authentication."""
     url = f"{DEFAULT_API_URL}/files/{task_id}"
+    headers = {}
+    if hf_token:
+        headers["Authorization"] = f"Bearer {hf_token}"
     try:
-        resp = requests.get(url, timeout=30)
-        print(f"  File request: HTTP {resp.status_code}, size={len(resp.content)}, "
-              f"content-type={resp.headers.get('content-type','?')}")
+        resp = requests.get(url, headers=headers, timeout=30)
+        print(f"  File [{task_id[:8]}]: HTTP {resp.status_code}, "
+              f"size={len(resp.content)}, ct={resp.headers.get('content-type','?')[:40]}")
         if resp.status_code != 200 or len(resp.content) == 0:
             return None, None
         cd = resp.headers.get("content-disposition", "")
@@ -57,12 +56,13 @@ def download_task_file(task_id):
             elif "excel" in ct or "spreadsheet" in ct: ext = ".xlsx"
             elif "csv" in ct: ext = ".csv"
             elif "image" in ct: ext = ".png"
+            elif "text" in ct: ext = ".txt"
             else: ext = ".bin"
             fname += ext
         tmp = tempfile.NamedTemporaryFile(delete=False, suffix=ext, prefix="gaia_")
         tmp.write(resp.content)
         tmp.close()
-        print(f"  Saved: {fname} -> {tmp.name}")
+        print(f"  Saved: {fname} ({len(resp.content)} bytes)")
         return tmp.name, fname
     except Exception as e:
         print(f"  Download error: {e}")
@@ -136,11 +136,12 @@ Reply with ONLY the final answer. No explanation. No prefix. No "The answer is".
 Give the bare answer: a name, number, word, or short phrase only."""
 
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, hf_token=None):
         self.key = os.getenv("GROQ_API_KEY", "")
         if not self.key:
             raise RuntimeError("GROQ_API_KEY not set!")
-        print(f"Agent ready. Key: {self.key[:8]}...")
+        self.hf_token = hf_token
+        print(f"Agent ready. Groq key: {self.key[:8]}... | HF token: {'YES' if hf_token else 'NO'}")
 
     def ask(self, prompt, max_tokens=128):
         return clean_answer(rate_limited_groq(self.key, prompt, SYSTEM, max_tokens))
@@ -156,10 +157,9 @@ class BasicAgent:
         file_ctx = ""
         is_py = False
 
-        # Download file
+        # Download file with HF token
         if task_id:
-            print(f"  Attempting file download for task_id={task_id}")
-            lp, fn = download_task_file(task_id)
+            lp, fn = download_task_file(task_id, self.hf_token)
             if lp and fn:
                 ext = os.path.splitext(fn)[-1].lower()
                 if ext == ".py":
@@ -175,8 +175,6 @@ class BasicAgent:
                 else:
                     contents = read_file_contents(lp, fn)
                     file_ctx = f"\n[File: {fn}]\n{contents[:4000]}\n"
-            else:
-                print(f"  No file found for this task.")
 
         # Web search
         search_ctx = ""
@@ -189,7 +187,7 @@ class BasicAgent:
         q = question.lower()
         fmt = ""
         if "studio album" in q:
-            fmt = "\nCount only SOLO studio albums (exclude collaborative albums). Single integer answer."
+            fmt = "\nCount only SOLO studio albums (not live, compilation, or collaborative). Single integer."
         elif "first name" in q:
             fmt = "\nFirst name only."
         elif "surname" in q or "last name" in q:
@@ -199,21 +197,19 @@ class BasicAgent:
         elif "how many" in q:
             fmt = "\nSingle integer only."
         elif "ioc" in q or ("country" in q and "olympic" in q):
-            fmt = "\nIOC country code only (3 letters, e.g. USA, GBR). If tied, alphabetically first."
+            fmt = "\nIOC country code only (3 letters). If tied, alphabetically first country."
         elif "excel" in q or ("sale" in q and "food" in q):
-            fmt = "\nUSD with two decimal places (e.g. 89.50). No $ sign."
+            fmt = "\nNumber with exactly two decimal places. No $ sign, no commas (e.g. 89.50)."
         elif "chess" in q:
-            fmt = "\nChess move in algebraic notation only."
+            fmt = "\nChess move in algebraic notation only (e.g. Qd8, e5)."
         elif "pitcher" in q and "number" in q:
-            fmt = "\nTwo last names, comma-separated, pitcher with lower jersey number first."
+            fmt = "\nTwo last names, comma-separated. Pitcher with lower jersey number first."
         elif "wikipedia" in q and "nominat" in q:
             fmt = "\nWikipedia username only."
         elif "grocery" in q or ("shopping" in q and "list" in q):
             fmt = "\nComma-separated list, alphabetical order."
         elif "youtube" in q or "video" in q:
-            fmt = "\nExact short answer only — quote, number, or name."
-        elif "grant" in q or "award number" in q:
-            fmt = "\nExact identifier only."
+            fmt = "\nExact short answer — quote, number, or brief phrase only."
 
         prompt = (
             f"Question: {question}"
@@ -240,9 +236,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if not profile:
         return "Please Login to Hugging Face.", None
+
     username = profile.username
+    # Get the HF OAuth token from the login profile
+    hf_token = getattr(profile, "token", None)
+    print(f"User: {username}, HF token present: {bool(hf_token)}")
+
     try:
-        agent = BasicAgent()
+        agent = BasicAgent(hf_token=hf_token)
     except RuntimeError as e:
         return f"❌ {e}", None
 
@@ -263,10 +264,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         print(f"\n[{i+1}/{len(questions_data)}]")
         try:
-            # Pass task_id directly — no injection needed
             ans = agent(question_text, task_id=task_id)
         except Exception as e:
             ans = ""
+            print(f"  Error: {e}")
         answers_payload.append({"task_id": task_id, "submitted_answer": ans})
         results_log.append({
             "Task ID": task_id,