Update app.py

app.py

@@ -254,7 +254,10 @@ class BasicAgent:
                 data = response.content
                 print(f"Successfully downloaded file, size: {len(data)} bytes")
                 
-                kind = mimetypes.guess_type(state["file_url"])[0] or ""
+                # 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
+                    kind = mimetypes.guess_type(state["file_url"])[0] or ""
                 print(f"Detected file type: {kind}")
                 
                 if "image" in kind:
@@ -379,11 +382,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         try:
             print(f"\nProcessing question {task_id}...")
             
-            # Handle file URL with fallback to generic attachment endpoint
+            # Handle file URL with conditional fallback to generic attachment endpoint
             raw_url = item.get("file_url") or ""
-            if not raw_url:  # fallback for empty field
+            if not raw_url and "attached" in item["question"].lower():
                 raw_url = f"/files/{task_id}"  # generic attachment endpoint
-            file_url = f"{api_url}{raw_url}"  # absolute URL
+            
+            # 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"
             
             answer = agent(
                 question=item.get("question", ""),