Update app.py

app.py

@@ -85,12 +85,12 @@ agent = create_react_agent(
     output_key="final_answer"  # Specify the output key here
 )
 
+# Create the agent executor
 agent_executor = AgentExecutor.from_agent_and_tools(
     agent=agent, 
     tools=tools, 
     verbose=True, 
-    memory=ConversationBufferMemory(),
-    output_key="final_answer"  # Ensure this matches the agent's output key
+    memory=ConversationBufferMemory()
 )
 
 # Streamlit interface
@@ -113,9 +113,8 @@ if lead_name:
         if st.button("Engage with Lead"):
             with st.spinner("AI is generating a response..."):
                 response = agent_executor.run(initial_message)
-            
-            st.write("AI Response:")
-            st.write(response)
+                st.write("AI Response:")
+                st.write(response["final_answer"])  # Access the final answer using the output key
 
 st.sidebar.title("About")
 st.sidebar.info("This is a demo of an AI CyberSecurity Program Advisor using an open-source LLM and LangChain. It's designed to engage with leads and attempt to book video calls for sales meetings.")