Update multi_agent.py

multi_agent.py

@@ -1,8 +1,6 @@
-import chess, chess.svg, getpass, io, json, os
+import chess, chess.svg
 from autogen import ConversableAgent, register_function
 from typing_extensions import Annotated
-from io import BytesIO
-from PIL import Image
 
 made_move = False
 
@@ -49,8 +47,9 @@ def check_made_move(msg):
     else:
         return False
 
-def run_multi_agent(llm, num_turns):   
-    llm_config = {"model": llm}
+def run_multi_agent(llm_white, llm_black, num_turns):   
+    llm_config_white = {"model": llm_white}
+    llm_config_black = {"model": llm_black}
     
     board_proxy = ConversableAgent(
         name="Board Proxy",
@@ -67,7 +66,7 @@ def run_multi_agent(llm, num_turns):
         "Then call make_move(move) to make a move. "
         "After a move is made, analyze the move using technical jargon, don't make any other comments. "
         "Then continue playing.",
-        llm_config=llm_config,
+        llm_config=llm_config_white,
     )
     
     player_black = ConversableAgent(
@@ -77,7 +76,7 @@ def run_multi_agent(llm, num_turns):
         "Then call make_move(move) to make a move. "
         "After a move is made, analyze the move using technical jargon, don't make any other comments. "
         "Then continue playing.",
-        llm_config=llm_config,
+        llm_config=llm_config_black,
     )
     
     for caller in [player_white, player_black]:
@@ -142,7 +141,7 @@ def run_multi_agent(llm, num_turns):
             player = "Player White"
 
         if turn_num > 0:
-            result += f"<b>{player}, {turn_num}:</b> {chat.get('content')}<br />{board_svgs[turn_num - 1]}<br />"
+            result += f"<b>{player}, Move {turn_num}:</b> {chat.get('content')}<br />{board_svgs[turn_num - 1]}<br />"
         
         turn_num += 1