Update app.py

app.py

@@ -65,8 +65,7 @@ class BasicAgent:
                 # No file attached
                 messages = [HumanMessage(content=question)]
     
-            # Invoke final agent and extract answer
-            response = self.final_agent.invoke({"messages": messages})
+            response = self.graph.invoke({"messages": messages})
             answer = response['messages'][-1].content
             return answer.split("FINAL ANSWER:")[-1].strip()