Update app.py

app.py

@@ -257,19 +257,28 @@ class BasicAgent:
             "logs": {},
             "file_url": file_url,
             "code_blocks": [],
-            "next_step": ""  # Initialize next_step
+            "next_step": ""
         }
         
         print(f"\nProcessing task {task_id}")
         print(f"Question: {state['question']}")
         print(f"File URL: {state['file_url']}")
         
-        final_state = self.workflow.invoke(state)
-        return final_state["final_answer"]
+        try:
+            final_state = self.workflow.invoke(state)
+            answer = final_state.get("final_answer", "")
+            if not answer:
+                print("Warning: No answer generated in final state")
+                return "No answer could be generated"
+            return answer
+        except Exception as e:
+            print(f"Error in workflow execution: {str(e)}")
+            return f"Error processing question: {str(e)}"
 
     def _route_to_tool(self, state: AgentState) -> Dict[str, Any]:
         """Route the state to the appropriate tool based on file type."""
         if not state["file_url"]:
+            print("No file URL, routing to text processing")
             return {"next_step": "process_text"}
             
         try:
@@ -291,6 +300,8 @@ class BasicAgent:
             elif 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:
                 return {"next_step": "process_image"}
             elif "video" in kind: