Update app.py

app.py

@@ -28,6 +28,9 @@ 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()]
@@ -53,7 +56,7 @@ tools = [
     )
 ]
 
-# Set up the agent with a specified output key
+# Define the prompt without needing an explicit output key
 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.
 
@@ -70,7 +73,7 @@ Action Input: the input to the action
 Observation: the result of the action
 ... (this Thought/Action/Action Input/Observation can repeat N times)
 Thought: I now know the final answer
-Final Answer: the final response you want to return. This is your output.
+Final Answer: [Insert your final response here]
 Begin!
 
 Question: {input}
@@ -78,13 +81,14 @@ Thought: Let's approach this step-by-step:
 {agent_scratchpad}"""
 )
 
+# Create the React agent (output_key is removed)
 agent = create_react_agent(
     llm=local_llm, 
     tools=tools, 
     prompt=prompt
 )
 
-# Create the agent executor without specifying output_keys
+# Create the agent executor
 agent_executor = AgentExecutor.from_agent_and_tools(
     agent=agent, 
     tools=tools, 
@@ -113,7 +117,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)  # The response should be the final output
+                st.write(response)  # The response should now directly be the final output
+
 
 
 st.sidebar.title("About")