Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import random
 
-# Initialize board, pieces, and game state
+# Initialize the board, pieces, and game state
 class CheckersGame:
     def __init__(self):
         self.board = self.create_board()
@@ -19,10 +19,10 @@ class CheckersGame:
                 board[row][col] = 'white'
         return board
 
-    def make_move(self):
+    def make_move(self, model_1, model_2):
         # AI logic: Random move for the sake of simplicity
         if self.game_over:
-            return self.board, "Game Over! Restart to play again."
+            return self.render_board(), "Game Over! Restart to play again."
         
         valid_moves = self.get_valid_moves(self.turn)
         if valid_moves:
@@ -30,10 +30,10 @@ class CheckersGame:
             self.board[move[2]][move[3]] = self.board[move[0]][move[1]]
             self.board[move[0]][move[1]] = ''
             self.turn = 'white' if self.turn == 'red' else 'red'
-            return self.board, f"{self.turn.capitalize()} moved."
+            return self.render_board(), f"{self.turn.capitalize()} moved using {model_1 if self.turn == 'white' else model_2}."
         else:
             self.game_over = True
-            return self.board, f"Game Over! {self.turn.capitalize()} has no moves left."
+            return self.render_board(), f"Game Over! {self.turn.capitalize()} has no moves left."
 
     def get_valid_moves(self, color):
         valid_moves = []
@@ -54,27 +54,39 @@ class CheckersGame:
             board_repr += ' '.join([cell[0].upper() if cell else '.' for cell in row]) + '\n'
         return board_repr
 
-# Create a Checkers game instance
+# Initialize the Checkers game instance
 game = CheckersGame()
 
 # Function to handle the game move and display
-def play_checkers():
-    board, status = game.make_move()
-    return game.render_board(), status
+def play_checkers(model_1, model_2):
+    board, status = game.make_move(model_1, model_2)
+    return board, status
 
 # Gradio interface
 def create_ui():
     with gr.Blocks() as demo:
         gr.Markdown("## AI vs AI - Watch the Game!")
         
-        # Display the board and game status
         with gr.Row():
-            board_display = gr.Textbox(label="Game Board", interactive=False, value=game.render_board())
-            game_status = gr.Textbox(label="Game Status", interactive=False, value="Game is running...")
+            # Display the board and game status
+            board_display = gr.Textbox(label="Game Board", interactive=False, value=game.render_board(), lines=8, max_lines=8)
+            game_status = gr.Textbox(label="Game Status", interactive=False, value="Game is running...", lines=2)
 
+        with gr.Row():
+            model_dropdown_1 = gr.Dropdown(
+                choices=["GPT-2", "DistilGPT-2", "BERT"], 
+                label="AI Model for Player 1 (White)",
+                value="GPT-2"
+            )
+            model_dropdown_2 = gr.Dropdown(
+                choices=["GPT-2", "DistilGPT-2", "BERT"], 
+                label="AI Model for Player 2 (Red)",
+                value="DistilGPT-2"
+            )
+        
         # Button to simulate a move and update the board
         play_button = gr.Button("AI makes a move")
-        play_button.click(play_checkers, outputs=[board_display, game_status])
+        play_button.click(play_checkers, inputs=[model_dropdown_1, model_dropdown_2], outputs=[board_display, game_status])
     
     return demo