Update app.py

app.py

@@ -90,7 +90,8 @@ agent_executor = AgentExecutor.from_agent_and_tools(
     agent=agent, 
     tools=tools, 
     verbose=True, 
-    memory=ConversationBufferMemory()
+    memory=ConversationBufferMemory(),
+    output_keys=["final_answer"]  # Explicitly define the output key(s) here
 )
 
 # Streamlit interface
@@ -114,7 +115,8 @@ if lead_name:
             with st.spinner("AI is generating a response..."):
                 response = agent_executor.run(initial_message)
                 st.write("AI Response:")
-                st.write(response["final_answer"])  # Access the final answer using the output key
+                st.write(response)  # Now 'response' will directly be the final answer
+
 
 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.")