Update app.py

app.py

@@ -223,23 +223,79 @@ class BasicAgent:
             # as a last resort, strip everything before the first colon
             return raw.split(':', 1)[-1].strip()
 
+    def __call__(self, question: str, task_id: str = "unknown") -> str:
+        # Parse question to get both text and file_url
+        try:
+            question_data = json.loads(question)
+            state: AgentState = {
+                "question": question_data.get("question", ""),
+                "current_step": "answer",
+                "final_answer": "",
+                "history": [],
+                "needs_search": False,
+                "search_query": "",
+                "task_id": task_id,
+                "logs": {},
+                "file_url": question_data.get("file_url", ""),
+                "code_blocks": question_data.get("code_blocks", [])
+            }
+            print(f"\nProcessing task {task_id}")
+            print(f"Question: {state['question']}")
+            print(f"File URL: {state['file_url']}")
+        except (json.JSONDecodeError, KeyError) as e:
+            print(f"Error parsing question data: {e}")
+            state: AgentState = {
+                "question": question,
+                "current_step": "answer",
+                "final_answer": "",
+                "history": [],
+                "needs_search": False,
+                "search_query": "",
+                "task_id": task_id,
+                "logs": {},
+                "file_url": "",
+                "code_blocks": []
+            }
+        
+        final_state = self.workflow.invoke(state)
+        return final_state["final_answer"]
+
     def _generate_answer(self, state: AgentState) -> AgentState:
         if state["file_url"]:
             try:
+                print(f"\nProcessing file: {state['file_url']}")
                 kind = mimetypes.guess_type(state["file_url"])[0] or ""
-                data = requests.get(state["file_url"]).content
+                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"])
                 elif "video" in kind:
+                    print("Processing as video...")
                     answer = video_label_bytes(data)
                 elif kind.endswith("spreadsheet") or state["file_url"].endswith((".xlsx", ".csv")):
+                    print("Processing as spreadsheet...")
                     answer = sheet_answer_bytes(data, state["question"])
                 elif state["file_url"].endswith(".py"):
+                    print("Processing as Python file...")
                     answer = run_python(data.decode())
                 else:
-                    answer = "Unsupported file type"
+                    print(f"Unsupported file type: {kind}")
+                    answer = f"Unsupported file type: {kind}"
                 
+                print(f"Generated answer: {answer}")
                 state["final_answer"] = answer
                 state["current_step"] = "done"
                 return state
@@ -250,6 +306,7 @@ class BasicAgent:
                 return state
 
         # For text-only questions, use the LLM
+        print("\nProcessing as text-only question...")
         prompt = f"""
 Answer this question using the materials provided.
 
@@ -262,6 +319,7 @@ Return ONLY this exact JSON object:
         try:
             raw = self._call_llm(prompt, 300)
             answer = self._safe_parse(raw)
+            print(f"Generated answer: {answer}")
             state["final_answer"] = answer
         except Exception as e:
             print(f"\nLLM Error in answer generation: {str(e)}")
@@ -277,23 +335,6 @@ Return ONLY this exact JSON object:
         sg.set_finish_point("answer")
         return sg.compile()
 
-    def __call__(self, question: str, task_id: str = "unknown") -> str:
-        state: AgentState = {
-            "question": question,
-            "current_step": "answer",
-            "final_answer": "",
-            "history": [],
-            "needs_search": False,
-            "search_query": "",
-            "task_id": task_id,
-            "logs": {},
-            "file_url": "",
-            "code_blocks": []
-        }
-        
-        final_state = self.workflow.invoke(state)
-        return final_state["final_answer"]
-
 # ----------------------------------------------------------------------------------
 # Gradio Interface & Submission Routines
 # ----------------------------------------------------------------------------------