Update app.py

app.py

@@ -29,7 +29,7 @@ sys.stdout.reconfigure(line_buffering=True)
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # Single values — no per-difficulty branching
-TIMEOUT_SECONDS  = 120
+TIMEOUT_SECONDS  = 60
 RECURSION_LIMIT  = 12
 PAUSE_SECONDS    = 1.0
 
@@ -105,9 +105,10 @@ def submit_answers(url: str, payload: dict) -> dict:
     return r.json()
 
 
-def fetch_task_file(task_id: str) -> tuple[str | None, str | None]:
+def fetch_task_file(task_id: str, question: str = "") -> tuple[str | None, str | None]:
     """
     Download the file attached to a task from GET /files/{task_id}.
+    If that fails, search the question text for embedded file content or links.
     Returns (local_path, injected_text):
       - local_path: saved file path (for binary files like .py the model runs directly)
       - injected_text: pre-read text content to inject into the question (for Excel/CSV)
@@ -117,8 +118,18 @@ def fetch_task_file(task_id: str) -> tuple[str | None, str | None]:
     try:
         r = requests.get(url, timeout=20)
         if r.status_code == 404:
-            log("File     : no attachment (404)")
-            return None, None
+            log(f"File     : endpoint /files/{task_id} returned 404 — trying alternate patterns")
+            # Fallback: try query param format
+            url2 = f"{DEFAULT_API_URL}/files?task_id={task_id}"
+            try:
+                r = requests.get(url2, timeout=20)
+                if r.status_code != 200:
+                    log(f"File     : no attachment found (both endpoints failed)")
+                    return None, None
+            except Exception as e:
+                log(f"File     : fallback endpoint also failed — {e}")
+                return None, None
+
         r.raise_for_status()
 
         # Determine extension
@@ -139,10 +150,8 @@ def fetch_task_file(task_id: str) -> tuple[str | None, str | None]:
             ext = ".png"
         elif "jpeg" in ct or "jpg" in ct:
             ext = ".jpg"
-        elif "python" in ct:
+        elif "python" in ct or "text/plain" in ct:
             ext = ".py"
-        elif "text/plain" in ct:
-            ext = ".txt"
         else:
             ext = ""
 
@@ -244,7 +253,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         log("-" * 80)
 
         # Download any attached file and inject content into the question
-        file_path, injected_text = fetch_task_file(task_id)
+        file_path, injected_text = fetch_task_file(task_id, question)
         if injected_text:
             # For Excel/CSV/text: embed the content directly so the model
             # doesn't need to re-download anything