Create app.py

app.py

@@ -0,0 +1,142 @@
+import gradio as gr
+import os
+from huggingface_hub import hf_hub_download
+from game_logic import AaduPulliEnv, NeuralNetwork, AlphaZeroAgent
+
+REPO_ID = "AaduPulliAttam/apa-ray" 
+MODEL_FILENAME = "alphazero_aadu_pulli_ray.weights.h5"
+HUMAN_PLAYER = 0  # 0 for Goat, 1 for Tiger
+AI_PLAYER = 1
+
+ai_agent = None
+try:
+    print(f"Downloading model from {REPO_ID}...")
+    model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
+    print(f"Model downloaded to: {model_path}")
+
+    env_for_init = AaduPulliEnv()
+    action_space_size = env_for_init.action_space.n
+    neural_net = NeuralNetwork(action_space_size)
+    
+    neural_net.model.load_weights(model_path)
+    print("Model weights loaded successfully from the Hub.")
+    
+    ai_agent = AlphaZeroAgent(env_for_init, neural_net, simulations_per_move=50)
+
+except Exception as e:
+    print(f"Could not load model from the Hub. AI will be disabled. Error: {e}")
+
+
+def new_game():
+    env = AaduPulliEnv()
+    env.reset()
+    board_image = env.render()
+    message = "New game started. You are the Goat (Blue). Make your first placement."
+    return env, board_image, message, False
+
+def get_action_from_str(env, move_str):
+    try:
+        is_placement_phase = env.goats_placed_count < env.NUM_GOATS
+        if is_placement_phase and env.player_turn == HUMAN_PLAYER:
+            pos = int(move_str)
+            if not 1 <= pos <= env.BOARD_POSITIONS:
+                return None, "Invalid position. Enter a number on the board."
+            return pos - 1, None  # Convert to 0-indexed action
+        else:
+            parts = move_str.split(',')
+            if len(parts) != 2:
+                return None, "Invalid move format. Use 'from,to' (e.g., '12,13')."
+            from_pos, to_pos = int(parts[0]), int(parts[1])
+            move_tuple = (from_pos, to_pos)
+            if move_tuple not in env._move_action_lookup:
+                return None, "Invalid move. That path does not exist."
+            action = env.placement_actions + env._move_action_lookup[move_tuple]
+            return action, None
+    except (ValueError, IndexError):
+        return None, "Invalid input. Please enter a position number or a 'from,to' move."
+
+def check_game_over(info):
+    winner = info.get('winner', -1)
+    if winner != -1:
+        if winner == HUMAN_PLAYER:
+            return "Congratulations, you won!", True
+        elif winner == AI_PLAYER:
+            return "The AI has won. Better luck next time!", True
+        else:
+            return "The game is a draw!", True
+    return None, False
+
+def play_turn(move_str, env, game_over):
+    if game_over:
+        return env, env.render(), "The game is over. Please start a new game.", True
+        
+    if not ai_agent:
+        return env, env.render(), "AI Agent is not available. Please check the logs.", True
+
+    if not move_str:
+        return env, env.render(), "Please enter a move.", False
+
+    # Human's Turn
+    action, error_msg = get_action_from_str(env, move_str)
+    if error_msg:
+        return env, env.render(), error_msg, False
+
+    is_valid, details = env.is_action_valid(action)
+    if not is_valid:
+        return env, env.render(), f"Invalid Move: {details.get('error', 'Unknown error')}", False
+
+    state, _, done, info = env.step(action)
+    message, game_over = check_game_over(info)
+    if game_over:
+        return env, env.render(), message, True
+
+    # AI's Turn
+    ai_action = ai_agent.get_action(state, training=False)
+    state, _, done, info = env.step(ai_action)
+    
+    action_type, details_txt = "Placement", ai_action
+    if ai_action >= env.placement_actions:
+        action_type, details_txt = "Move", env._move_action_map[ai_action - env.placement_actions]
+    
+    message, game_over = check_game_over(info)
+    if game_over:
+        return env, env.render(), message, True
+        
+    message = f"AI played {action_type} {details_txt}. Your turn."
+    return env, env.render(), message, False
+
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    game_state = gr.State()
+    game_over_state = gr.State(False)
+
+    gr.Markdown("# Aadu Puli Aattam: Human vs AlphaZero AI")
+    gr.Markdown("You play as the Goats (Blue). Your goal is to trap the three Tigers (Red). The AI plays as the Tigers, trying to capture 10 of your goats.")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            board_image = gr.Image(label="Game Board", interactive=False)
+        with gr.Column(scale=1):
+            status_message = gr.Markdown(label="Game Status")
+            move_input = gr.Textbox(label="Your Move", placeholder="Enter placement # or 'from,to'")
+            submit_button = gr.Button("Submit Move")
+            new_game_button = gr.Button("New Game")
+
+    demo.load(new_game, outputs=[game_state, board_image, status_message, game_over_state])
+
+    submit_button.click(
+        fn=play_turn,
+        inputs=[move_input, game_state, game_over_state],
+        outputs=[game_state, board_image, status_message, game_over_state]
+    ).then(
+        fn=lambda: "",
+        outputs=[move_input]
+    )
+    
+    new_game_button.click(
+        fn=new_game,
+        outputs=[game_state, board_image, status_message, game_over_state]
+    )
+
+if __name__ == "__main__":
+    demo.launch(share=True)