Update app.py

app.py

@@ -247,22 +247,18 @@ class BasicAgent:
     def _generate_answer(self, state: AgentState) -> AgentState:
         if state["file_url"]:
             try:
-                print(f"\nProcessing file: {state['file_url']}")
+                print(f"Downloading {state['file_url']} …")
+                headers = {}
+                if HF_TOKEN:
+                    headers["Authorization"] = f"Bearer {HF_TOKEN}"
+                response = requests.get(state["file_url"], headers=headers, timeout=30)
+                response.raise_for_status()
+                data = response.content
+                print(f"Successfully downloaded file, size: {len(data)} bytes")
+                
                 kind = mimetypes.guess_type(state["file_url"])[0] or ""
                 print(f"Detected file type: {kind}")
                 
-                # Download file with timeout and error handling
-                try:
-                    response = requests.get(state["file_url"], timeout=30)
-                    response.raise_for_status()
-                    data = response.content
-                    print(f"Successfully downloaded file, size: {len(data)} bytes")
-                except requests.exceptions.RequestException as e:
-                    print(f"Error downloading file: {e}")
-                    state["final_answer"] = f"Error downloading file: {str(e)}"
-                    state["current_step"] = "done"
-                    return state
-                
                 if "image" in kind:
                     print("Processing as image...")
                     answer = image_qa_bytes(data, state["question"])
@@ -283,6 +279,11 @@ class BasicAgent:
                 state["final_answer"] = answer
                 state["current_step"] = "done"
                 return state
+            except requests.exceptions.RequestException as e:
+                print(f"Error downloading file: {e}")
+                state["final_answer"] = f"Error downloading file: {str(e)}"
+                state["current_step"] = "done"
+                return state
             except Exception as e:
                 print(f"\nError processing file {state['file_url']}: {str(e)}")
                 state["final_answer"] = f"Error processing file: {str(e)}"