Update app.py

app.py

@@ -256,7 +256,16 @@ class BasicAgent:
                 
                 # Get content type from response headers first, fallback to URL-based detection
                 kind = response.headers.get("Content-Type", "")
-                if not kind:  # fallback if header missing
+                if kind in ("application/octet-stream", ""):
+                    # rough sniff: look at the first few bytes
+                    sig = data[:4]
+                    if sig.startswith(b"\x89PNG"):
+                        kind = "image/png"
+                    elif sig.startswith(b"\xFF\xD8"):
+                        kind = "image/jpeg"
+                    elif sig[:2] == b"PK":  # XLSX = ZIP
+                        kind = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                elif not kind:  # fallback if header missing
                     kind = mimetypes.guess_type(state["file_url"])[0] or ""
                 print(f"Detected file type: {kind}")
                 
@@ -266,7 +275,7 @@ class BasicAgent:
                 elif "video" in kind:
                     print("Processing as video...")
                     answer = video_label_bytes(data)
-                elif kind.endswith("spreadsheet") or state["file_url"].endswith((".xlsx", ".csv")):
+                elif "spreadsheet" in kind or "excel" in kind:
                     print("Processing as spreadsheet...")
                     answer = sheet_answer_bytes(data, state["question"])
                 elif state["file_url"].endswith(".py"):
@@ -384,11 +393,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             
             # Handle file URL with conditional fallback to generic attachment endpoint
             raw_url = item.get("file_url") or ""
-            if not raw_url and "attached" in item["question"].lower():
-                raw_url = f"/files/{task_id}"  # generic attachment endpoint
-            
-            # Only construct file_url if we have a raw_url
-            file_url = f"{api_url}{raw_url}" if raw_url else ""  # empty means "no file"
+            if not raw_url:  # field missing
+                raw_url = attachment_url(task_id, api_url, sess) or ""
+            file_url = raw_url  # already absolute
             
             answer = agent(
                 question=item.get("question", ""),
@@ -440,6 +447,17 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {str(e)}", pd.DataFrame(results_log)
 
+def attachment_url(task_id: str, api_url: str, sess: requests.Session) -> str | None:
+    """Probe if a task has an attachment, return URL if it exists."""
+    probe = f"{api_url}/files/{task_id}"
+    try:
+        r = sess.head(probe, timeout=10)
+        if r.status_code == 200:
+            return probe  # attachment exists
+    except requests.RequestException:
+        pass
+    return None  # no file
+
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")