Update app.py

app.py

@@ -29,8 +29,6 @@ def load_model():
 local_llm = load_model()
 
 
-from langchain.agents import create_react_agent, AgentExecutor, Tool
-
 # Define the tools for the agent
 def search_leads(query):
     results = [lead for lead in LEADS if query.lower() in lead['name'].lower()]
@@ -56,7 +54,7 @@ tools = [
     )
 ]
 
-# Define the prompt without needing an explicit output key
+# Define the prompt
 prompt = PromptTemplate.from_template(
     """You are an AI CyberSecurity Program Advisor. Your goal is to engage with leads and get them to book a video call for an in-person sales meeting. You have access to a list of leads and can send emails.
 
@@ -81,7 +79,7 @@ Thought: Let's approach this step-by-step:
 {agent_scratchpad}"""
 )
 
-# Create the React agent (output_key is removed)
+# Create the React agent
 agent = create_react_agent(
     llm=local_llm, 
     tools=tools, 
@@ -115,9 +113,10 @@ if lead_name:
         
         if st.button("Engage with Lead"):
             with st.spinner("AI is generating a response..."):
-                response = agent_executor.run(initial_message)
+                response = agent_executor.call({"input": initial_message})
                 st.write("AI Response:")
-                st.write(response)  # The response should now directly be the final output
+                st.write(response['output'])  # Access the final output using the appropriate key
+