fix answering

app.py

@@ -76,6 +76,9 @@ def assistant(state: AgentState):
     system_prompt = SystemMessage(
         content=f"""
         You are an expert assistant. Try to answer the question as accurately as possible.
+        If you don't know the answer, you can use the tools available to you. 
+        But try to answer the question first and only use the tools if you are not sure.
+        If you get a response from a tool, try to come to the final answer.
         You can use the following tools to help you:
         {textual_description_of_tool}
         """
@@ -166,8 +169,8 @@ class BasicAgent:
 
         print(f"Initial state: {initial_state}")
         answer = compiled_graph.invoke(initial_state)
-        print(f"Agent returning answer: {answer}")
-        return answer
+        print(f"Agent returning answer: {answer.get('answer')}")
+        return answer.get("answer")
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """