Update app.py

app.py

@@ -1,11 +1,13 @@
+import os
 import gradio as gr
-from langchain.agents import AgentExecutor
+from langchain.agents import initialize_agent
 from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain.tools import Tool
 
 # Define the tool
+
 def create_your_own(query: str) -> str:
     """This function can do whatever you would like once you fill it in"""
     return query[::-1]
@@ -28,25 +30,28 @@ tools = [
 class cbfs:
     
     def __init__(self, tools):
-        self.panels = []
-        self.functions = [tool.func for tool in tools]
-        self.model = ChatOpenAI(temperature=0)
+        self.model = ChatOpenAI(temperature=0, openai_api_key=os.getenv("OPENAI_API_KEY"))
         self.memory = ConversationBufferMemory(return_messages=True, memory_key="chat_history")
         self.prompt = ChatPromptTemplate.from_messages([
-            ("system", "You are helpful but sassy assistant"),
+            ("system", "You are a helpful but sassy assistant"),
             MessagesPlaceholder(variable_name="chat_history"),
             ("user", "{input}"),
             MessagesPlaceholder(variable_name="agent_scratchpad")
         ])
-        self.chain = self.model.bind_tools(tools)  # Connecting the tools to the model
-        self.qa = AgentExecutor(agent=self.chain, tools=tools, verbose=False, memory=self.memory)
+        
+        self.chain = initialize_agent(
+            tools=tools, 
+            llm=self.model, 
+            agent="chat-conversational-react-description",
+            memory=self.memory,
+            verbose=True
+        )
     
     def convchain(self, query):
         if not query:
             return "Please enter a query."
-        result = self.qa.invoke({"input": query})
-        self.answer = result['output'] 
-        return self.answer
+        result = self.chain.invoke({"input": query})
+        return result.get('output_text', "No response generated.")
 
 # Create an instance of the agent
 cb = cbfs(tools)
@@ -60,7 +65,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         inp = gr.Textbox(placeholder="Enter text here…", label="User Input")
         output = gr.Textbox(placeholder="Response...", label="ChatBot Output", interactive=False)
-    inp.submit(process_query, inputs=inp, outputs=output)
     
+    inp.submit(process_query, inputs=inp, outputs=output)
+
 demo.launch(share=True)
 
+