Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
-import os
+import os, threading
 
 from multi_agent import run_multi_agent
 
+lock = threading.Lock()
+
 LLM_WHITE = "gpt-4o"
 LLM_BLACK = "gpt-4o"
 
@@ -10,9 +12,11 @@ def invoke(openai_api_key, num_moves = 10):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
 
-    os.environ["OPENAI_API_KEY"] = openai_api_key
-
-    return run_multi_agent(LLM_WHITE, LLM_BLACK, num_moves)
+    with lock:
+        os.environ["OPENAI_API_KEY"] = openai_api_key
+        result = run_multi_agent(LLM_WHITE, LLM_BLACK, num_moves)
+        del os.environ["OPENAI_API_KEY"]
+        return result
 
 gr.close_all()