Update gemini_agent.py

gemini_agent.py

@@ -117,7 +117,7 @@ class GEMINI_AGENT:
         for attempt in range(max_retries):
             try:
                 response = self.app.invoke({"messages": messages, "input_file": None})
-                final_ans = self.extract_after_final_answer(response)
+                final_ans = self.extract_after_final_answer(response['messages'][-1].content)
                 time.sleep(60) # avoid rate limit
                 return final_ans
             except Exception as e: