| import gradio as gr |
| import datetime, os, threading |
|
|
| from multi_agent import run_multi_agent |
|
|
| lock = threading.Lock() |
|
|
| LLM = "gpt-4o" |
|
|
| def invoke(openai_api_key, task): |
| if not openai_api_key: |
| raise gr.Error("OpenAI API Key is required.") |
|
|
| if not task: |
| raise gr.Error("Task is required.") |
|
|
| raise gr.Error("Please clone space due to local code execution.") |
| |
| with lock: |
| os.environ["OPENAI_API_KEY"] = openai_api_key |
| result = run_multi_agent(LLM, task) |
| del os.environ["OPENAI_API_KEY"] |
| return result |
|
|
| gr.close_all() |
|
|
| demo = gr.Interface(fn = invoke, |
| inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1), |
| gr.Textbox(label = "Task", value = f"Today is {datetime.date.today()}. {os.environ['INPUT']}")], |
| outputs = [gr.Markdown(label = "Output", value = os.environ["OUTPUT"], line_breaks = True, sanitize_html = False)], |
| title = "Multi-Agent AI: Coding", |
| description = os.environ["DESCRIPTION"]) |
|
|
| demo.launch() |
