Update app.py

app.py

@@ -76,7 +76,6 @@ def get_chain(vectorstore):
     
     
 class ChatWrapper:
-    global chain
 
     def __init__(self):
         self.lock = Lock()
@@ -88,7 +87,7 @@ class ChatWrapper:
         try:
             history = history or []
             # Set OpenAI key
-            chain = get_chain(vectorstore)
+            # chain = get_chain(vectorstore)
             # Run chain and append input.
             output = chain({"question": inp, "chat_history": history})["answer"]
             history.append((inp, output))
@@ -110,7 +109,7 @@ with block:
         embeddings.change(on_value_change, embeddings)
     
     vectorstore = load_vectorstore(embeddings.value)
-
+    chain = get_chain(vectorstore)
     chatbot = gr.Chatbot()
 
     chat = ChatWrapper()
@@ -144,8 +143,8 @@ with block:
     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, chain], outputs=[chatbot, state])
+    message.submit(chat, inputs=[message, state, chain], outputs=[chatbot, state])
 
     # agent_state = chain