| import gradio as gr | |
| import os | |
| from multi_agent import MultiAgent | |
| LLM_WHITE = "gpt-4o" | |
| LLM_BLACK = "gpt-4o" | |
| 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 | |
| multi_agent = MultiAgent() | |
| return multi_agent.run_multi_agent(LLM_WHITE, LLM_BLACK, num_moves) | |
| gr.close_all() | |
| demo = gr.Interface(fn = invoke, | |
| inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1), | |
| gr.Number(label = "Number of Moves", value = 3, minimum = 1, maximum = 300)], | |
| outputs = [gr.Markdown(label = "Game", value=os.environ["OUTPUT"], line_breaks=True, sanitize_html=False)], | |
| title = "Chess: AI vs. AI", | |
| description = os.environ["DESCRIPTION"]) | |
| demo.launch() |