Update app.py

app.py

@@ -39,7 +39,7 @@ class BasicAgent:
                     {"type": "image_url", "image_url": f"data:image/png;base64,{encoded_image}"}
                 ])]
 
-                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()
 
@@ -61,7 +61,7 @@ class BasicAgent:
                     }
                 ])]
 
-                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()
 
@@ -71,7 +71,7 @@ class BasicAgent:
 
         else:
             messages = [HumanMessage(content=full_prompt)]
-            messages = self.final_agent.invoke({"messages": messages})
+            messages = self.graph.invoke({"messages": messages})
             answer = messages['messages'][-1].content
             return answer.split("FINAL ANSWER:")[-1].strip()