Update app.py

app.py

@@ -29,19 +29,33 @@ def set_anthropic_api_key(api_key: str):
         return agent
 
 
-def chat(
-    inp: str, history: Optional[Tuple[str, str]], agent: Optional[create_pandas_dataframe_agent]
-):
-    """Execute the chat functionality."""
-    history = history or []
-    # If agent is None, that is because no API key was provided.
-    if agent is None:
-        history.append((inp, "Please paste your Anthropic key to use"))
+class ChatWrapper:
+
+    def __init__(self):
+        self.lock = Lock()
+    def __call__(
+        self, api_key: str, inp: str, history: Optional[Tuple[str, str]], agent: Optional[create_pandas_dataframe_agent]
+    ):
+        """Execute the chat functionality."""
+        self.lock.acquire()
+        try:
+            history = history or []
+            # If chain is None, that is because no API key was provided.
+            if agent is None:
+                history.append((inp, "Please paste your OpenAI key to use"))
+                return history, history
+            # Set OpenAI key
+            agent = set_anthropic_api_key(api_key)
+            # Run chain and append input.
+            output = agent.run(input=inp)
+            history.append((inp, output))
+        except Exception as e:
+            raise e
+        finally:
+            self.lock.release()
         return history, history
-    # Run agent and append input.
-    output = agent.run(input=inp)
-    history.append((inp, output))
-    return history, history
+
+chat = ChatWrapper()
 
 
 block = gr.Blocks(css=".gradio-container {background-color: lightgray}")