Update app.py

app.py

@@ -1,7 +1,9 @@
 import gradio as gr
 import random
 
-# Simulated AI models for checkers
+# Simulated AI models for checkers (replace these with actual AI model imports later)
+from transformers import pipeline  # Example for using real AI models
+
 class BasicAI:
     def move(self, board):
         # Basic AI makes a random valid move
@@ -22,10 +24,12 @@ class AdvancedAI:
         # Simulate valid moves as random pairs of coordinates
         return [(random.randint(0, 7), random.randint(0, 7)) for _ in range(random.randint(1, 5))]
 
-# Initialize AI models
+# Replace BasicAI and AdvancedAI with actual models (example Hugging Face pipelines)
 ai_models = {
     "Basic AI": BasicAI(),
-    "Advanced AI": AdvancedAI()
+    "Advanced AI": AdvancedAI(),
+    "HuggingFace AI 1": pipeline("text-generation", model="gpt2"),
+    "HuggingFace AI 2": pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 }
 
 def play_ai_vs_ai(ai1_name, ai2_name, rounds):
@@ -36,12 +40,23 @@ def play_ai_vs_ai(ai1_name, ai2_name, rounds):
     history = []
 
     for i in range(rounds):
-        ai1_move = ai1.move(board)
-        ai2_move = ai2.move(board)
+        ai1_move = ai1.move(board) if hasattr(ai1, 'move') else "Simulated AI 1 Move"
+        ai2_move = ai2.move(board) if hasattr(ai2, 'move') else "Simulated AI 2 Move"
         history.append(f"Round {i + 1}: {ai1_name} -> {ai1_move}, {ai2_name} -> {ai2_move}")
 
     return "\n".join(history)
 
+def render_board():
+    board_html = "<table style='border-collapse: collapse; width: 100px; height: 100px;'>"
+    for i in range(8):
+        board_html += "<tr>"
+        for j in range(8):
+            color = "#f0d9b5" if (i + j) % 2 == 0 else "#b58863"
+            board_html += f"<td style='width: 40px; height: 40px; background-color: {color}; border: 1px solid black;'></td>"
+        board_html += "</tr>"
+    board_html += "</table>"
+    return board_html
+
 # Gradio Interface
 def build_interface():
     with gr.Blocks() as interface:
@@ -60,17 +75,26 @@ def build_interface():
             start_button = gr.Button("Start AI vs AI")
 
         output_box = gr.Textbox(label="Game History", lines=10)
+        board_output = gr.HTML(render_board(), label="Checkers Board")
 
         start_button.click(
             play_ai_vs_ai,
             inputs=[ai1_dropdown, ai2_dropdown, rounds_slider],
-            outputs=output_box
+            outputs=[output_box]
         )
 
+        gr.HTML("""
+        <div style='margin-top: 20px;'>
+            <h3>Game Board</h3>
+            <div>""" + render_board() + """</div>
+        </div>
+        """)
+
         gr.Markdown("""
         ### How It Works
         - **Basic AI**: Makes random valid moves.
         - **Advanced AI**: Uses a placeholder strategy to prioritize specific moves.
+        - **HuggingFace AI Models**: These use text generation pipelines for creative move generation.
         
         Select models, adjust the rounds, and press "Start AI vs AI" to see the results!
         """)