Update multi_agent.py

multi_agent.py

@@ -1,12 +1,11 @@
-import cairosvg, chess, chess.svg, os
+import chess, chess.svg
 from autogen import ConversableAgent, register_function
-from IPython.display import SVG
 from typing_extensions import Annotated
 
 made_move = False
 
 board = chess.Board()
-board_pngs = []
+board_svgs = []
 
 def convert_svg_to_png(svg_file_path: str, png_file_path: str) -> None:
     cairosvg.svg2png(url=svg_file_path, write_to=png_file_path)
@@ -22,15 +21,13 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
     global made_move
     made_move = True
 
-    png = SVG(convert_svg_to_png(chess.svg.board(
+    svg = convert_svg_to_png(chess.svg.board(
         board,
         arrows=[(move.from_square, move.to_square)],
         fill={move.from_square: "gray"},
-        size=200
-    )))
+        size=250
+    ))
 
-    print(png)
-    
     piece = board.piece_at(move.to_square)
     piece_symbol = piece.unicode_symbol()
     piece_name = (
@@ -39,7 +36,7 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
         else chess.piece_name(piece.piece_type)
     )
 
-    board_pngs.append(png)
+    board_svgs.append(svg)
 
     return f"Moved {piece_name} ({piece_symbol}) from "\
     f"{chess.SQUARE_NAMES[move.from_square]} to "\
@@ -147,7 +144,7 @@ def run_multi_agent(llm_white, llm_black, num_turns):
             player = "Player White"
 
         if turn_num > 0:
-            result += f"**{player}, Move {turn_num}**\n<br>{chat.get('content')}<br>{board_pngs[turn_num - 1]}<br>"
+            result += f"**{player}, Move {turn_num}**\n<br>{chat.get('content')}<br>{board_svgs[turn_num - 1]}<br>"
         
         turn_num += 1