Update app.py

app.py

@@ -280,21 +280,37 @@ def read_task_file(file_path: str) -> str:
         return f.read()
 
 
+_DOWNLOAD_DIR = os.path.join(os.environ.get("TMPDIR", "/tmp"), "gaia_files")
+os.makedirs(_DOWNLOAD_DIR, exist_ok=True)
+
+
 def _download_task_file(task_id: str, file_name: str, api_url: str = DEFAULT_API_URL) -> str | None:
     """Download a file attached to a GAIA task and save it locally. Returns local path or None."""
     url = f"{api_url}/files/{task_id}"
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    try:
-        resp = requests.get(url, headers=headers, timeout=30)
-        resp.raise_for_status()
-        local_path = os.path.join(os.getcwd(), f"task_{task_id}_{file_name}")
-        with open(local_path, "wb") as fh:
-            fh.write(resp.content)
-        print(f"Downloaded task file: {local_path}")
-        return local_path
-    except Exception as e:
-        print(f"Could not download file for task {task_id}: {e}")
-        return None
+    local_path = os.path.join(_DOWNLOAD_DIR, f"task_{task_id}_{file_name}")
+
+    # Try with auth first, then without (some endpoints don't require it)
+    for headers in [
+        {"Authorization": f"Bearer {HF_TOKEN}"},
+        {},
+    ]:
+        try:
+            resp = requests.get(url, headers=headers, timeout=30)
+            resp.raise_for_status()
+            with open(local_path, "wb") as fh:
+                fh.write(resp.content)
+            print(f"Downloaded task file: {local_path} ({len(resp.content)} bytes)")
+            return local_path
+        except requests.exceptions.HTTPError as e:
+            status = e.response.status_code if e.response is not None else "?"
+            print(f"Download attempt for {task_id} returned {status} (auth={'yes' if headers else 'no'})")
+            continue
+        except Exception as e:
+            print(f"Download error for {task_id}: {e}")
+            continue
+
+    print(f"Could not download file for task {task_id} after all attempts")
+    return None
 
 
 class WebSearchAgent:
@@ -309,13 +325,6 @@ class WebSearchAgent:
                 api_base="https://api.groq.com/openai/v1",
                 api_key=GROQ_API_KEY,
                 timeout=60,
-                # --- Groq alternatives ---
-                # model_id="meta-llama/llama-4-scout-17b-16e-instruct",
-                # --- OpenRouter: free models (often 429/rate-limited) ---
-                # model_id="arcee-ai/trinity-large-preview:free",
-                # model_id="google/gemma-3-27b-it:free",
-                # api_base="https://openrouter.ai/api/v1",
-                # api_key=OPENROUTER_API_KEY,
             ),
             tools=[
                 web_search,