bstraehle commited on
Commit
965981d
1 Parent(s): 6e747e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,8 +1,10 @@
1
  import gradio as gr
2
- import os
3
 
4
  from multi_agent import run_multi_agent
5
 
 
 
6
  os.environ["LANGCHAIN_TRACING_V2"] = "true"
7
  os.environ["LANGCHAIN_PROJECT"] = "langgraph-multi-agent"
8
 
@@ -14,9 +16,11 @@ def invoke(openai_api_key, topic):
14
  if (topic == ""):
15
  raise gr.Error("Topic is required.")
16
 
17
- os.environ["OPENAI_API_KEY"] = openai_api_key
18
-
19
- return run_multi_agent(LLM, topic)
 
 
20
 
21
  gr.close_all()
22
 
 
1
  import gradio as gr
2
+ import os, threading
3
 
4
  from multi_agent import run_multi_agent
5
 
6
+ lock = threading.Lock()
7
+
8
  os.environ["LANGCHAIN_TRACING_V2"] = "true"
9
  os.environ["LANGCHAIN_PROJECT"] = "langgraph-multi-agent"
10
 
 
16
  if (topic == ""):
17
  raise gr.Error("Topic is required.")
18
 
19
+ with lock:
20
+ os.environ["OPENAI_API_KEY"] = openai_api_key
21
+ result = run_multi_agent(LLM, topic)
22
+ del os.environ["OPENAI_API_KEY"]
23
+ return result
24
 
25
  gr.close_all()
26