Update app.py

app.py

@@ -10,9 +10,6 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 
 embeddings = OpenAIEmbeddings()
 
-# PINECONE_API_KEY = '6af52b8a-a3df-4189-899b-b21163027bb8'
-# PINECONE_API_ENV = 'asia-southeast1-gcp'
-
 PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
 PINECONE_API_ENV = os.getenv("PINECONE_API_ENV")
 PINECONE_INDEX = os.getenv("PINECONE_INDEX")
@@ -26,13 +23,16 @@ pinecone.init(
 index_name = PINECONE_INDEX
 vectorstore = Pinecone.from_existing_index(index_name=index_name, embedding=embeddings)
 
+chain = get_chain(vectorstore)
+
+
 class ChatWrapper:
 
     def __init__(self):
         self.lock = Lock()
 
     def __call__(
-            self, inp: str, history: Optional[Tuple[str, str]], chain
+            self, inp: str, history: Optional[Tuple[str, str]]
     ):
         """Execute the chat functionality."""
         self.lock.acquire()
@@ -44,8 +44,6 @@ class ChatWrapper:
             #     return history, history
             # Set OpenAI key
 
-            chain = get_chain(vectorstore)
-
             import openai
             openai.api_key = os.getenv("OPENAI_API_KEY")
             # Run chain and append input.
@@ -94,8 +92,8 @@ with block:
 
     state = gr.State()
     agent_state = gr.State()
-    submit.click(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
-    message.submit(chat, inputs=[message, state, agent_state], outputs=[chatbot, state])
+    submit.click(chat, inputs=[message, state], outputs=[chatbot, state])
+    message.submit(chat, inputs=[message, state], outputs=[chatbot, state])
 
     # openai_api_key_textbox.change(
     #     set_openai_api_key,