Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 
-from multi_agent import run_multi_agent
+from multi_agent import MultiAgent
 
 LLM_WHITE = "gpt-4o"
 LLM_BLACK = "gpt-4o"
@@ -11,8 +11,10 @@ def invoke(openai_api_key, num_moves = 10):
         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)
+
+    multi_agent = MultiAgent()
+    
+    return multi_agent.run_multi_agent(LLM_WHITE, LLM_BLACK, num_moves)
 
 gr.close_all()